Progress on moving setup into pytest setup

Author: nealrichardson

pins/tests/conftest.py

@@ -1,3 +1,4 @@
+import json
 import os
 import tempfile
 from pathlib import Path
@@ -12,6 +13,7 @@
     RscBoardBuilder,
     Snapshot,
     rm_env,
+    set_rsc_api_keys,
 )
 
 EXAMPLE_REL_PATH = "pins/tests/pins-compat"
@@ -86,5 +88,38 @@ def tmp_data_dir():
             yield Path(tmp_dir)
 
 
+@pytest.fixture(scope="session", autouse=True)
+def rsc_api_keys(request):
+    if not any(item.get_closest_marker("fs_rsc") for item in request.session.items):
+        return None
+
+    from pins.rsconnect.api import LoginConnectApi, RsConnectApi
+
+    extra_users = [
+        {"username": "susan", "password": "susansusan"},
+        {"username": "derek", "password": "derekderek"},
+    ]
+
+    admin_client = RsConnectApi()
+
+    guid = admin_client.get_user()["guid"]
+    admin_client.query_v1(f"users/{guid}", "PUT", json={"username": "admin"})
+
+    api_keys = {"admin": os.getenv("CONNECT_API_KEY")}
+
+    for user in extra_users:
+        admin_client.create_user(
+            username=user["username"],
+            password=user["password"],
+            __confirmed=True,
+        )
+        api_keys[user["username"]] = LoginConnectApi(
+            user["username"], user["password"]
+        ).create_api_key()
+
+    set_rsc_api_keys(api_keys)
+    return api_keys
+
+
 def pytest_addoption(parser):
     parser.addoption("--snapshot-update", action="store_true")

pins/tests/helpers.py

@@ -19,8 +19,7 @@
 DEFAULT_CREATION_DATE = datetime(2020, 1, 13, 23, 58, 59)
 
 RSC_SERVER_URL = os.getenv("CONNECT_SERVER")
-# TODO: should use pkg_resources for this path?
-RSC_KEYS_FNAME = "pins/tests/rsconnect_api_keys.json"
+_RSC_API_KEYS = None
 
 DATABRICKS_VOLUME = "/Volumes/workshops/my-board/my-volume/test"
 
@@ -76,12 +75,18 @@ def wrapper(*args, **kwargs):
     return wrapper
 
 
+def set_rsc_api_keys(keys):
+    global _RSC_API_KEYS
+    _RSC_API_KEYS = keys
+
+
 def rsc_from_key(name):
     from pins.rsconnect.api import RsConnectApi
 
-    with open(RSC_KEYS_FNAME) as f:
-        api_key = json.load(f)[name]
-        return RsConnectApi(RSC_SERVER_URL, api_key)
+    if _RSC_API_KEYS is None:
+        raise RuntimeError("RSC API keys not initialized. This should be set by the pytest fixture.")
+    api_key = _RSC_API_KEYS[name]
+    return RsConnectApi(RSC_SERVER_URL, api_key)
 
 
 def rsc_fs_from_key(name):