basic test for the list of attributes returned in an item

ECN · Jan 11, 2021 · 6dcad0d · 6dcad0d
1 parent 46807e9
commit 6dcad0d
Showing 1 changed file with 8 additions and 9 deletions.
diff --git a/api/pytest/test_api_tester.py b/api/pytest/test_api_tester.py
@@ -87,16 +87,15 @@ def test_ECNQueue_Item():
         "department",
         "building",
         "dateReceived",
-        "jsonData"
+        "jsonData",
+        "toJson"
     ]
 
-    # Returns a list of all attributes in the item object
-    returnedItemKeys = item.__dict__
+    # Returns a list of all attributes in the item object, omiting attribute names starting with "_" 
+    returnedItemKeys = []
+    for attribute in item.__dir__():
+        if not attribute.startswith("_"):
+            returnedItemKeys.append(attribute)
 
-    # Checks that all necessary keys are in the item
+    # Checks that all necessary keys are in the item by comparing it to the itemkeys list
     assert all(keys in itemKeys for keys in returnedItemKeys)
-
-    print()
-
-if __name__ == "__main__":
-    test_ECNQueue_Item()