def test_agent_fails_no_runner_token(monkeypatch):
post = MagicMock(return_value=MagicMock(json=MagicMock(return_value=dict(
data=dict(authInfo=MagicMock(apiTokenScope="USER"))))))
session = MagicMock()
session.return_value.post = post
monkeypatch.setattr("requests.Session", session)
with pytest.raises(AuthorizationError):
agent = Agent()
agent.query_tenant_id()
def test_query_tenant_id(monkeypatch, runner_token):
post = MagicMock(return_value=MagicMock(json=MagicMock(return_value=dict(
data=dict(tenant=[dict(id="id")])))))
session = MagicMock()
session.return_value.post = post
monkeypatch.setattr("requests.Session", session)
agent = Agent()
tenant_id = agent.query_tenant_id()
assert tenant_id == "id"
def test_query_tenant_id_not_found(monkeypatch):
with set_temporary_config({"cloud.agent.auth_token": "token"}):
post = MagicMock(return_value=MagicMock(json=MagicMock(
return_value=dict(data=dict(tenant=[])))))
session = MagicMock()
session.return_value.post = post
monkeypatch.setattr("requests.Session", session)
agent = Agent()
tenant_id = agent.query_tenant_id()
assert not tenant_id
def test_agent_fails_no_auth_token():
with pytest.raises(AuthorizationError):
agent = Agent()
agent.query_tenant_id()
def test_agent_fails_no_auth_token():
with set_temporary_config({"cloud.agent.auth_token": None}):
with pytest.raises(AuthorizationError):
agent = Agent()
agent.query_tenant_id()
