diff options
| -rw-r--r-- | morphlib/gitindex.py | 4 | ||||
| -rw-r--r-- | morphlib/gitindex_tests.py | 5 |
2 files changed, 9 insertions, 0 deletions
diff --git a/morphlib/gitindex.py b/morphlib/gitindex.py index a119389e..978ea0e2 100644 --- a/morphlib/gitindex.py +++ b/morphlib/gitindex.py @@ -153,3 +153,7 @@ class GitIndex(object): # 50 is an arbitrary limit for paths in morphlib.util.iter_trickle(paths, 50): self._run_git('add', *paths) + + def write_tree(self): + '''Transform the index into a tree in the object store.''' + return self._run_git('write-tree').strip() diff --git a/morphlib/gitindex_tests.py b/morphlib/gitindex_tests.py index 94444091..7a8953f2 100644 --- a/morphlib/gitindex_tests.py +++ b/morphlib/gitindex_tests.py @@ -85,3 +85,8 @@ class GitIndexTests(unittest.TestCase): idx = gd.get_index() self.assertRaises(morphlib.gitdir.NoWorkingTreeError, idx.add_files_from_working_tree, ['foo']) + + def test_write_tree(self): + gd = morphlib.gitdir.GitDirectory(self.dirname) + idx = gd.get_index() + self.assertEqual(idx.write_tree(), gd.resolve_ref_to_tree(gd.HEAD)) |