diff --git a/test/test_opsdb_queries.py b/test/test_opsdb_queries.py index 021a2adda1cc9429993cad268f55c3e8e557b3fd..da219e271fd717a46655b5b6721ffa027d3fbe01 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 = {