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)