Added tests for file creation.

db.py

@@ -13,9 +13,6 @@ def __init__(self, db):
         """
         from os import path, stat
 
-        # TODO: Create test to make sure file is created if none exists. Test should verify that an empty {} is placed in the file.
-        # TODO: Test that when a file exists but is empty, a {} is added.
-
         # Create the file if it does not exist or is empty
         if not path.exists(db) or stat(db).st_size == 0:
             new_file = open(db, "w+")

tests.py

@@ -25,6 +25,45 @@ def tearDown(self):
 
 class Tests(BaseTestCase):
 
+    def test_create_none_exists(self):
+        """
+        A file should be created if none exists
+        and an empty {} is placed in the file.
+        """
+        from os import path
+        import os
+
+        database = Database("new.db")
+
+        data = ""
+        with open ("new.db", "r") as db:
+            data = db.read()
+
+        self.assertTrue(path.exists("new.db"))
+        self.assertTrue("{}" in data)
+
+        os.remove("new.db")
+
+    def test_empty_file_initialized(self):
+        """
+        Test that when a file exists but is empty,
+        a {} is added.
+        """
+        import os
+
+        # Create an new empty file
+        open("empty.db", 'a').close()
+
+        database = Database("empty.db")
+
+        data = ""
+        with open ("empty.db", "r") as db:
+            data = db.read()
+
+        self.assertTrue("{}" in data)
+
+        os.remove("empty.db")
+
     def test_get_key(self):
         key = self.database.data(key="url")