From 2b7ec9f92f18a7c7f9da92728dc9c2194fe997b5 Mon Sep 17 00:00:00 2001
From: Robert Latta <robert.latta@geant.org>
Date: Fri, 25 Jan 2019 11:18:35 +0000
Subject: [PATCH] trap metadata api testing
---
test/test_classifier_routes.py | 36 ++++++++++++++++++++++++++++++++++
1 file changed, 36 insertions(+)
diff --git a/test/test_classifier_routes.py b/test/test_classifier_routes.py
index 114112d1..a01236f2 100644
--- a/test/test_classifier_routes.py
+++ b/test/test_classifier_routes.py
@@ -73,3 +73,39 @@ def test_juniper_addresses(mocker, client):
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)
+
+
+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.classifier.db.get_redis")
+ mocked_redis.return_value.hget.return_value = json.dumps(test_data)\
+ .encode("utf-8")
+
+ 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',
+ headers=DEFAULT_REQUEST_HEADERS)
+ assert rv.status_code == 200
+ assert rv.is_json
+ response_data = json.loads(rv.data.decode('utf-8'))
+ jsonschema.validate(response_data, response_schema)
--
GitLab