def test_multimaster() -> None:
with use_test_config_dir():
conf1 = {
"address": ("localhost", 12345),
"credentials": ("user1", "password1", "token1"),
"ssl_keys": CERTS1,
}
conf2 = {
"address": ("localhost", 12346),
"credentials": ("user2", "password2", "token2"),
"ssl_keys": CERTS2,
}
with run_api_server(**conf1) as master_url1: # type: ignore
with run_api_server(**conf2) as master_url2: # type: ignore
d1 = Determined(
master_url1,
user="******",
password="******",
cert_path=str(CERTS1["certfile"]),
)
d2 = Determined(
master_url2,
user="******",
password="******",
cert_path=str(CERTS2["certfile"]),
)
d1.get_models()
d2.get_models()
def test_auth_with_store(requests_mock: requests_mock.Mocker,
user: Optional[str]) -> None:
with use_test_config_dir() as config_dir:
auth_json_path = config_dir / "auth.json"
with open(auth_json_path, "w") as f:
json.dump(AUTH_JSON, f)
expected_user = "******" if user == "determined" else "bob"
expected_token = "det.token" if user == "determined" else "bob.token"
requests_mock.get(
"/users/me",
status_code=200,
json={"username": expected_user},
)
authentication = Authentication(MOCK_MASTER_URL, user)
assert authentication.session.username == expected_user
assert authentication.session.token == expected_token
def test_auth_json_v0_upgrade() -> None:
with use_test_config_dir() as config_dir:
auth_json_path = config_dir / "auth.json"
shutil.copy2(AUTH_V0_PATH, auth_json_path)
ts = TokenStore(MOCK_MASTER_URL, auth_json_path)
assert ts.get_active_user() == "determined"
assert ts.get_token("determined") == "v2.public.this.is.a.test"
ts.set_token("determined", "ai")
ts2 = TokenStore(MOCK_MASTER_URL, auth_json_path)
assert ts2.get_token("determined") == "ai"
with auth_json_path.open() as fin:
data = json.load(fin)
assert data.get("version") == 1
assert "masters" in data and list(data["masters"].keys()) == [MOCK_MASTER_URL]
def test_auth_user_from_env(requests_mock: requests_mock.Mocker,
user: Optional[str],
has_token_store: bool) -> None:
with use_test_config_dir() as config_dir, set_container_env_vars():
if has_token_store:
auth_json_path = config_dir / "auth.json"
with open(auth_json_path, "w") as f:
json.dump(AUTH_JSON, f)
requests_mock.get("/users/me",
status_code=200,
json={"username": "******"})
authentication = Authentication(MOCK_MASTER_URL, user)
if has_token_store:
assert authentication.session.username == user or "determined"
assert authentication.session.token == (
"det.token" if user == "determined" else "bob.token")
else:
assert authentication.session.username == "alice"
assert authentication.session.token == "alice.token"
def test_cert_v0_upgrade() -> None:
with use_test_config_dir() as config_dir:
cert_path = config_dir / "master.crt"
shutil.copy2(UNTRUSTED_CERT_PATH, cert_path)
with cert_path.open() as fin:
cert_data = fin.read()
cert = certs_default_load(MOCK_MASTER_URL)
assert isinstance(cert.bundle, str)
with open(cert.bundle) as fin:
loaded_cert_data = fin.read()
assert loaded_cert_data.endswith(cert_data)
assert not cert_path.exists()
v1_certs_path = config_dir / "certs.json"
assert v1_certs_path.exists()
# Load once again from v1.
cert2 = certs_default_load(MOCK_MASTER_URL)
assert isinstance(cert2.bundle, str)
with open(cert2.bundle) as fin:
loaded_cert_data = fin.read()
assert loaded_cert_data.endswith(cert_data)
def test_auth_no_store_no_reauth(user: Optional[str]) -> None:
with use_test_config_dir():
with pytest.raises(UnauthenticatedException):
Authentication(MOCK_MASTER_URL, user)
