def test_agent_connect_handled_error(monkeypatch, runner_token, cloud_api): post = MagicMock(side_effect=Exception) session = MagicMock() session.return_value.post = post monkeypatch.setattr("requests.Session", session) agent = Agent() assert agent.agent_connect() is None
def test_agent_connect(monkeypatch, runner_token, cloud_api): post = MagicMock(return_value=MagicMock(json=MagicMock(return_value="hello"))) session = MagicMock() session.return_value.post = post monkeypatch.setattr("requests.Session", session) agent = Agent() assert agent.agent_connect() is None
def test_agent_connect(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() assert agent.agent_connect() == "id"
def test_agent_connect_no_tenant_id(monkeypatch, runner_token): post = MagicMock(return_value=MagicMock(json=MagicMock(return_value=dict( data=dict(tenant=[dict(id=None)]))))) session = MagicMock() session.return_value.post = post monkeypatch.setattr("requests.Session", session) agent = Agent() with pytest.raises(ConnectionError): assert agent.agent_connect()
def test_agent_connect(monkeypatch): with set_temporary_config({"cloud.agent.auth_token": "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() assert agent.agent_connect() == "id"
def test_agent_connect_no_tenant_id(monkeypatch): with set_temporary_config({"cloud.agent.auth_token": "token"}): post = MagicMock(return_value=MagicMock(json=MagicMock( return_value=dict(data=dict(tenant=[dict(id=None)]))))) session = MagicMock() session.return_value.post = post monkeypatch.setattr("requests.Session", session) agent = Agent() with pytest.raises(ConnectionError): assert agent.agent_connect()