def test_get_peer(self):
cra = RelationshipAssociation(relationship=self.m2m_1, source=self.racks[0], destination=self.vlans[0])
cra.save()
self.assertEqual(cra.get_peer(self.racks[0]), self.vlans[0])
self.assertEqual(cra.get_peer(self.vlans[0]), self.racks[0])
self.assertEqual(cra.get_peer(self.vlans[1]), None)
def test_get_peer(self):
"""Validate that the get_peer() method works correctly."""
cra = RelationshipAssociation(relationship=self.m2m_1,
source=self.racks[0],
destination=self.vlans[0])
cra.validated_save()
self.assertEqual(cra.get_peer(self.racks[0]), self.vlans[0])
self.assertEqual(cra.get_peer(self.vlans[0]), self.racks[0])
self.assertEqual(cra.get_peer(self.vlans[1]), None)