Răsfoiți Sursa

Add equality testing method for key pairs and a test.

Enis Afgan 10 ani în urmă
părinte
comite
d462c5dc5e
2 a modificat fișierele cu 8 adăugiri și 0 ștergeri
  1. 5 0
      cloudbridge/providers/base.py
  2. 3 0
      test/test_provider_security_service.py

+ 5 - 0
cloudbridge/providers/base.py

@@ -217,6 +217,11 @@ class BaseKeyPair(KeyPair):
         self._provider = provider
         self._key_pair = key_pair
 
+    def __eq__(self, other):
+        return isinstance(other, KeyPair) and \
+            self._provider == other._provider and \
+            self.name == other.name
+
     @property
     def name(self):
         """

+ 3 - 0
test/test_provider_security_service.py

@@ -53,6 +53,9 @@ class ProviderSecurityServiceTestCase(ProviderTestBase):
             self.assertIsNotNone(
                 kp.material,
                 "KeyPair material is empty but it should not be.")
+            self.assertTrue(
+                kp == kp,
+                "The same key pair should be equal to self.")
             kp.delete()
             kpl = self.provider.security.key_pairs.list()
             found_kp = [k for k in kpl if k.name == name]