#!/usr/bin/env python
# coding: UTF-8

# testDirectoryTree
#
# Copyright(c) 2015 Uptime Technologies, LLC.

import unittest
import os
import DirectoryTree

class TestDirectoryTree(unittest.TestCase):
    def setUp(self):
        os.system("rm -rf t1")
        os.system("mkdir -p t1/t2/t3")
        os.system("touch t1/a t1/t2/b t1/t2/t3/c")

    def testDirectoryTree001(self):
        d = DirectoryTree.DirectoryTree('t1', debug=False)

        l = d.get_file_list()
#        print(l)

        self.assertTrue(d is not None)
        self.assertTrue(len(l) == 1)
        self.assertEqual(l[0], 't1/a')

    def testDirectoryTree002(self):
        d = DirectoryTree.DirectoryTree('t1', recursive=True, debug=False)

        l = d.get_file_list()
#        print(sorted(l))

        self.assertTrue(d is not None)
        self.assertTrue(len(l) == 3)
        self.assertEqual(sorted(l)[0], 't1/a')
        self.assertEqual(sorted(l)[1], 't1/t2/b')
        self.assertEqual(sorted(l)[2], 't1/t2/t3/c')

    def testDirectoryTree003(self):
        d = DirectoryTree.DirectoryTree('t1/a', recursive=False, debug=True)

        l = d.get_file_list()
#        print(sorted(l))

        self.assertTrue(d is not None)
        self.assertTrue(len(l) == 1)
        self.assertEqual(sorted(l)[0], 't1/a')

    def _tearDown(self):
        print("teardown")

if __name__ == '__main__':
    unittest.main()