]> git.rkrishnan.org Git - tahoe-lafs/tahoe-lafs.git/commitdiff
more runner.py test coverage: don't bypass argv parsing
authorBrian Warner <warner@lothar.com>
Tue, 26 Jun 2007 23:51:38 +0000 (16:51 -0700)
committerBrian Warner <warner@lothar.com>
Tue, 26 Jun 2007 23:51:38 +0000 (16:51 -0700)
src/allmydata/test/test_system.py
src/allmydata/test/test_vdrive.py

index 509b7f46aec6edc5df4125a88d3f867b33cb073c..4f14731c2527643ca665c494a4d8ac7e2dc4db8a 100644 (file)
@@ -376,9 +376,12 @@ class SystemTest(testutil.SignalMixin, unittest.TestCase):
         log.msg("test_system.SystemTest._test_runner using %s" % dirpath)
 
         filename = os.path.join(dirpath, "uri_extension")
-        s = StringIO()
-        rc = runner.dump_uri_extension({'filename': filename}, s)
-        output = s.getvalue()
+
+        out,err = StringIO(), StringIO()
+        rc = runner.runner(["dump-uri-extension",
+                            filename],
+                           stdout=out, stderr=err)
+        output = out.getvalue()
         self.failUnlessEqual(rc, 0)
 
         # we only upload a single file, so we can assert some things about
index e5a49b11597769d728bc17c550d4084b480905e7..bfd9415a1f254302369b142ed79542a118eb19eb 100644 (file)
@@ -244,19 +244,22 @@ class Test(unittest.TestCase):
 
     def _test_one_3(self, res):
         # now test some of the diag tools with the data we've created
-        s = StringIO()
-        rc = runner.dump_root_dirnode("vdrive/test_one", {}, s)
-        output = s.getvalue()
+        out,err = StringIO(), StringIO()
+        rc = runner.runner(["dump-root-dirnode", "vdrive/test_one"],
+                           stdout=out, stderr=err)
+        output = out.getvalue()
         self.failUnless(output.startswith("URI:DIR:fakeFURL:"))
         self.failUnlessEqual(rc, 0)
 
-        s = StringIO()
-        args = {'uri': self.bar_node.get_uri(),
-                'verbose': True,
-                }
-        rc = runner.dump_directory_node("vdrive/test_one", args, s)
-        output = s.getvalue()
+        out,err = StringIO(), StringIO()
+        rc = runner.runner(["dump-dirnode",
+                            "--basedir", "vdrive/test_one",
+                            "--verbose",
+                            self.bar_node.get_uri()],
+                           stdout=out, stderr=err)
+        output = out.getvalue()
         #print output
+        self.failUnlessEqual(rc, 0)
         self.failUnless("dirnode uri: URI:DIR:myFURL" in output)
         self.failUnless("write_enabler" in output)
         self.failIf("write_enabler: None" in output)
@@ -265,15 +268,16 @@ class Test(unittest.TestCase):
         self.failUnless(" read: URI:DIR-RO:myFURL:" in output)
         self.failUnless("key file4\n" in output)
         self.failUnless("H_key " in output)
-        self.failUnlessEqual(rc, 0)
 
-        s = StringIO()
-        args = {'uri': self.bar_node.get_uri(),
-                'verbose': False,
-                }
-        rc = runner.dump_directory_node("vdrive/test_one", args, s)
-        output = s.getvalue()
+        out,err = StringIO(), StringIO()
+        rc = runner.runner(["dump-dirnode",
+                            "--basedir", "vdrive/test_one",
+                            # non-verbose
+                            "--uri", self.bar_node.get_uri()],
+                           stdout=out, stderr=err)
+        output = out.getvalue()
         #print output
+        self.failUnlessEqual(rc, 0)
         self.failUnless("dirnode uri: URI:DIR:myFURL" in output)
         self.failUnless("write_enabler" in output)
         self.failIf("write_enabler: None" in output)
@@ -282,22 +286,22 @@ class Test(unittest.TestCase):
         self.failUnless(" read: URI:DIR-RO:myFURL:" in output)
         self.failUnless("key file4\n" in output)
         self.failIf("H_key " in output)
-        self.failUnlessEqual(rc, 0)
 
-        s = StringIO()
-        args = {'uri': self.bar_node_readonly.get_uri(),
-                'verbose': True,
-                }
-        rc = runner.dump_directory_node("vdrive/test_one", args, s)
-        output = s.getvalue()
+        out,err = StringIO(), StringIO()
+        rc = runner.runner(["dump-dirnode",
+                            "--basedir", "vdrive/test_one",
+                            "--verbose",
+                            self.bar_node_readonly.get_uri()],
+                           stdout=out, stderr=err)
+        output = out.getvalue()
         #print output
+        self.failUnlessEqual(rc, 0)
         self.failUnless("dirnode uri: URI:DIR-RO:myFURL" in output)
         self.failUnless("write_enabler: None" in output)
         self.failUnless("key baz\n" in output)
         self.failIf(" write: URI:DIR:myFURL:" in output)
         self.failUnless(" read: URI:DIR-RO:myFURL:" in output)
         self.failUnless("key file4\n" in output)
-        self.failUnlessEqual(rc, 0)
 
     def shouldFail(self, res, expected_failure, which, substring=None):
         if isinstance(res, failure.Failure):