Add tests for the short OID support

test/test_refs.py

@@ -84,6 +84,12 @@ class ReferencesTest(utils.RepoTestCase):
         reference.oid = NEW_COMMIT
         self.assertEqual(reference.hex, NEW_COMMIT)
 
+    def test_reference_set_sha_prefix(self):
+        NEW_COMMIT = '5ebeeebb320790caf276b9fc8b24546d63316533'
+        reference = self.repo.lookup_reference('refs/heads/master')
+        reference.oid = NEW_COMMIT[0:6]
+        self.assertEqual(reference.hex, NEW_COMMIT)
+
 
     def test_reference_get_type(self):
         reference = self.repo.lookup_reference('refs/heads/master')

test/test_repository.py

@@ -76,7 +76,9 @@ class RepositoryTest(utils.BareRepoTestCase):
         self.assertRaises(TypeError, lambda: 123 in self.repo)
         self.assertTrue(A_BIN_SHA in self.repo)
         self.assertTrue(A_HEX_SHA in self.repo)
+        self.assertTrue(A_HEX_SHA[0:10] in self.repo)
         self.assertFalse('a' * 40 in self.repo)
+        self.assertFalse('a' * 20 in self.repo)
 
     def test_lookup_blob(self):
         self.assertRaises(TypeError, lambda: self.repo[123])

test/test_revwalk.py

@@ -64,6 +64,11 @@ class WalkerTest(utils.RepoTestCase):
         walker.hide('4ec4389a8068641da2d6578db0419484972284c8')
         self.assertEqual(len(list(walker)), 2)
 
+    def test_hide_prefix(self):
+        walker = self.repo.walk(log[0], GIT_SORT_TIME)
+        walker.hide('4ec4389a')
+        self.assertEqual(len(list(walker)), 2)
+
     def test_reset(self):
         walker = self.repo.walk(log[0], GIT_SORT_TIME)
         walker.reset()