Python Agent._retrieve_agent_configの例

コード例 #1

def test_agent_retrieve_config(monkeypatch, cloud_api):
    monkeypatch.setattr(
        "prefect.agent.agent.Client.get_agent_config",
        MagicMock(return_value={"settings": "yes"}),
    )

    agent = Agent(max_polls=1, agent_config_id="foo")
    assert agent._retrieve_agent_config() == {"settings": "yes"}

コード例 #2

ファイル: test_agent.py プロジェクト: zpencerq/prefect

def test_agent_rerieve_config(monkeypatch, cloud_api):
    monkeypatch.setattr("prefect.agent.agent.Agent._verify_token", MagicMock())
    monkeypatch.setattr("prefect.agent.agent.Client.register_agent",
                        MagicMock(return_value="ID"))
    monkeypatch.setattr(
        "prefect.agent.agent.Client.get_agent_config",
        MagicMock(return_value={"settings": "yes"}),
    )

    agent = Agent(max_polls=1)
    agent.start()
    assert agent._retrieve_agent_config() == {"settings": "yes"}

コード例 #3

def test_agent_retrieve_config_requires_config_id_set(cloud_api):
    agent = Agent(max_polls=1)
    with pytest.raises(ValueError, match="agent_config_id"):
        assert agent._retrieve_agent_config() == {"settings": "yes"}