added unit test for cleanup_db

modified: adsdocmatch/tests/unittests/test_oracle_util.py
adsabs · Aug 7, 2023 · 825e9a5 · 825e9a5
1 parent 0b1c1e3
commit 825e9a5
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/adsdocmatch/tests/unittests/test_oracle_util.py b/adsdocmatch/tests/unittests/test_oracle_util.py
@@ -570,6 +570,17 @@ def test_query(self):
         # remove temp files
         os.remove(tmp_output_filename)
 
+    def test_cleanup(self):
+        """ """
+        with mock.patch('requests.get') as mock_oracle_util:
+            mock_oracle_util.return_value = mock_response = mock.Mock()
+            mock_response.status_code = 200
+            result = self.match_metadata.ORACLE_UTIL.cleanup_db()
+            self.assertEqual(result, None)
+            mock_response.status_code = 403
+            result = self.match_metadata.ORACLE_UTIL.cleanup_db()
+            self.assertEqual(result, None)
+
 
 if __name__ == '__main__':
     unittest.main()