1 files changed, 83 insertions, 0 deletions
diff --git a/test/git/test_tree.py b/test/git/test_tree.py
new file mode 100644
index 00000000..3812f213
--- /dev/null
+++ b/test/git/test_tree.py
@@ -0,0 +1,83 @@
+from mock import *
+from gitalicious.test.asserts import *
+from gitalicious.lib import *
+from gitalicious.test.helper import *
+
+class TestTree(object):
+    def setup(self):
+        self.repo = Repo(GIT_REPO)
+        self.tree = Tree(self.repo)
+
+    @patch(Git, 'method_missing')
+    def test_contents_should_cache(self, git):
+        git.return_value = fixture('ls_tree_a') + fixture('ls_tree_b')
+    
+        tree = self.repo.tree('master')
+
+        child = tree.contents[-1]
+        child.contents
+        child.contents
+        
+        assert_true(git.called)
+        assert_equal(2, git.call_count)
+        assert_equal(git.call_args, (('ls_tree', '34868e6e7384cb5ee51c543a8187fdff2675b5a7'), {}))
+  
+    def test_content_from_string_tree_should_return_tree(self):
+        text = fixture('ls_tree_a').splitlines()[-1]
+        tree = self.tree.content_from_string(None, text)
+
+        assert_equal(Tree, tree.__class__)
+        assert_equal("650fa3f0c17f1edb4ae53d8dcca4ac59d86e6c44", tree.id)
+        assert_equal("040000", tree.mode)
+        assert_equal("test", tree.name)
+  
+    def test_content_from_string_tree_should_return_blob(self):
+        text = fixture('ls_tree_b').split("\n")[0]
+        
+        tree = self.tree.content_from_string(None, text)
+
+        assert_equal(Blob, tree.__class__)
+        assert_equal("aa94e396335d2957ca92606f909e53e7beaf3fbb", tree.id)
+        assert_equal("100644", tree.mode)
+        assert_equal("grit.rb", tree.name)
+  
+    def test_content_from_string_tree_should_return_commit(self):
+        text = fixture('ls_tree_commit').split("\n")[1]
+    
+        tree = self.tree.content_from_string(None, text)
+        assert_none(tree)
+    
+    @raises(TypeError)
+    def test_content_from_string_invalid_type_should_raise(self):
+        self.tree.content_from_string(None, "040000 bogus 650fa3f0c17f1edb4ae53d8dcca4ac59d86e6c44	test")
+
+    @patch(Blob, '__len__')
+    @patch(Git, 'method_missing')
+    def test_slash(self, blob, git):
+        git.return_value = fixture('ls_tree_a')
+        blob.return_value = 1
+        
+        tree = self.repo.tree('master')
+        
+        assert_equal('aa06ba24b4e3f463b3c4a85469d0fb9e5b421cf8', (tree/'lib').id)
+        assert_equal('8b1e02c0fb554eed2ce2ef737a68bb369d7527df', (tree/'README.txt').id)
+        
+        assert_true(git.called)
+        assert_equal(git.call_args, (('ls_tree', 'master'), {}))
+  
+    @patch(Git, 'method_missing')
+    def test_slash_with_commits(self, git):
+        git.return_value = fixture('ls_tree_commit')
+
+        tree = self.repo.tree('master')
+    
+        assert_none(tree/'bar')
+        assert_equal('2afb47bcedf21663580d5e6d2f406f08f3f65f19', (tree/'foo').id)
+        assert_equal('f623ee576a09ca491c4a27e48c0dfe04be5f4a2e', (tree/'baz').id)
+
+        assert_true(git.called)
+        assert_equal(git.call_args, (('ls_tree', 'master'), {}))
+
+    def test_repr(self):
+        self.tree = Tree(self.repo, **{'id': 'abc'})
+        assert_equal('<GitPython.Tree "abc">', repr(self.tree))