d11c40dcac
This ensures `isort` is applied to all files in the repo. Change-Id: Ib7ced1c924ef1639542bf0d1a01c5737f6ba43e9
409 lines
15 KiB
Python
409 lines
15 KiB
Python
# Copyright (c) 2023 The Regents of the University of California
|
|
# All rights reserved.
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without
|
|
# modification, are permitted provided that the following conditions are
|
|
# met: redistributions of source code must retain the above copyright
|
|
# notice, this list of conditions and the following disclaimer;
|
|
# redistributions in binary form must reproduce the above copyright
|
|
# notice, this list of conditions and the following disclaimer in the
|
|
# documentation and/or other materials provided with the distribution;
|
|
# neither the name of the copyright holders nor the names of its
|
|
# contributors may be used to endorse or promote products derived from
|
|
# this software without specific prior written permission.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
import copy
|
|
import json
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
import mongomock
|
|
from api.mongo_client import MongoDBClient
|
|
from bson import json_util
|
|
from server import app
|
|
|
|
|
|
class TestComprehensive(unittest.TestCase):
|
|
@patch.object(
|
|
MongoDBClient,
|
|
"_get_database",
|
|
return_value=mongomock.MongoClient().db.collection,
|
|
)
|
|
def setUp(self, mock_get_database):
|
|
"""This method sets up the test environment."""
|
|
self.ctx = app.app_context()
|
|
self.ctx.push()
|
|
self.app = app
|
|
self.test_client = app.test_client()
|
|
self.alias = "test"
|
|
objects = []
|
|
with open("./test/refs/resources.json", "rb") as f:
|
|
objects = json.loads(f.read(), object_hook=json_util.object_hook)
|
|
self.collection = mock_get_database()
|
|
for obj in objects:
|
|
self.collection.insert_one(obj)
|
|
|
|
self.test_client.post(
|
|
"/validateMongoDB",
|
|
json={
|
|
"uri": "mongodb://localhost:27017",
|
|
"database": "test",
|
|
"collection": "test",
|
|
"alias": self.alias,
|
|
},
|
|
)
|
|
|
|
def tearDown(self):
|
|
"""This method tears down the test environment."""
|
|
self.collection.drop()
|
|
self.ctx.pop()
|
|
|
|
def test_insert_find_update_find(self):
|
|
test_resource = {
|
|
"category": "diskimage",
|
|
"id": "test-resource",
|
|
"author": ["test-author"],
|
|
"description": "test-description",
|
|
"license": "test-license",
|
|
"source_url": (
|
|
"https://github.com/gem5/gem5-resources/"
|
|
"tree/develop/src/x86-ubuntu"
|
|
),
|
|
"tags": ["test-tag", "test-tag2"],
|
|
"example_usage": " test-usage",
|
|
"gem5_versions": [
|
|
"22.1",
|
|
],
|
|
"resource_version": "1.0.0",
|
|
}
|
|
original_resource = test_resource.copy()
|
|
test_id = test_resource["id"]
|
|
test_resource_version = test_resource["resource_version"]
|
|
# insert resource
|
|
response = self.test_client.post(
|
|
"/insert", json={"resource": test_resource, "alias": self.alias}
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json, {"status": "Inserted"})
|
|
# find resource
|
|
response = self.test_client.post(
|
|
"/find",
|
|
json={
|
|
"id": test_id,
|
|
"resource_version": test_resource_version,
|
|
"alias": self.alias,
|
|
},
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertTrue(response.json == test_resource)
|
|
|
|
# update resource
|
|
test_resource["description"] = "test-description-2"
|
|
test_resource["author"].append("test-author-2")
|
|
response = self.test_client.post(
|
|
"/update",
|
|
json={
|
|
"original_resource": original_resource,
|
|
"resource": test_resource,
|
|
"alias": self.alias,
|
|
},
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json, {"status": "Updated"})
|
|
# find resource
|
|
response = self.test_client.post(
|
|
"/find",
|
|
json={
|
|
"id": test_id,
|
|
"resource_version": test_resource_version,
|
|
"alias": self.alias,
|
|
},
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertTrue(response.json == test_resource)
|
|
|
|
def test_find_new_insert(self):
|
|
test_resource = {
|
|
"category": "diskimage",
|
|
"id": "test-resource",
|
|
"author": ["test-author"],
|
|
"description": "test-description",
|
|
"license": "test-license",
|
|
"source_url": (
|
|
"https://github.com/gem5/gem5-resources/"
|
|
"tree/develop/src/x86-ubuntu"
|
|
),
|
|
"tags": ["test-tag", "test-tag2"],
|
|
"example_usage": " test-usage",
|
|
"gem5_versions": [
|
|
"22.1",
|
|
],
|
|
"resource_version": "1.0.0",
|
|
}
|
|
test_id = test_resource["id"]
|
|
test_resource_version = test_resource["resource_version"]
|
|
# find resource
|
|
response = self.test_client.post(
|
|
"/find",
|
|
json={
|
|
"id": test_id,
|
|
"resource_version": test_resource_version,
|
|
"alias": self.alias,
|
|
},
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json, {"exists": False})
|
|
# insert resource
|
|
response = self.test_client.post(
|
|
"/insert", json={"resource": test_resource, "alias": self.alias}
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json, {"status": "Inserted"})
|
|
# find resource
|
|
response = self.test_client.post(
|
|
"/find",
|
|
json={
|
|
"id": test_id,
|
|
"resource_version": test_resource_version,
|
|
"alias": self.alias,
|
|
},
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertTrue(response.json == test_resource)
|
|
|
|
def test_insert_find_new_version_find_older(self):
|
|
test_resource = {
|
|
"category": "diskimage",
|
|
"id": "test-resource",
|
|
"author": ["test-author"],
|
|
"description": "test-description",
|
|
"license": "test-license",
|
|
"source_url": (
|
|
"https://github.com/gem5/gem5-resources/"
|
|
"tree/develop/src/x86-ubuntu"
|
|
),
|
|
"tags": ["test-tag", "test-tag2"],
|
|
"example_usage": " test-usage",
|
|
"gem5_versions": [
|
|
"22.1",
|
|
],
|
|
"resource_version": "1.0.0",
|
|
}
|
|
test_id = test_resource["id"]
|
|
test_resource_version = test_resource["resource_version"]
|
|
# insert resource
|
|
response = self.test_client.post(
|
|
"/insert", json={"resource": test_resource, "alias": self.alias}
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json, {"status": "Inserted"})
|
|
# find resource
|
|
response = self.test_client.post(
|
|
"/find",
|
|
json={
|
|
"id": test_id,
|
|
"resource_version": test_resource_version,
|
|
"alias": self.alias,
|
|
},
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertTrue(response.json == test_resource)
|
|
|
|
# add new version
|
|
test_resource_new_version = copy.deepcopy(test_resource)
|
|
test_resource_new_version["description"] = "test-description-2"
|
|
test_resource_new_version["author"].append("test-author-2")
|
|
test_resource_new_version["resource_version"] = "1.0.1"
|
|
|
|
response = self.test_client.post(
|
|
"/insert",
|
|
json={"resource": test_resource_new_version, "alias": self.alias},
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json, {"status": "Inserted"})
|
|
|
|
# get resource versions
|
|
response = self.test_client.post(
|
|
"/versions", json={"id": test_id, "alias": self.alias}
|
|
)
|
|
return_json = json.loads(response.data)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(
|
|
return_json,
|
|
[{"resource_version": "1.0.1"}, {"resource_version": "1.0.0"}],
|
|
)
|
|
|
|
resource_version = return_json[1]["resource_version"]
|
|
# find older version
|
|
response = self.test_client.post(
|
|
"/find",
|
|
json={
|
|
"id": test_id,
|
|
"resource_version": resource_version,
|
|
"alias": self.alias,
|
|
},
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertTrue(response.json == test_resource)
|
|
|
|
def test_find_add_new_version_delete_older(self):
|
|
test_resource = {
|
|
"category": "binary",
|
|
"id": "binary-example",
|
|
"description": "binary-example documentation.",
|
|
"architecture": "ARM",
|
|
"is_zipped": False,
|
|
"md5sum": "71b2cb004fe2cda4556f0b1a38638af6",
|
|
"url": (
|
|
"http://dist.gem5.org/dist/develop/"
|
|
"test-progs/hello/bin/arm/linux/hello64-static"
|
|
),
|
|
"source": "src/simple",
|
|
"resource_version": "1.0.0",
|
|
"gem5_versions": ["23.0"],
|
|
}
|
|
test_id = test_resource["id"]
|
|
test_resource_version = test_resource["resource_version"]
|
|
# find resource
|
|
response = self.test_client.post(
|
|
"/find",
|
|
json={
|
|
"id": test_id,
|
|
"resource_version": test_resource_version,
|
|
"alias": self.alias,
|
|
},
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertTrue(response.json == test_resource)
|
|
|
|
# add new version
|
|
test_resource_new_version = copy.deepcopy(test_resource)
|
|
test_resource_new_version["description"] = "test-description-2"
|
|
test_resource_new_version["resource_version"] = "1.0.1"
|
|
|
|
response = self.test_client.post(
|
|
"/insert",
|
|
json={"resource": test_resource_new_version, "alias": self.alias},
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json, {"status": "Inserted"})
|
|
|
|
# get resource versions
|
|
response = self.test_client.post(
|
|
"/versions", json={"id": test_id, "alias": self.alias}
|
|
)
|
|
return_json = json.loads(response.data)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(
|
|
return_json,
|
|
[{"resource_version": "1.0.1"}, {"resource_version": "1.0.0"}],
|
|
)
|
|
# delete older version
|
|
response = self.test_client.post(
|
|
"/delete", json={"resource": test_resource, "alias": self.alias}
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json, {"status": "Deleted"})
|
|
|
|
# get resource versions
|
|
response = self.test_client.post(
|
|
"/versions", json={"id": test_id, "alias": self.alias}
|
|
)
|
|
return_json = json.loads(response.data)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(return_json, [{"resource_version": "1.0.1"}])
|
|
|
|
def test_find_add_new_version_update_older(self):
|
|
test_resource = {
|
|
"category": "binary",
|
|
"id": "binary-example",
|
|
"description": "binary-example documentation.",
|
|
"architecture": "ARM",
|
|
"is_zipped": False,
|
|
"md5sum": "71b2cb004fe2cda4556f0b1a38638af6",
|
|
"url": (
|
|
"http://dist.gem5.org/dist/develop/"
|
|
"test-progs/hello/bin/arm/linux/hello64-static"
|
|
),
|
|
"source": "src/simple",
|
|
"resource_version": "1.0.0",
|
|
"gem5_versions": ["23.0"],
|
|
}
|
|
original_resource = test_resource.copy()
|
|
test_id = test_resource["id"]
|
|
test_resource_version = test_resource["resource_version"]
|
|
# find resource
|
|
response = self.test_client.post(
|
|
"/find",
|
|
json={
|
|
"id": test_id,
|
|
"resource_version": test_resource_version,
|
|
"alias": self.alias,
|
|
},
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertTrue(response.json == test_resource)
|
|
|
|
# add new version
|
|
test_resource_new_version = copy.deepcopy(test_resource)
|
|
test_resource_new_version["description"] = "test-description-2"
|
|
test_resource_new_version["resource_version"] = "1.0.1"
|
|
|
|
response = self.test_client.post(
|
|
"/insert",
|
|
json={"resource": test_resource_new_version, "alias": self.alias},
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json, {"status": "Inserted"})
|
|
|
|
# get resource versions
|
|
response = self.test_client.post(
|
|
"/versions", json={"id": test_id, "alias": self.alias}
|
|
)
|
|
return_json = json.loads(response.data)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(
|
|
return_json,
|
|
[{"resource_version": "1.0.1"}, {"resource_version": "1.0.0"}],
|
|
)
|
|
|
|
resource_version = return_json[1]["resource_version"]
|
|
|
|
# update older version
|
|
test_resource["description"] = "test-description-3"
|
|
|
|
response = self.test_client.post(
|
|
"/update",
|
|
json={
|
|
"original_resource": original_resource,
|
|
"resource": test_resource,
|
|
"alias": self.alias,
|
|
},
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json, {"status": "Updated"})
|
|
|
|
# find resource
|
|
response = self.test_client.post(
|
|
"/find",
|
|
json={
|
|
"id": test_id,
|
|
"resource_version": resource_version,
|
|
"alias": self.alias,
|
|
},
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertTrue(response.json == test_resource)
|