Merge pull request #706 from opendatacube/startup_test_coverage

Startup test coverage
opendatacube · Aug 31, 2021 · 2884603 · 2884603
2 parents 19ecc3a + d38ed4f
commit 2884603
Showing 1 changed file with 34 additions and 2 deletions.
diff --git a/tests/test_startup.py b/tests/test_startup.py
@@ -31,6 +31,28 @@ def test_fake_creds(monkeypatch):
         initialise_aws_credentials()
         assert os.getenv("AWS_ACCESS_KEY_ID") == "fake"
 
+
+def test_renewable_creds(monkeypatch):
+    from datacube_ows.startup_utils import (CredentialManager,
+                                            RefreshableCredentials,
+                                            initialise_aws_credentials)
+    CredentialManager._instance = None
+    log = MagicMock()
+    monkeypatch.setenv("AWS_DEFAULT_REGION", "")
+    initialise_aws_credentials(log=log)
+    CredentialManager._instance = None
+    monkeypatch.setenv("AWS_DEFAULT_REGION", "us-west-1")
+    monkeypatch.setenv("AWS_NO_SIGN_REQUEST", "false")
+    monkeypatch.setenv("AWS_REQUEST_PAYER", "requester")
+    with patch("datacube_ows.startup_utils.configure_s3_access") as s3a:
+        mock_creds = MagicMock(spec=RefreshableCredentials)
+        s3a.return_value = mock_creds
+        initialise_aws_credentials(log=log)
+        CredentialManager.check_cred()
+        mock_creds.refresh_needed.return_value = True
+        CredentialManager.check_cred()
+
+
 def test_s3_endpoint_default(monkeypatch):
     from datacube_ows.startup_utils import (CredentialManager,
                                             initialise_aws_credentials)
@@ -53,21 +75,31 @@ def test_initialise_ign_warn():
     initialise_ignorable_warnings()
 
 
-def test_initialise_debugging(monkeypatch):
+def test_initialise_nodebugging(monkeypatch):
     monkeypatch.setenv("PYDEV_DEBUG", "")
     from datacube_ows.startup_utils import initialise_debugging
     initialise_debugging()
 
 
+def test_initialise_debugging(monkeypatch):
+    monkeypatch.setenv("PYDEV_DEBUG", "YES")
+    from datacube_ows.startup_utils import initialise_debugging
+    fake_mod = MagicMock()
+    with patch.dict("sys.modules", pydevd_pycharm=fake_mod) as set_trc:
+        initialise_debugging()
+        fake_mod.settrace.assert_called()
+
+
 def test_initialise_sentry(monkeypatch):
     monkeypatch.setenv("SENTRY_KEY", "")
     monkeypatch.setenv("SENTRY_PROJECT", "")
     from datacube_ows.startup_utils import initialise_sentry
     initialise_sentry()
     monkeypatch.setenv("SENTRY_KEY", "dummy_key")
     monkeypatch.setenv("SENTRY_PROJECT", "dummy_project")
+    log = MagicMock()
     try:
-        initialise_sentry()
+        initialise_sentry(log)
     except Exception:
         pass