diff --git a/validator/bridgedomain.py b/validator/bridgedomain.py index 4907c45..e80120e 100644 --- a/validator/bridgedomain.py +++ b/validator/bridgedomain.py @@ -21,6 +21,16 @@ class NullHandler(logging.Handler): pass +def get_bridgedomains(yaml): + """ Return a list of all bridgedomains. """ + ret = [] + if not 'bridgedomains' in yaml: + return ret + for ifname, iface in yaml['bridgedomains'].items(): + ret.append(ifname) + return ret + + def get_by_name(yaml, ifname): """ Return the BridgeDomain by name, if it exists. Return None,None otherwise. """ try: diff --git a/validator/test_bridgedomain.py b/validator/test_bridgedomain.py index ac27f19..52b58b0 100644 --- a/validator/test_bridgedomain.py +++ b/validator/test_bridgedomain.py @@ -34,3 +34,7 @@ class TestBridgeDomainMethods(unittest.TestCase): self.assertIn("BondEthernet0", ifs) self.assertIn("GigabitEthernet1/0/0", ifs) self.assertIn("GigabitEthernet2/0/0.100", ifs) + + def test_get_bridgedomains(self): + ifs = bridgedomain.get_bridgedomains(self.cfg) + self.assertEqual(len(ifs), 3)