Add files from the old svn, r101.

[matthijs/upstream/mobilegtd.git] / tests / specs / persistence / file_based_spec.py

diff --git a/tests/specs/persistence/file_based_spec.py b/tests/specs/persistence/file_based_spec.py
new file mode 100644 (file)
index 0000000..c13b774
--- /dev/null
+++ b/tests/specs/persistence/file_based_spec.py
@@ -0,0 +1,43 @@
+import unittest
+import os,tempfile,shutil
+import inout.io
+from inout import io
+
+
+class FileSystemBasedBehaviour(unittest.TestCase):
+       def setUp(self):
+               self.current_dir = os.getcwd()
+               self.tempdir = tempfile.mkdtemp()
+               os.chdir(self.tempdir)
+       def tearDown(self):
+               os.chdir(self.current_dir)
+               #shutil.rmtree(self.tempdir,True)
+
+       def create_file(self,path):
+               inout.io.create_file(path).close()
+
+       def assertCreatedFile(self,path,error_message = None):
+               if not error_message:
+                       error_message = u"The file %s should have been created"%repr(path)
+               self.assertTrue(os.path.isfile(io.os_encode(path)),error_message)
+
+class FileBasedBehaviour(FileSystemBasedBehaviour):
+
+       def file_content(self):
+               f=file(self.path(),'r')
+               raw=f.read()
+               f.close()
+               return raw
+
+       def create_file(self,path=None):
+               if path == None:
+                       path = self.path()
+               super(FileBasedBehaviour,self).create_file(path)
+
+       def write(self,content,path=None):
+               if path == None:
+                       path = self.path()
+               inout.io.write(path,content)
+
+
+__all__= ["FileBasedBehaviour","FileSystemBasedBehaviour"]