import unittest

from sources.hash_chains import QueryProcessor, Query

class MyTestCase(unittest.TestCase):
    def test_something(self):
        proc = QueryProcessor(5)
        self.assertEqual(4, proc.hash_func('world'))
        self.assertEqual(4, proc.hash_func('HellO'))
        self.assertEqual(2, proc.hash_func('GooD'))
        self.assertEqual(2, proc.hash_func('luck'))

    def test_something1(self):
        proc = QueryProcessor(3)
        self.assertEqual(1, proc.hash_func('add'))
        self.assertEqual(1, proc.hash_func('help'))
        self.assertEqual(2, proc.hash_func('del'))

        proc.elems = [[] for i in range(12)]
        proc.process_query(Query(['check', 0]))
        proc.process_query(Query(['find', 'help']))
        proc.process_query(Query(['add', 'help']))
        proc.process_query(Query(['add', 'del']))
        proc.process_query(Query(['add', 'add']))
        proc.process_query(Query(['find', 'add']))
        proc.process_query(Query(['find', 'del']))
        proc.process_query(Query(['del', 'del']))
        proc.process_query(Query(['find', 'del']))
        proc.process_query(Query(['check', 0]))
        proc.process_query(Query(['check', 1]))
        proc.process_query(Query(['check', 2]))


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