def test_network_failure(
head: mock.MagicMock,
run: mock.MagicMock,
):
head.side_effect = requests.ConnectionError()
with pytest.raises(requests.ConnectionError):
login.Auth()._run_server()
run.assert_not_called()
head.side_effect = requests.RequestException()
with pytest.raises(requests.RequestException):
login.Auth()._run_server()
run.assert_not_called()
def test_credentials_file_io(self):
auth = login.Auth()
assert not auth.secrets_file.exists()
assert auth.load() is False
auth.save(username="", user_id="123", api_key="123")
assert auth.secrets_file.exists()
assert auth.load() is True
def test_can_store_credentials(self):
auth = login.Auth()
auth.save(username="******", user_id="kr-1234")
assert auth.secrets_file.exists()
auth.clear()
assert not auth.secrets_file.exists()
def test_login_with_browser(click_launch: mock.MagicMock, head: mock.MagicMock,
run: mock.MagicMock, port: mock.MagicMock):
port.return_value = 1234
login.Auth()._run_server()
url = f"{LIGHTNING_CLOUD_URL}/sign-in?redirectTo=http%3A%2F%2Flocalhost%3A1234%2Flogin-complete" # E501
head.assert_called_once_with(url)
click_launch.assert_called_once_with(url)
run.assert_called_once()
def test_with_api_key_only():
auth = login.Auth()
auth.save(user_id="7c8455e3-7c5f-4697-8a6d-105971d6b9bd",
api_key="e63fae57-2b50-498b-bc46-d6204cbf330e")
assert (
auth.authenticate() ==
"Basic N2M4NDU1ZTMtN2M1Zi00Njk3LThhNmQtMTA1OTcxZDZiOWJkOmU2M2ZhZTU3LTJiNTAtNDk4Yi1iYzQ2LWQ2MjA0Y2JmMzMwZQ==" # noqa E501
)
def test_authentication_with_environment_vars(browser_login: mock.MagicMock):
os.environ.setdefault("LIGHTNING_USER_ID", "abc")
os.environ.setdefault("LIGHTNING_API_KEY", "abc")
auth = login.Auth()
assert auth.user_id == "abc"
assert auth.auth_header == "Basic YWJjOmFiYw=="
assert auth._with_env_var is True
assert auth.authenticate() == auth.auth_header
# should not run login flow when env vars are passed
browser_login.assert_not_called()
def test_auth_header(self):
# fake credentials
os.environ.setdefault("LIGHTNING_USER_ID",
"7c8455e3-7c5f-4697-8a6d-105971d6b9bd")
os.environ.setdefault("LIGHTNING_API_KEY",
"e63fae57-2b50-498b-bc46-d6204cbf330e")
auth = login.Auth()
assert "Basic" in auth.auth_header
assert (
auth.auth_header ==
"Basic N2M4NDU1ZTMtN2M1Zi00Njk3LThhNmQtMTA1OTcxZDZiOWJkOmU2M2ZhZTU3LTJiNTAtNDk4Yi1iYzQ2LWQ2MjA0Y2JmMzMwZQ==" # noqa E501
)
def test_authenticate(click_launch: mock.MagicMock, head: mock.MagicMock,
run: mock.MagicMock, port: mock.MagicMock):
port.return_value = 1234
auth = login.Auth()
auth.user_id = "user_id"
auth.api_key = "api_key"
auth.authenticate()
url = f"{LIGHTNING_CLOUD_URL}/sign-in?redirectTo=http%3A%2F%2Flocalhost%3A1234%2Flogin-complete" # E501
head.assert_called_with(url)
click_launch.assert_called_with(url)
run.assert_called()
assert auth.auth_header == "Basic dXNlcl9pZDphcGlfa2V5"
auth.authenticate()
assert auth.auth_header == "Basic dXNlcl9pZDphcGlfa2V5"
def test_authentication_with_invalid_environment_vars():
# if api key is passed without user id
os.environ.setdefault("LIGHTNING_API_KEY", "123")
with pytest.raises(ValueError):
login.Auth()
def test_get_auth_header_should_raise_error(self):
with pytest.raises(AttributeError):
login.Auth().auth_header
def test_e2e(self):
auth = login.Auth()
assert auth._with_env_var is False
auth.save(username="******", user_id="kr-1234")
assert auth.secrets_file.exists()