Add OS.__eq__()

Signed-off-by: Zack Cerza <zack@redhat.com>

diff --git a/teuthology/orchestra/opsys.py b/teuthology/orchestra/opsys.py
index 7d6daec2bd376d6a55252bf2c7a75330077afa32..87383403bdc0767f5e5aa6ad72119a29f3b32e0a 100644 (file)
--- a/teuthology/orchestra/opsys.py
+++ b/teuthology/orchestra/opsys.py
@@ -188,3 +188,9 @@ class OS(object):
             .format(name=repr(self.name),
                     version=repr(self.version),
                     codename=repr(self.codename))
+
+    def __eq__(self, other):
+        for slot in self.__slots__:
+            if not getattr(self, slot) == getattr(other, slot):
+                return False
+        return True

diff --git a/teuthology/orchestra/test/test_opsys.py b/teuthology/orchestra/test/test_opsys.py
index c20ff443bf76af1f9997226811877dc6b9233980..b98f3527f4a9c9fa825d7a54d31a059677017523 100644 (file)
--- a/teuthology/orchestra/test/test_opsys.py
+++ b/teuthology/orchestra/test/test_opsys.py
@@ -175,3 +175,11 @@ class TestOS(object):
     def test_codename_no_version(self):
         os = OS(name='ubuntu', codename='trusty')
         assert os.version == '14.04'
+
+    def test_eq_equal(self):
+        os = OS(name='ubuntu', codename='trusty', version='14.04')
+        assert OS(name='ubuntu', codename='trusty', version='14.04') == os
+
+    def test_eq_not_equal(self):
+        os = OS(name='ubuntu', codename='trusty', version='16.04')
+        assert OS(name='ubuntu', codename='trusty', version='14.04') != os