store GSL_KEY in app.config

NOAA-GSL · Dec 19, 2024 · b23926f · b23926f
1 parent e97d3f0
commit b23926f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/nwsc_dummy_service/ncd_web_service.py b/python/nwsc_dummy_service/ncd_web_service.py
@@ -53,7 +53,7 @@ def __init__(self, base_dir: str):
     def handler(self):
         """Logic for requests to /all-events"""
         # check that this request has proper key to get or add data
-        if request.headers.get('X-Api-Key') != GSL_KEY:
+        if request.headers.get('X-Api-Key') != current_app.config['GSL_KEY']:
             return jsonify({'message': 'ERROR: Unauthorized'}), 401
 
         if request.method == 'POST':
@@ -99,7 +99,6 @@ def handler(self):
         return jsonify({'profiles': profiles, 'errors': []}), 200
 
 
-
 class AppWrapper:
     """Web server class wrapping Flask operations"""
     def __init__(self, base_dir: str):

diff --git a/python/nwsc_dummy_service/test/test_ncd_web_service.py b/python/nwsc_dummy_service/test/test_ncd_web_service.py
@@ -58,6 +58,7 @@ def mock_current_app(monkeypatch: MonkeyPatch) -> Mock:
     mock_obj = Mock(name='MockCurrentApp', spec=Flask)
     mock_obj.logger.info.return_value = None
     mock_obj.logger.error.return_value = None
+    mock_obj.config = MultiDict({'GSL_KEY': GSL_KEY})
     monkeypatch.setattr('python.nwsc_dummy_service.ncd_web_service.current_app', mock_obj)
     return mock_obj