Test coverage for filestore module.

Signed-off-by: Tim Parkin & Matt Goodall <info@ish.io>
mcdonc · Sep 15, 2009 · f73bb17 · f73bb17
1 parent ce6c1ef
commit f73bb17
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/formish/tests/unittests/test_filestores.py b/formish/tests/unittests/test_filestores.py
@@ -27,6 +27,10 @@ def test_put(self):
         finally:
             f.close()
 
+    def test_put_headersdict(self):
+        self.store.put('foo', {'Cache-Tag':'1', 'Content-Type':'text/plain'}, StringIO("Yay!"))
+        assert file(os.path.join(self.dirname, 'foo'), 'rb').read() == 'Cache-Tag: 1\nContent-Type: text/plain\n\nYay!'
+
     def test_get(self):
         self.store.put('foo', [('Cache-Tag', '1'), ('Content-Type', 'text/plain')], StringIO("Yay!"))
         headers, f = self.store.get('foo')
@@ -53,6 +57,11 @@ def test_delete(self):
         self.store.delete('foo')
         self.assertRaises(KeyError, self.store.get, 'foo')
 
+    def test_delete_glob(self):
+        self.store.put('foo', [], StringIO('foo'))
+        self.store.delete('fo', glob=True)
+        self.assertRaises(KeyError, self.store.get, 'foo')
+
     def test_unicode(self):
         gbp = '£'.decode('utf-8')
         self.store.put('foo', [('a', gbp)], StringIO('foo'))
@@ -75,6 +84,11 @@ def test_put(self):
         self.store.put('foo', StringIO('bar'), '1', [('Content-Type', 'text/plain')])
         assert file(os.path.join(self.dirname, 'foo'), 'rb').read() == 'Cache-Tag: 1\nContent-Type: text/plain\n\nbar'
 
+    def test_put_noheaders(self):
+        self.store.put('foo', StringIO('bar'), '1')
+        self.assertEqual(open(os.path.join(self.dirname, 'foo'), 'rb').read(),
+        'Cache-Tag: 1\n\nbar')
+
     def test_get(self):
         self.store.put('foo', StringIO('bar'), '1', [('Content-Type', 'text/plain')])
         (cache_tag, headers, f) = self.store.get('foo')
@@ -122,3 +136,6 @@ def test_none_cache_tag(self):
         finally:
             f.close()
 
+    def test_delete(self):
+        self.assertRaises(OSError, self.store.delete, 'not_found')
+