1 files changed, 146 insertions, 0 deletions
diff --git a/test/test-gall.tree.lua b/test/test-gall.tree.lua
new file mode 100644
index 0000000..e1d5449
--- /dev/null
+++ b/test/test-gall.tree.lua
@@ -0,0 +1,146 @@
+-- test/test-gall.tree.lua
+--
+-- Git Abstraction layer for Lua - Tree object tests
+--
+-- Copyright 2012 Daniel Silverstone <dsilvers@digital-scurf.org>
+--
+-- For Licence terms, see COPYING
+--
+
+-- This test includes tree wibbling
+dofile "test/withrepo.lua"
+
+-- Step one, start coverage
+
+local luacov = require 'luacov'
+
+local gall = require 'gall'
+
+local testnames = {}
+
+local real_assert = assert
+local total_asserts = 0
+local function assert(...)
+   local retval = real_assert(...)
+   total_asserts = total_asserts + 1
+   return retval
+end
+
+local function add_test(suite, name, value)
+   rawset(suite, name, value)
+   testnames[#testnames+1] = name
+end
+
+local suite = setmetatable({}, {__newindex = add_test})
+
+function suite.get_tree_ok()
+   local repo = test_repo()
+   local commit = repo:get("HEAD").content
+   local tree = commit.tree.content
+   assert(tree)
+end
+
+function suite.tree_str()
+   local repo = test_repo()
+   local commit = repo:get("HEAD").content
+   local treeobj = commit.tree
+   local tree = treeobj.content
+   local treestr = tostring(tree)
+   assert(treestr:find("GitTree"))
+   assert(treestr:find(treeobj.sha))
+end
+
+function suite.flatten_tree()
+   local repo = test_repo()
+   local commit = repo:get("refs/heads/heirarchy").content
+   local tree = commit.tree.content
+   local flat = gall.tree.flatten(tree)
+   assert(flat)
+   assert(flat.testblob)
+   assert(flat["blobs/second"])
+end
+
+function suite.bad_diff()
+   local repo = test_repo()
+   local commit = repo:get("refs/heads/heirarchy").content
+   local tree = commit.tree.content
+   local ok, msg = tree:diff_to()
+   assert(not ok)
+   assert(msg:find("%?%?%?"))
+end
+
+function suite.diff_trees()
+   local repo = test_repo()
+   local heir_tree = repo:get("refs/heads/heirarchy").content.tree.content
+   local master_tree = repo:get("refs/heads/master").content.tree.content
+   local diff = assert(master_tree:diff_to(heir_tree))
+end
+
+function suite.flatten_then_create_tree()
+   local repo = test_repo()
+   local commit = repo:get("refs/heads/heirarchy").content
+   local tree = commit.tree.content
+   local flat = gall.tree.flatten(tree)
+   assert(flat)
+   assert(flat.testblob)
+   assert(flat["blobs/second"])
+   local newtree = gall.tree.create(repo, flat)
+   assert(newtree == commit.tree)
+end
+
+function suite.create_bad_tree_type()
+   local repo = test_repo()
+   local bad_tree = {
+      thingy = { obj = repo:get("HEAD") }
+   }
+   local ok, msg = gall.tree.create(repo, bad_tree)
+   assert(not ok)
+   assert(msg:find("commit"))
+end
+
+function suite.create_bad_tree_type_flat()
+   local repo = test_repo()
+   local bad_tree = {
+      thingy = { type = "FISH" }
+   }
+   local ok, msg = gall.tree.create(repo, bad_tree)
+   assert(not ok)
+   assert(msg:find("FISH"))
+end
+
+function suite.create_bad_tree_type_flat_deep()
+   local repo = test_repo()
+   local bad_tree = {
+      ["thingy/wotsit"] = { type = "FISH" }
+   }
+   local ok, msg = gall.tree.create(repo, bad_tree)
+   assert(not ok)
+   assert(msg:find("FISH"))
+end
+
+function suite.create_bad_sha()
+   local repo = test_repo()
+   local bad_tree = {
+      ["thingy/wotsit"] = { type = "blob", sha = "FISHFISH" }
+   }
+   local ok, msg = gall.tree.create(repo, bad_tree)
+   assert(not ok)
+   assert(msg:find("mktree returned"))
+end
+
+
+local count_ok = 0
+for _, testname in ipairs(testnames) do
+--   print("Run: " .. testname)
+   local ok, err = xpcall(suite[testname], debug.traceback)
+   if not ok then
+      print(err)
+      print()
+   else
+      count_ok = count_ok + 1
+   end
+end
+
+print(tostring(count_ok) .. "/" .. tostring(#testnames) .. " [" .. tostring(total_asserts) .. "] OK")
+
+os.exit(count_ok == #testnames and 0 or 1)