From: Daira Hopwood <daira@jacaranda.org>
Date: Tue, 5 Feb 2013 19:42:01 +0000 (+0000)
Subject: test_client.py: cleanups to S3 config tests.
X-Git-Url: https://git.rkrishnan.org/vdrive/components/COPYING.GPL?a=commitdiff_plain;h=0d5b76d7a81d3d014f7eeb199bf3a790b2c5b7b9;p=tahoe-lafs%2Ftahoe-lafs.git

test_client.py: cleanups to S3 config tests.

Signed-off-by: David-Sarah Hopwood <david-sarah@jacaranda.org>
---

diff --git a/src/allmydata/test/test_client.py b/src/allmydata/test/test_client.py
index 2f3ec972..75070ad0 100644
--- a/src/allmydata/test/test_client.py
+++ b/src/allmydata/test/test_client.py
@@ -202,15 +202,15 @@ class Basic(testutil.ReallyEqualMixin, unittest.TestCase):
                                     "reserved_space = bogus\n")
         self.failUnlessRaises(InvalidValueError, client.Client, basedir)
 
-    def _write_s3secret(self, basedir, secret="dummy"):
-        os.mkdir(os.path.join(basedir, "private"))
-        fileutil.write(os.path.join(basedir, "private", "s3secret"), secret)
+    def _write_secret(self, basedir, filename, secret="dummy"):
+        fileutil.make_dirs(os.path.join(basedir, "private"))
+        fileutil.write(os.path.join(basedir, "private", filename), secret)
 
     @mock.patch('allmydata.storage.backends.cloud.s3.s3_container.S3Container')
     def test_s3_config_good_defaults(self, mock_S3Container):
         basedir = "client.Basic.test_s3_config_good_defaults"
         os.mkdir(basedir)
-        self._write_s3secret(basedir)
+        self._write_secret(basedir, "s3secret")
         config = (BASECONFIG +
                   "[storage]\n" +
                   "enabled = true\n" +
@@ -225,11 +225,11 @@ class Basic(testutil.ReallyEqualMixin, unittest.TestCase):
         self.failUnless(isinstance(server.backend, CloudBackend), server.backend)
 
         mock_S3Container.reset_mock()
-        fileutil.write(os.path.join(basedir, "private", "s3producttoken"), "{ProductToken}")
+        self._write_secret(basedir, "s3producttoken", secret="{ProductToken}")
         self.failUnlessRaises(InvalidValueError, client.Client, basedir)
 
         mock_S3Container.reset_mock()
-        fileutil.write(os.path.join(basedir, "private", "s3usertoken"), "{UserToken}")
+        self._write_secret(basedir, "s3usertoken", secret="{UserToken}")
         fileutil.write(os.path.join(basedir, "tahoe.cfg"), config + "s3.url = http://s3.example.com\n")
 
         c = client.Client(basedir)
@@ -239,7 +239,7 @@ class Basic(testutil.ReallyEqualMixin, unittest.TestCase):
     def test_s3_readonly_bad(self):
         basedir = "client.Basic.test_s3_readonly_bad"
         os.mkdir(basedir)
-        self._write_s3secret(basedir)
+        self._write_secret(basedir, "s3secret")
         fileutil.write(os.path.join(basedir, "tahoe.cfg"),
                                     BASECONFIG +
                                     "[storage]\n" +
@@ -319,7 +319,7 @@ class Basic(testutil.ReallyEqualMixin, unittest.TestCase):
     def test_s3_config_no_access_key_id(self):
         basedir = "client.Basic.test_s3_config_no_access_key_id"
         os.mkdir(basedir)
-        self._write_s3secret(basedir)
+        self._write_secret(basedir, "s3secret")
         fileutil.write(os.path.join(basedir, "tahoe.cfg"),
                                     BASECONFIG +
                                     "[storage]\n" +
@@ -331,7 +331,7 @@ class Basic(testutil.ReallyEqualMixin, unittest.TestCase):
     def test_s3_config_no_bucket(self):
         basedir = "client.Basic.test_s3_config_no_bucket"
         os.mkdir(basedir)
-        self._write_s3secret(basedir)
+        self._write_secret(basedir, "s3secret")
         fileutil.write(os.path.join(basedir, "tahoe.cfg"),
                                     BASECONFIG +
                                     "[storage]\n" +