process_tracker_python-102 Add relationship to extract and extract location

🐛 Fixed ProcessTracker tests

Needed table cleanup for new tables in ProcessTracker.

Closes #102

Author: Alex Meadows

tests/test_process_tracker.py

@@ -19,6 +19,7 @@
     ExtractDatasetType,
     ExtractProcess,
     ExtractStatus,
+    ExtractSource,
     Location,
 )
 from process_tracker.models.process import (
@@ -43,9 +44,11 @@
     Source,
     SourceContact,
     SourceDatasetType,
+    SourceLocation,
     SourceObject,
     SourceObjectAttribute,
     SourceObjectDatasetType,
+    SourceObjectLocation,
 )
 
 from process_tracker.utilities.data_store import DataStore, ClusterProcess
@@ -117,12 +120,16 @@ def tearDown(self):
         :return:
         """
         self.session.query(ExtractProcess).delete()
+        self.session.query(ExtractSource).delete()
         self.session.query(ExtractDatasetType).delete()
         self.session.query(SourceDatasetType).delete()
         self.session.query(SourceObjectDatasetType).delete()
+        self.session.query(SourceLocation).delete()
+        self.session.query(SourceObjectLocation).delete()
         self.session.query(ProcessDatasetType).delete()
         self.session.query(ErrorTracking).delete()
         self.session.query(ProcessTracking).delete()
+        self.session.query(Process)
         self.session.query(Extract).delete()
         self.session.query(ErrorType).delete()
         self.session.commit()