TASK: Add test for Dictwrapper

pastepwn/util/dictwrapper_test.py

@@ -0,0 +1,23 @@
+# -*- coding: utf-8 -*-
+import unittest
+from unittest import mock
+
+from pastepwn.util.dictwrapper import DictWrapper
+
+
+class TestDictwrapper(unittest.TestCase):
+    def setUp(self):
+        self.test_dict = {"key1": "value1", "key2": "value2"}
+        self.obj = mock.Mock()
+
+    def test_existing_key(self):
+        test_wrapped = DictWrapper(self.test_dict)
+        self.assertEqual(test_wrapped["key1"], "value1")
+
+    def test_non_existing_key(self):
+        test_wrapped = DictWrapper(self.test_dict)
+        self.assertEqual(test_wrapped["key3"], "{key3}")
+
+
+if __name__ == '__main__':
+    unittest.main()