ips: Added secondary marker

wobcom · Aug 30, 2024 · db48ad1 · db48ad1
1 parent ca44163
commit db48ad1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/cosmo/serializer.py b/cosmo/serializer.py
@@ -52,8 +52,9 @@ def add_ip(self, ipa, is_secondary=False):
     def render_addresses(self):
         retVal = {}
         primaryMarker = {"primary": True} if len(self.secondaryIPs) > 0 else {}
+        secondaryMarker = {"secondary": True} if len(self.secondaryIPs) > 0 else {}
         retVal[self.primaryIP.with_prefixlen] = primaryMarker
-        secondaryIPs = {x.with_prefixlen: {} for x in self.secondaryIPs}
+        secondaryIPs = {x.with_prefixlen: secondaryMarker for x in self.secondaryIPs}
         retVal.update(secondaryIPs)
         return retVal
 

diff --git a/cosmo/tests/test_serializer.py b/cosmo/tests/test_serializer.py
@@ -90,11 +90,11 @@ def test_router_ips():
 
     assert unit_14['families']['inet']['address']['45.139.138.1/29'] == {}
     assert unit_14['families']['inet']['address']['45.139.138.8/29'] == {"primary": True}
-    assert unit_14['families']['inet']['address']['45.139.138.9/29'] == {}
+    assert unit_14['families']['inet']['address']['45.139.138.9/29'] == {"secondary": True}
 
     assert unit_16['families']['inet6']['address']['2a0e:b941:2::/122'] == {}
     assert unit_16['families']['inet6']['address']['2a0e:b941:2::40/122'] == {"primary": True}
-    assert unit_16['families']['inet6']['address']['2a0e:b941:2::41/122'] == {}
+    assert unit_16['families']['inet6']['address']['2a0e:b941:2::41/122'] == {"secondary": True}
 
 def test_router_case_mpls_evpn():