merge

arkindex_worker/worker/base.py

@@ -135,9 +135,8 @@ class BaseWorker(object):
     def configure_for_developers(self):
         # CLI args are stored on the instance so that implementations can access them
         self.args = self.parser.parse_args()
-
-        # Setup logging level if verbose or if ARKINDEX_DEBUG is set to true
-        if self.args.verbose or os.environ.get("ARKINDEX_DEBUG"):
+        # Setup logging level
+        if self.args.verbose:
             logger.setLevel(logging.DEBUG)
             logger.debug("Debug output enabled")
         if self.args.config:
@@ -167,14 +166,18 @@ class BaseWorker(object):
         self.api_client = ArkindexClient(**options_from_env())
         logger.debug(f"Setup Arkindex API client on {self.api_client.document.url}")
 
-        # Load process information except in developer mode
-        if not self.args.dev:
-            assert os.environ.get(
-                "ARKINDEX_PROCESS_ID"
-            ), "ARKINDEX_PROCESS_ID environment variable is not defined"
-            self.process_information = self.request(
-                "RetrieveDataImport", id=os.environ["ARKINDEX_PROCESS_ID"]
-            )
+        # Load features available on backend, and check authentication
+        user = self.request("RetrieveUser")
+        logger.debug(f"Connected as {user['display_name']} - {user['email']}")
+        self.features = user["features"]
+
+        # Load process information
+        assert os.environ.get(
+            "ARKINDEX_PROCESS_ID"
+        ), "ARKINDEX_PROCESS_ID environment variable is not defined"
+        self.process_information = self.request(
+            "RetrieveDataImport", id=os.environ["ARKINDEX_PROCESS_ID"]
+        )
 
         if self.worker_version_id:
             # Retrieve initial configuration from API

tests/test_elements_worker/test_worker.py

@@ -99,7 +99,7 @@ def test_activities_disabled(responses, monkeypatch, mock_worker_version_api):
     ] == BASE_API_CALLS
 
 
-def test_activities_dev_mode(mocker, mock_worker_version_api):
+def test_activities_dev_mode(mocker):
     """
     Worker activities are not stored in dev mode
     """