Up test coverage + fix exceptions

Author: Code0x58

jsonstore/__init__.py

@@ -136,7 +136,7 @@ def __get_obj(self, full_path):
             path.append(step)
             try:
                 obj = obj[step]
-            except AttributeError:
+            except KeyError:
                 raise KeyError('.'.join(path))
         return obj
 
@@ -151,7 +151,7 @@ def __setitem__(self, name, value):
 
     def __getitem__(self, key):
         obj = self.__get_obj(key)
-        if obj is self:
+        if obj is self._data:
             raise KeyError
         return deepcopy(obj)
 

jsonstore/tests.py

@@ -109,6 +109,16 @@ def test_has_values(self):
 
         self.assertFalse('foo' in self.store)
 
+    def test_empty_key(self):
+        with self.assertRaises(KeyError):
+            self.store['']
+
+    def test_empty_store(self):
+        store_file = mktemp()
+        with open(store_file, 'wb') as f:
+            f.write(b"")
+        JsonStore(f.name)
+
     def test_assign_cycle(self):
         test_list = []
         test_dict = {}