Add run.Raw.__eq__()

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

diff --git a/teuthology/orchestra/run.py b/teuthology/orchestra/run.py
index 7f56900e46b526ba417a916e2f930b355c12d984..80b3b0727235c4955a0660127438083a925d047a 100644 (file)
--- a/teuthology/orchestra/run.py
+++ b/teuthology/orchestra/run.py
@@ -168,6 +168,9 @@ class Raw(object):
             value=self.value,
             )
 
+    def __eq__(self, value):
+        return self.value == value
+
 
 def quote(args):
     """

diff --git a/teuthology/orchestra/test/test_run.py b/teuthology/orchestra/test/test_run.py
index 5132768cc6334d0ff07f3c827b5f701fec995e68..40464f19c69d33867a9c33e99b6561e77f1ad304 100644 (file)
--- a/teuthology/orchestra/test/test_run.py
+++ b/teuthology/orchestra/test/test_run.py
@@ -431,3 +431,10 @@ class TestRun(object):
     def test_quote_and_raw(self):
         got = run.quote(['true', run.Raw('&&'), 'echo', 'yay'])
         assert got == "true && echo yay"
+
+
+class TestRaw(object):
+    def test_eq(self):
+        str_ = "I am a raw something or other"
+        raw = run.Raw(str_)
+        assert raw == run.Raw(str_)