test_object_store: use temp dirs instead of 'foo'
authorTay Ray Chuan <rctay89@gmail.com>
Sun, 11 Apr 2010 00:29:21 +0000 (02:29 +0200)
committerJelmer Vernooij <jelmer@samba.org>
Sun, 11 Apr 2010 00:29:21 +0000 (02:29 +0200)
dulwich/tests/test_object_store.py

index b012658720497c8b3a747889afdc884a683fed78..94dbe6a6ae8f8648bab06ac74b8212945ef9225e 100644 (file)
@@ -31,6 +31,7 @@ from dulwich.object_store import (
     )
 import os
 import shutil
+import tempfile
 
 
 testobject = Blob()
@@ -39,12 +40,18 @@ testobject.data = "yummy data"
 
 class SpecificDiskObjectStoreTests(TestCase):
 
+    def setUp(self):
+        self.store_dir = tempfile.mkdtemp()
+
+    def tearDown(self):
+        shutil.rmtree(self.store_dir)
+
     def test_pack_dir(self):
-        o = DiskObjectStore("foo")
-        self.assertEquals(os.path.join("foo", "pack"), o.pack_dir)
+        o = DiskObjectStore(self.store_dir)
+        self.assertEquals(os.path.join(self.store_dir, "pack"), o.pack_dir)
 
     def test_empty_packs(self):
-        o = DiskObjectStore("foo")
+        o = DiskObjectStore(self.store_dir)
         self.assertEquals([], o.packs)
 
 
@@ -95,10 +102,12 @@ class DiskObjectStoreTests(ObjectStoreTests,TestCase):
 
     def setUp(self):
         TestCase.setUp(self)
-        if os.path.exists("foo"):
-            shutil.rmtree("foo")
-        os.makedirs(os.path.join("foo", "pack"))
-        self.store = DiskObjectStore("foo")
+        self.store_dir = tempfile.mkdtemp()
+        self.store = DiskObjectStore(self.store_dir)
+
+    def tearDown(self):
+        TestCase.tearDown(self)
+        shutil.rmtree(self.store_dir)
 
 
 # TODO: MissingObjectFinderTests