Python Agent.agent_connect示例

示例#1

文件： test_agent.py 项目： zhangxsgit/prefect

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

示例#2

文件： test_agent.py 项目： zhangxsgit/prefect

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

示例#3

文件： test_agent.py 项目： bmedx/prefect

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"

示例#4

文件： test_agent.py 项目： bmedx/prefect

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()

示例#5

文件： test_agent.py 项目： marplemr/prefect

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"

示例#6

文件： test_agent.py 项目： marplemr/prefect

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()