move check_sourcedata to SourceCommandTestMixin

djmitche · Jun 10, 2010 · 5da76a2 · 5da76a2
1 parent a074a7e
commit 5da76a2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/slave/buildslave/test/util/command.py b/slave/buildslave/test/util/command.py
@@ -103,11 +103,3 @@ def clean_environ(self):
         Temporarily clean out os.environ to { 'PWD' : '.' }
         """
         self.patch(os, 'environ', { 'PWD' : '.' })
-
-    def check_sourcedata(self, _, expected_sourcedata):
-        """
-        Assert that the sourcedata (from the patched sourcedata_fns - see
-        make_command) is correct.  Use this as a deferred callback.
-        """
-        self.assertEqual(self.sourcedata, expected_sourcedata)
-        return _
diff --git a/slave/buildslave/test/util/sourcecommand.py b/slave/buildslave/test/util/sourcecommand.py
@@ -52,3 +52,11 @@ def doCopy(_):
                 self.builder.basedir)
             return r.start()
         cmd.doCopy = doCopy
+
+    def check_sourcedata(self, _, expected_sourcedata):
+        """
+        Assert that the sourcedata (from the patched functions - see
+        make_command) is correct.  Use this as a deferred callback.
+        """
+        self.assertEqual(self.sourcedata, expected_sourcedata)
+        return _