update link to upper-constraints.txt

[optf/osdf.git] / test / mainapp / test_osdfapp.py

diff --git a/test/mainapp/test_osdfapp.py b/test/mainapp/test_osdfapp.py
index c162ac0..e4f12e6 100644 (file)
--- a/test/mainapp/test_osdfapp.py
+++ b/test/mainapp/test_osdfapp.py
@@ -15,21 +15,27 @@
 #
 # -------------------------------------------------------------------------
 #
-import osdfapp
+
 import unittest
+from unittest import TestCase
+from unittest import mock
+from unittest.mock import patch
 
-from osdf.operation.exceptions import BusinessException
-from requests import Request, RequestException
+from requests import Request
+from requests import RequestException
 from schematics.exceptions import DataError
-from unittest import mock, TestCase
-from unittest.mock import patch
+
+from osdf.apps import baseapp
+from osdf.apps.baseapp import app
+from osdf.operation.exceptions import BusinessException
 
 
 class TestOSDFApp(TestCase):
 
     def setUp(self):
-        self.patcher_g = patch('osdfapp.g', return_value={'request_id':'DUMMY-REQ'})
-        self.Mock_g = self.patcher_g.start()
+        with app.app_context():
+            self.patcher_g = patch('osdf.apps.baseapp.g', return_value={'request_id': 'DUMMY-REQ'})
+            self.Mock_g = self.patcher_g.start()
         # self.patcher2 = patch('package.module.Class2')
         # self.MockClass2 = self.patcher2.start()
 
@@ -44,31 +50,30 @@ class TestOSDFApp(TestCase):
         e.response.content = "Some request exception occurred"
         # request().raise_for_status.side_effect = e
         return e
- 
+
     def test_handle_business_exception(self):
         e = BusinessException("Business Exception Description")
-        resp = osdfapp.handle_business_exception(e)
+        resp = baseapp.handle_business_exception(e)
         assert resp.status_code == 400
 
     def test_handle_request_exception(self):
         e = self.dummy_request_exception()
-        resp = osdfapp.handle_request_exception(e)
+        resp = baseapp.handle_request_exception(e)
         assert resp.status_code == 400
 
     def test_handle_data_error(self):
         e = DataError({"A1": "A1 Data Error"})
-        resp = osdfapp.handle_data_error(e)
+        resp = baseapp.handle_data_error(e)
         assert resp.status_code == 400
 
     def test_internal_failure(self):
         e = Exception("An Internal Error")
-        resp = osdfapp.internal_failure(e)
+        resp = baseapp.internal_failure(e)
         assert resp.status_code == 500
 
-    def test_getOptions_default(self):
-        opts = osdfapp.getOptions(["PROG"])  # ensure nothing breaks
+    def test_get_options_default(self):
+        baseapp.get_options(["PROG"])  # ensure nothing breaks
 
 
 if __name__ == "__main__":
     unittest.main()
-