From 0ab907f15595c365521fe38c568d0319a5dbb8ab Mon Sep 17 00:00:00 2001 From: Pim van Pelt Date: Tue, 22 Mar 2022 18:57:45 +0000 Subject: [PATCH] Add get_bridgedomains() --- validator/bridgedomain.py | 10 ++++++++++ validator/test_bridgedomain.py | 4 ++++ 2 files changed, 14 insertions(+) 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)