import unittest

from httpie.prompt.tree import Node


class TestNode(unittest.TestCase):

    def setUp(self):
        # Make a tree like this:
        #          root
        #     a             h
        #  b     d        i   n
        # c f   e g     k     o
        #             l m p
        self.root = Node('root')
        self.root.add_path('a', 'b', 'c')
        self.root.add_path('a', 'b', 'f')
        self.root.add_path('a', 'd', 'e')
        self.root.add_path('a', 'd', 'g')
        self.root.add_path('h', 'i', 'k', 'l')
        self.root.add_path('h', 'i', 'k', 'm')
        self.root.add_path('h', 'i', 'k', 'p')
        self.root.add_path('h', 'n', 'o')

    def test_illegal_name(self):
        self.assertRaises(ValueError, Node, '.')
        self.assertRaises(ValueError, Node, '..')

    def test_str(self):
        node = Node('my node')
        self.assertEqual(str(node), 'my node')

    def test_cmp_same_type(self):
        a = Node('a', data={'type': 'dir'})
        b = Node('b', data={'type': 'dir'})
        self.assertTrue(a < b)

    def test_cmp_different_type(self):
        a = Node('a', data={'type': 'file'})
        b = Node('b', data={'type': 'dir'})
        self.assertTrue(b < a)

    def test_eq(self):
        a = Node('a', data={'type': 'file'})
        b = Node('b', data={'type': 'dir'})
        self.assertNotEqual(a, b)

        a = Node('a', data={'type': 'file'})
        b = Node('a', data={'type': 'file'})
        self.assertEqual(a, b)

    def test_add_path_and_find_child(self):
        # Level 1 (root)
        self.assertEqual(set(c.name for c in self.root.children), set('ah'))

        # Level 2
        node_a = self.root.find_child('a')
        node_h = self.root.find_child('h')
        self.assertEqual(set(c.name for c in node_a.children), set('bd'))
        self.assertEqual(set(c.name for c in node_h.children), set('in'))

        # Level 3
        node_b = node_a.find_child('b')
        node_i = node_h.find_child('i')
        self.assertEqual(set(c.name for c in node_b.children), set('cf'))
        self.assertEqual(set(c.name for c in node_i.children), set('k'))

        # Level 4
        node_c = node_b.find_child('c')
        node_k = node_i.find_child('k')
        self.assertEqual(set(c.name for c in node_c.children), set())
        self.assertEqual(set(c.name for c in node_k.children), set('lmp'))

        # Return None if child can't be found
        self.assertFalse(node_c.find_child('x'))

    def test_find_child_wildcard(self):
        root = Node('root')
        root.add_path('a')
        root.add_path('{b}')
        root.add_path('c')

        self.assertEqual(root.find_child('a').name, 'a')
        self.assertEqual(root.find_child('c').name, 'c')
        self.assertEqual(root.find_child('x').name, '{b}')
        self.assertFalse(root.find_child('x', wildcard=False))

    def test_ls(self):
        self.assertEqual([n.name for n in self.root.ls('a')], list('bd'))
        self.assertEqual([n.name for n in self.root.ls('a', 'b')], list('cf'))
        self.assertEqual([n.name for n in self.root.ls('a', 'b', 'c')], [])
        self.assertEqual([n.name for n in self.root.ls('h', 'i', 'k')],
                         list('lmp'))

    def test_ls_root(self):
        self.assertEqual([n.name for n in self.root.ls()], list('ah'))

    def test_ls_non_existing(self):
        self.assertEqual([n.name for n in self.root.ls('x')], [])
        self.assertEqual([n.name for n in self.root.ls('a', 'b', 'x')], [])

    def test_ls_parent(self):
        self.assertEqual([n.name for n in self.root.ls('..')], list('ah'))
        self.assertEqual([n.name for n in self.root.ls('..', '..', '..')],
                         list('ah'))
        self.assertEqual([n.name for n in self.root.ls('..', '..', 'h')],
                         list('in'))
        self.assertEqual(
            [n.name for n in self.root.ls('..', '..', 'h', '..', 'a')],
            list('bd'))

    def test_ls_dot(self):
        self.assertEqual([n.name for n in self.root.ls('.')], list('ah'))
        self.assertEqual([n.name for n in self.root.ls('.', '.', '.')],
                         list('ah'))
        self.assertEqual([n.name for n in self.root.ls('.', 'a', 'b')],
                         list('cf'))
        self.assertEqual([n.name for n in self.root.ls('.', 'h', '.')],
                         list('in'))
        self.assertEqual(
            [n.name for n in self.root.ls('.', 'h', '.', '.', 'n')], ['o'])

    def test_ls_sort_by_types(self):
        self.root.add_path('q', 'r')
        self.root.add_path('q', 's', node_type='file')
        self.root.add_path('q', 't', node_type='file')
        self.root.add_path('q', 'u')
        self.root.add_path('q', 'v', node_type='file')

        self.assertEqual([n.name for n in self.root.ls('q')],
                         list('rustv'))