mirror of
https://github.com/cea-sec/miasm
synced 2026-06-21 13:48:18 +00:00
Camille Mougey
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
#!/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 dictionnary
|
|
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))
|