"""
author: deadc0de6 (https://github.com/deadc0de6)
Copyright (c) 2017, deadc0de6

Basic unittest for indexing
"""

import os
import unittest

from catcli.catcli import cmd_index
from catcli.noder import Noder
from catcli.catalog import Catalog
from tests.helpers import get_tempdir, create_rnd_file, clean, \
        get_rnd_string, create_dir


class TestIndexing(unittest.TestCase):
    """test index"""

    def test_index(self):
        """test index"""
        # init
        workingdir = get_tempdir()
        catalogpath = create_rnd_file(workingdir, 'catalog.json', content='')
        self.addCleanup(clean, workingdir)

        dirpath = get_tempdir()
        self.addCleanup(clean, dirpath)

        # create 3 files
        file1 = create_rnd_file(dirpath, get_rnd_string(5))
        file2 = create_rnd_file(dirpath, get_rnd_string(5))
        file3 = create_rnd_file(dirpath, get_rnd_string(5))

        # create 2 directories
        dir1 = create_dir(dirpath, get_rnd_string(3))
        dir2 = create_dir(dirpath, get_rnd_string(3))

        # fill directories with files
        _ = create_rnd_file(dir1, get_rnd_string(4))
        _ = create_rnd_file(dir1, get_rnd_string(4))
        _ = create_rnd_file(dir2, get_rnd_string(6))

        noder = Noder()
        top = noder.new_top_node()
        catalog = Catalog(catalogpath, force=True, debug=False)

        # create fake args
        tmpdirname = 'tmpdir'
        args = {'<path>': dirpath, '<name>': tmpdirname,
                '--hash': True, '--meta': ['some meta'],
                '--verbose': True}

        # index the directory
        cmd_index(args, noder, catalog, top)
        self.assertTrue(os.stat(catalogpath).st_size != 0)

        # explore the top node to find all nodes
        self.assertTrue(len(top.children) == 1)
        storage = top.children[0]
        self.assertTrue(len(storage.children) == 5)

        # ensures files and directories are in
        names = [x.get_name() for x in storage.children]
        self.assertTrue(os.path.basename(file1) in names)
        self.assertTrue(os.path.basename(file2) in names)
        self.assertTrue(os.path.basename(file3) in names)
        self.assertTrue(os.path.basename(dir1) in names)
        self.assertTrue(os.path.basename(dir2) in names)

        for node in storage.children:
            if node.get_name() == os.path.basename(dir1):
                self.assertTrue(len(node.children) == 2)
            elif node.get_name() == os.path.basename(dir2):
                self.assertTrue(len(node.children) == 1)


def main():
    """entry point"""
    unittest.main()


if __name__ == '__main__':
    main()