1 files changed, 114 insertions, 0 deletions
diff --git a/tests/test_hgrepo.py b/tests/test_hgrepo.py
new file mode 100644
index 0000000..c67c506
--- /dev/null
+++ b/tests/test_hgrepo.py
@@ -0,0 +1,114 @@
+import unittest
+from testutils import blit, tmpData
+
+from olpc.datastore import hg_backingstore
+import os
+
+DEFAULT_STORE = '/tmp/hgtest'
+DATA_1 = '/tmp/data1'
+DATA_2 = '/tmp/data2'
+
+class Test(unittest.TestCase):
+    def setUp(self):
+        os.system("rm -rf %s" % DEFAULT_STORE)
+        os.system("rm -rf %s" % DATA_1)
+        os.system("rm -rf %s" % DATA_2)
+
+        os.makedirs(DATA_1)
+        os.makedirs(DATA_2)
+        
+    def tearDown(self):
+        os.system("rm -rf %s" % DEFAULT_STORE)
+        os.system("rm -rf %s" % DATA_1)
+        os.system("rm -rf %s" % DATA_2)
+        
+    def test_hgrepo(self):
+        repo = hg_backingstore.FileRepo(DEFAULT_STORE)
+
+        # create a test file in DATA_1
+        TEXT_1= "this is a test"
+        fn1 = blit(TEXT_1, os.path.join(DATA_1, "s1"))
+        # and the version we will use later in DATA_2
+        # we do this to test tracking from different source dirs
+        TEXT_2 = "another test"
+        fn2 = blit(TEXT_2, os.path.join(DATA_2, "s2"))
+
+        # this should add foo to the repo with TEXT_1 data
+        repo.put("foo", fn1, text="initial")
+
+        # now commit fn2 to the same path, will create another
+        # revision
+        # of the existing data
+        repo.put("foo", fn2, text="updated")
+
+        # now verify access to both revisions and their data
+        # we check things out into DATA_1
+        co1 = os.path.join(DATA_1, "co1")
+        co2 = os.path.join(DATA_1, "co2")
+
+        repo.dump("foo", 0, co1)
+        repo.dump("foo", 1, co2)
+
+        assert open(co1, 'r').read() == TEXT_1
+        assert open(co2, 'r').read() == TEXT_2
+        
+    def test_hgbs(self):
+        bs = hg_backingstore.HgBackingStore("hg:%s" % DEFAULT_STORE)
+        bs.initialize_and_load()
+        bs.create_descriptor()
+        desc = bs.descriptor()
+        assert 'id' in desc
+        assert 'uri' in desc
+        assert 'title' in desc
+        assert desc['title'] is not None
+
+        d = """This is a test"""
+        d2 = "Different"
+
+        uid, rev = bs.checkin(dict(title="A", filename="a.txt"), tmpData(d))
+
+        bs.complete_indexing()
+
+        props, fn = bs.checkout(uid)
+
+        assert props.get('title') == "A"
+        got = open(fn, 'r').read()
+        assert got == d
+
+        uid, rev = bs.checkin(dict(uid=uid, title="B"), tmpData(d2))
+
+        bs.complete_indexing()
+        
+        props, fn = bs.checkout(uid)
+        assert props.get('title') == "B"
+        got = open(fn, 'r').read()
+        assert got == d2
+
+        # go back and check out the first version
+        props, fn = bs.checkout(uid, 1)
+        assert props.get('title') == "A"
+        got = open(fn, 'r').read()
+        assert got == d
+        
+        bs.delete(uid, props['vid'])
+        bs.complete_indexing()
+
+        # There is no more revision 2
+        self.failUnlessRaises(KeyError, bs.get, uid, 1)
+
+##         props, fn = bs.checkout(uid)
+
+##         import pdb;pdb.set_trace()
+##         assert props.get('title') == "A"
+##         got = open(fn, 'r').read()
+##         assert got == d
+        
+        
+def test_suite():
+    suite = unittest.TestSuite()
+    suite.addTest(unittest.makeSuite(Test))
+    return suite
+
+if __name__ == "__main__":
+    unittest.main()
+