Python Agent.update_states Examples

Example #1

File: test_agent.py Project: bmedx/prefect

def test_update_states_passes_task_runs(monkeypatch, runner_token):
    gql_return = MagicMock(return_value=MagicMock(
        data=MagicMock(set_flow_run_state=None, set_task_run_state=None)))
    client = MagicMock()
    client.return_value.graphql = gql_return
    monkeypatch.setattr("prefect.agent.agent.Client", client)

    agent = Agent()
    assert not agent.update_states(flow_runs=[
        GraphQLResult({
            "id":
            "id",
            "serialized_state":
            Scheduled().serialize(),
            "version":
            1,
            "task_runs": [
                GraphQLResult({
                    "id": "id",
                    "version": 1,
                    "serialized_state": Scheduled().serialize(),
                })
            ],
        })
    ])

Example #2

File: test_agent.py Project: bmedx/prefect

def test_update_states_passes_empty(monkeypatch, runner_token):
    gql_return = MagicMock(return_value=MagicMock(
        data=MagicMock(set_flow_run_state=None, set_task_run_state=None)))
    client = MagicMock()
    client.return_value.graphql = gql_return
    monkeypatch.setattr("prefect.agent.agent.Client", client)

    agent = Agent()
    assert not agent.update_states(flow_runs=[])

Example #3

File: test_agent.py Project: marplemr/prefect

def test_update_states_passes_empty(monkeypatch):
    with set_temporary_config({"cloud.agent.auth_token": "token"}):
        gql_return = MagicMock(return_value=MagicMock(
            data=MagicMock(set_flow_run_state=None, set_task_run_state=None)))
        client = MagicMock()
        client.return_value.graphql = gql_return
        monkeypatch.setattr("prefect.agent.agent.Client", client)

        agent = Agent()
        assert not agent.update_states(flow_runs=[])