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(), }) ], }) ])
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=[])
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=[])