From ffd0585418c57c058bc28e0a53d0f0f73bacf9ae Mon Sep 17 00:00:00 2001
From: Erik Reid <erik.reid@geant.org>
Date: Tue, 8 Dec 2020 13:48:47 +0100
Subject: [PATCH] added a unit test for the new patterns

---
 test/test_opsdb_queries.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/test/test_opsdb_queries.py b/test/test_opsdb_queries.py
index 021a2add..da219e27 100644
--- a/test/test_opsdb_queries.py
+++ b/test/test_opsdb_queries.py
@@ -294,9 +294,12 @@ def test_lookup_lg_routers(connection, cached_test_data):
     assert not any(
         r['equipment name'].lower().startswith('junosspace') for r in routers)
 
-    srx = filter(
-        lambda r: r['equipment name'].lower().startswith('srx'), routers)
-    assert all(s['type'] == 'INTERNAL' for s in srx)
+    internal_prefix_patterns = ['srx', 'gts', 'qfx']
+    for prefix in internal_prefix_patterns:
+        filtered = filter(
+            lambda r: r['equipment name'].lower().startswith(prefix), routers)
+        assert all(r['type'] == 'INTERNAL' for r in filtered), \
+            f'not all {prefix}* routers are INTERNAL'
 
 
 CIRCUIT_INFO_SCHEMA = {
-- 
GitLab