import tempfile
import unittest
from os.path import join

from int2intmaplike import Int2IntMapLike


class Int2IntMapLikeTest(unittest.TestCase):

    def setUp(self):
        self.test_dir = tempfile.TemporaryDirectory()
        self.tmpfile = join(self.test_dir.name, "my.log")
        self.map = Int2IntMapLike(self.tmpfile)

    def tearDown(self):
        self.map.close()
        self.test_dir.cleanup()

    def test_empty_map(self):
        map = self.map
        self.assertEqual(None, map.find(0))

    def test_one_line_one_byte(self):
        map = self.map
        map.add(10, 1, 1)  # add only the key 10
        self.assertEqual(None, map.find(9))  # directly before
        self.assertEqual(1, map.find(10))
        self.assertEqual(None, map.find(11))  # directly after

    def test_one_line_two_bytes(self):
        map = self.map
        map.add(10, 2, 1)  # added keys 10 and 11
        self.assertEqual(None, map.find(9))  # directly before
        self.assertEqual(1, map.find(10))
        self.assertEqual(1, map.find(11))
        self.assertEqual(None, map.find(12))  # directly after

    def test_two_lines(self):
        map = self.map
        map.add(10, 1, 1)  # added key 10
        map.add(12, 1, 2)  # added key 12
        self.assertEqual(None, map.find(9))  # directly before
        self.assertEqual(1, map.find(10))
        self.assertEqual(None, map.find(11))  # between
        self.assertEqual(2, map.find(12))
        self.assertEqual(None, map.find(13))  # directly after

    def test_fill_map(self):
        map = self.map
        map.blocksize = 64

        # fill map with
        # 10,5,1
        # 20,5,2
        # 30,5,3
        # ...
        for i in range(1, 20):
            map.add(i * 10, 5, i)

        self.assertEqual(2, map.find(20))
        self.assertEqual(7, map.find(71))
        self.assertEqual(13, map.find(134))
        self.assertEqual(19, map.find(194))

        # values that are not in the map
        self.assertEqual(None, map.find(0))
        self.assertEqual(None, map.find(9))
        self.assertEqual(None, map.find(15))
        self.assertEqual(None, map.find(16))
        self.assertEqual(None, map.find(107))  # a value in the second block
        self.assertEqual(None, map.find(188))  # a value in the third block


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