SharedDict unittests

keymapper/ipc/shared_dict.py

@@ -91,7 +91,6 @@ class SharedDict:
 
         If it doesn't exist, returns None.
         """
-        # TODO it returns None right?
         return self.__getitem__(key)
 
     def is_alive(self, timeout=None):

tests/testcases/test_ipc.py

@@ -23,8 +23,27 @@ import unittest
 import select
 
 from keymapper.ipc.pipe import Pipe
+from keymapper.ipc.shared_dict import SharedDict
 from keymapper.ipc.socket import Server, Client, Base
 
+from tests.test import quick_cleanup
+
+
+class TestSharedDict(unittest.TestCase):
+    def tearDown(self):
+        quick_cleanup()
+
+    def test_returns_none(self):
+        shared_dict = SharedDict()
+        self.assertIsNone(shared_dict.get("a"))
+        self.assertIsNone(shared_dict["a"])
+
+    def test_set_get(self):
+        shared_dict = SharedDict()
+        shared_dict["a"] = 3
+        self.assertEqual(shared_dict.get("a"), 3)
+        self.assertEqual(shared_dict["a"], 3)
+
 
 class TestSocket(unittest.TestCase):
     def test_socket(self):