Skip to content
Snippets Groups Projects
Commit 8478646f authored by Erik Reid's avatar Erik Reid
Browse files

use cache data for classifer test

parent 8c799a73
No related branches found
No related tags found
No related merge requests found
......@@ -41,24 +41,6 @@ def test_coriant_addresses(client):
def test_juniper_addresses(mocker, client):
test_data = [
{"ip_address": "a.b.c.d", "project_name": "AAABBB"},
{"ip_address": "b.c.d.e", "project_name": "CCCCDDDDD"},
{"ip_address": "c.d.e.f", "project_name": "EFEFEFEF"},
{"ip_address": "::1", "project_name": "GGHHGGHH"}
]
class MockedRedis():
def __init__(self, *args, **kwargs):
pass
def get(self, ignored):
return json.dumps(test_data).encode('utf-8')
mocker.patch(
'inventory_provider.routes.common.redis.StrictRedis',
MockedRedis)
response_schema = {
"$schema": "http://json-schema.org/draft-07/schema#",
"type": "array",
......@@ -71,39 +53,16 @@ def test_juniper_addresses(mocker, client):
assert rv.status_code == 200
response_data = json.loads(rv.data.decode('utf-8'))
jsonschema.validate(response_data, response_schema)
assert len(response_data) == len(test_data)
assert set([x['ip_address'] for x in test_data]) == set(response_data)
assert len(response_data) > 0 # test data is not empty
def test_trap_metadata(mocker, client):
test_data = [
{
"equipment": "test-equipment",
"manufacturer": "test-manufacturer",
"vendor": "test-vendor",
"absid": "1234",
"name": "test-service_1"
},
{
"equipment": "test-equipment",
"manufacturer": "test-manufacturer",
"vendor": "test-vendor",
"absid": "2345",
"name": "test-service_2"
}
]
mocked_redis = mocker.patch(
"inventory_provider.routes.common.get_redis")
mocked_redis.return_value.hget.return_value = json.dumps(test_data)\
.encode("utf-8")
def test_trap_metadata(client_with_mocked_data):
response_schema = {
"$schema": "http://json-schema.org/draft-07/schema#",
"type": "object"
}
rv = client.get(
'/classifier/trap-metadata/BGP/dummy-equipment/xe-2/3/1',
rv = client_with_mocked_data.get(
'/classifier/trap-metadata/BGP/mx1.ams.nl.geant.net/ae15.1500',
headers=DEFAULT_REQUEST_HEADERS)
assert rv.status_code == 200
assert rv.is_json
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment