1 files changed, 93 insertions, 0 deletions
diff --git a/test/git/test_base.py b/test/git/test_base.py
new file mode 100644
index 00000000..402cdba3
--- /dev/null
+++ b/test/git/test_base.py
@@ -0,0 +1,93 @@
+# test_base.py
+# Copyright (C) 2008, 2009 Michael Trier (mtrier@gmail.com) and contributors
+#
+# This module is part of GitPython and is released under
+# the BSD License: http://www.opensource.org/licenses/bsd-license.php
+
+import time
+from test.testlib import *
+from git import *
+import git.objects.base as base
+import git.refs as refs
+from itertools import chain
+from git.objects.utils import get_object_type_by_name
+
+class TestBase(object):
+	
+	type_tuples = (  ("blob", "8741fc1d09d61f02ffd8cded15ff603eff1ec070"), 
+					 ("tree", "3a6a5e3eeed3723c09f1ef0399f81ed6b8d82e79"),
+					 ("commit", "4251bd59fb8e11e40c40548cba38180a9536118c"),
+					 ("tag", "e56a60e8e9cd333cfba0140a77cd12b0d9398f10") ) 
+	
+	def setup(self):
+		self.repo = Repo(GIT_REPO)
+		
+	def test_base_object(self):	
+		# test interface of base object classes
+		types = (Blob, Tree, Commit, TagObject)
+		assert len(types) == len(self.type_tuples)
+		
+		s = set()
+		num_objs = 0
+		num_index_objs = 0
+		for obj_type, (typename, hexsha) in zip(types, self.type_tuples):
+			item = obj_type(self.repo,hexsha)
+			num_objs += 1
+			assert item.id == hexsha
+			assert item.type == typename
+			assert item.size
+			assert item.data
+			assert item == item
+			assert not item != item
+			assert str(item) == item.id
+			assert repr(item)
+			s.add(item)
+			
+			if isinstance(item, base.IndexObject):
+				num_index_objs += 1
+				if hasattr(item,'path'):						# never runs here
+					assert not item.path.startswith("/")		# must be relative
+					assert isinstance(item.mode, int)
+			# END index object check
+		# END for each object type to create
+		
+		# each has a unique sha
+		assert len(s) == num_objs
+		assert len(s|s) == num_objs
+		assert num_index_objs == 2
+		
+		
+	def test_tags(self):
+		# tag refs can point to tag objects or to commits
+		s = set()
+		ref_count = 0
+		for ref in chain(self.repo.tags, self.repo.heads):
+			ref_count += 1
+			assert isinstance(ref, refs.Ref)
+			assert str(ref) == ref.name
+			assert repr(ref)
+			assert ref == ref
+			assert not ref != ref
+			s.add(ref)
+		# END for each ref
+		assert len(s) == ref_count
+		assert len(s|s) == ref_count
+		
+	def test_heads(self):
+		# see how it dynmically updates its object
+		for head in self.repo.heads:
+			head.name
+			head.path
+			prev_object = head.object
+			cur_object = head.object
+			assert prev_object == cur_object		# represent the same git object
+			assert prev_object is not cur_object	# but are different instances
+		# END for each head
+		
+	def test_get_object_type_by_name(self):
+		for tname in base.Object.TYPES:
+			assert base.Object in get_object_type_by_name(tname).mro()
+		# END for each known type 
+		
+		assert_raises( ValueError, get_object_type_by_name, "doesntexist" )
+