Getting Dimensions tests to work

sul-dlss-labs · Jun 19, 2024 · f7a8e74 · f7a8e74
1 parent 3376e2b
commit f7a8e74
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/rialto_airflow/harvest/dimensions.py b/rialto_airflow/harvest/dimensions.py
@@ -11,8 +11,8 @@
 dotenv.load_dotenv()
 
 dimcli.login(
-    os.environ.get("DIMENSIONS_API_USER"),
-    os.environ.get("DIMENSIONS_API_PASS"),
+    os.environ.get("AIRFLOW_VAR_DIMENSIONS_API_USER"),
+    os.environ.get("AIRFLOW_VAR_DIMENSIONS_API_PASS"),
     "https://app.dimensions.ai",
 )
 

diff --git a/test/harvest/test_dimensions.py b/test/harvest/test_dimensions.py
@@ -7,18 +7,16 @@
 
 dotenv.load_dotenv()
 
-dimensions_user = os.environ.get("DIMENSIONS_API_USER")
-dimensions_password = os.environ.get("DIMENSIONS_API_PASS")
+dimensions_user = os.environ.get("AIRFLOW_VAR_DIMENSIONS_API_USER")
+dimensions_password = os.environ.get("AIRFLOW_VAR_DIMENSIONS_API_PASS")
 
 no_auth = not (dimensions_user and dimensions_password)
 
 
 @pytest.mark.skipif(no_auth, reason="no dimensions key")
 def test_dimensions_doi_orcids_dict(tmpdir):
     pickle_file = tmpdir / "dimensions.pickle"
-    dimensions_doi_orcids_dict(
-        "data/rialto_app/authors_2024-03-18.csv", pickle_file, limit=5
-    )
+    dimensions_doi_orcids_dict("test/data/orcids.csv", pickle_file, limit=5)
     assert pickle_file.isfile()
 
     with open(pickle_file, "rb") as handle: