def test_run_flow(monkeypatch, tmpdir): environment = Environment() flow_runner = MagicMock() flow_runner_class = MagicMock(return_value=flow_runner) monkeypatch.setattr( "prefect.engine.get_default_flow_runner_class", MagicMock(return_value=flow_runner_class), ) d = Local(str(tmpdir)) d.add_flow(Flow("name")) gql_return = MagicMock(return_value=MagicMock(data=MagicMock(flow_run=[ GraphQLResult({ "flow": GraphQLResult({ "name": "name", "storage": d.serialize() }) }) ], ))) client = MagicMock() client.return_value.graphql = gql_return monkeypatch.setattr("prefect.environments.execution.base.Client", client) with set_temporary_config({"cloud.auth_token": "test"}), prefect.context({"flow_run_id": "id"}): environment.run_flow() assert flow_runner_class.call_args[1]["flow"].name == "name" assert flow_runner.run.call_args[1]["executor"] is not None
def test_create_environment(): environment = Environment() assert environment assert environment.labels == set() assert environment.on_start is None assert environment.on_exit is None assert environment.logger.name == "prefect.Environment"
def create_object(self, data: dict, **kwargs: Any) -> Environment: """ Because we cannot deserialize a custom class, we return an empty Base Environment with the appropriate labels. """ return Environment(labels=data.get("labels"), metadata=data.get("metadata"))
def test_create_environment_callbacks(): def f(): pass environment = Environment(on_start=f, on_exit=f) assert environment.on_start is f assert environment.on_exit is f
def test_run_flow_no_flow_run_id_in_context(monkeypatch, tmpdir): environment = Environment() d = Local(str(tmpdir)) d.add_flow(Flow("name")) with set_temporary_config({"cloud.auth_token": "test"}): with pytest.raises(ValueError): environment.run_flow()
def test_run_flow_no_flow_run_id_in_context(monkeypatch): environment = Environment() with tempfile.TemporaryDirectory() as directory: d = Local(directory) d.add_flow(Flow("name")) with set_temporary_config({"cloud.auth_token": "test"}): with pytest.raises(ValueError): environment.run_flow()
def test_serialize_environment(): environment = Environment() env = environment.serialize() assert env["type"] == "Environment"
def test_create_environment(): environment = Environment() assert environment
def test_setup_environment_passes(): environment = Environment() environment.setup(storage=Docker()) assert environment
def test_execute_environment_passes(): environment = Environment() environment.execute(storage=Docker(), flow_location="") assert environment
def test_execute_environment_passes(): environment = Environment() environment.execute(flow=Flow("test", storage=Docker())) assert environment
def test_environment_build_error(): environment = Environment() with pytest.raises(NotImplementedError): environment.build(1)
def test_setup_environment_passes(): environment = Environment() environment.setup(flow=Flow("test", storage=Docker())) assert environment
def test_environment_dependencies(): environment = Environment() assert environment.dependencies == []
def test_create_environment(): environment = Environment() assert environment assert environment.labels == set() assert environment.logger.name == "prefect.Environment"
def test_create_environment_metadata(): environment = Environment(metadata={"test": "here"}) assert environment assert environment.metadata == {"test": "here"}
def test_create_environment_converts_labels_to_set(): environment = Environment(labels=["a", "b", "a"]) assert environment assert environment.labels == set(["a", "b"]) assert environment.logger.name == "prefect.Environment"