From e2f33c99b059f315c813379c21fc947ed7d5ed50 Mon Sep 17 00:00:00 2001
From: deadc0de6 <info@deadc0de.ch>
Date: Thu, 14 Dec 2017 20:09:20 +0100
Subject: [PATCH] add test for the tree command

---
 tests/test_tree.py | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 tests/test_tree.py

diff --git a/tests/test_tree.py b/tests/test_tree.py
new file mode 100644
index 0000000..8adefff
--- /dev/null
+++ b/tests/test_tree.py
@@ -0,0 +1,39 @@
+"""
+author: deadc0de6 (https://github.com/deadc0de6)
+Copyright (c) 2017, deadc0de6
+
+Basic unittest for tree
+"""
+
+import unittest
+
+from catcli.catcli import *
+from catcli.noder import Noder
+from catcli.walker import Walker
+from catcli.catalog import Catalog
+from tests.helpers import *
+
+
+class TestTree(unittest.TestCase):
+
+    def test_tree(self):
+        # init
+        path = 'fake'
+        self.addCleanup(clean, path)
+        catalog = Catalog(path, force=True, verbose=False)
+        top = catalog._restore_json(get_fakecatalog())
+        noder = Noder()
+
+        # create fake args dict
+        args = {'<path>': path, '--verbose': True}
+
+        # print tree and wait for any errors
+        cmd_tree(args, noder, top)
+
+
+def main():
+    unittest.main()
+
+
+if __name__ == '__main__':
+    main()