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_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()