test/core/utils.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

#!/usr/bin/env python
#-*- coding:utf-8 -*-

import unittest


class TestUtils(unittest.TestCase):

    def test_boundedDict(self):
        from miasm2.core.utils import BoundedDict

        # Use a callback
        def logger(key):
            print "DELETE", key

        # Create a 5/2 dictionary
        bd = BoundedDict(5, 2, initialdata={"element": "value"},
                         delete_cb=logger)
        bd["element2"] = "value2"
        assert("element" in bd)
        assert("element2" in bd)
        self.assertEqual(bd["element"], "value")
        self.assertEqual(bd["element2"], "value2")

        # Increase 'element2' use
        _ = bd["element2"]

        for i in xrange(6):
            bd[i] = i
            print "Insert %d -> %s" % (i, bd)

        assert(len(bd) == 2)

        assert("element2" in bd)
        self.assertEqual(bd["element2"], "value2")


if __name__ == '__main__':
    testsuite = unittest.TestLoader().loadTestsFromTestCase(TestUtils)
    report = unittest.TextTestRunner(verbosity=2).run(testsuite)
    exit(len(report.errors + report.failures))