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