TeachingData
diff --git a/‎ETL/FileParsers/fileparser.py‎
Lines changed: 1 addition & 1 deletion b/‎ETL/FileParsers/fileparser.py‎
Lines changed: 1 addition & 1 deletion
diff --git a/‎ETL/FileParsers/jsonparser.py‎
Lines changed: 5 additions & 1 deletion b/‎ETL/FileParsers/jsonparser.py‎
Lines changed: 5 additions & 1 deletion
diff --git a/‎ETL/test_loadJSON.py‎
Lines changed: 13 additions & 5 deletions b/‎ETL/test_loadJSON.py‎
Lines changed: 13 additions & 5 deletions
@@ -17,7 +17,7 @@ def file(self, file: str):
 
     @property
     def directory(self):
-        return self.__data_dir.toString()
+        return self.__data_dir
 
     @directory.setter
     def directory(self, directory: str):
 
@@ -6,10 +6,14 @@ class JSONParser(FileParser):
 
     def parse_file(self):
         with open(self.file, "r") as jd:
+            # Add call to validator here
             self.responses = json.loads(jd.read())
 
+    def get_responses(self):
+        return self.responses
+
     def print_json(self):
         for i in self.responses:
-            print(f"\nWe are on {i} iteration\n")
+            print("\n")
             for k in i.keys():
                 print(f"The key is {k} with value(s): {i[k]}")
@@ -1,8 +1,16 @@
-import pytest
+import unittest
 from ETL.FileParsers.jsonparser import JSONParser
 
-# This will be replaced by specific fileParser unit tests (as it will be the abstract)
-def test_check_file():
+# TODO: Add mock of json.loads & open
+
+class TestJSONParser(unittest.TestCase):
     jp = JSONParser()
-    jp.file = "student_poll.json"
-    assert jp.check_file() is True
+
+    def test_check_file(self):
+        self.jp.file = "student_poll.json"
+        assert self.jp.check_file() is True
+
+    def test_get_responses(self):
+        test_dict = {'grade': 'freshman', 'TA': False}
+        self.jp.responses = [{'grade': 'freshman', 'TA': False}, {'grade': 'sophomore', 'TA': False}]
+        self.assertDictEqual(self.jp.get_responses()[0], test_dict)