def test_serialize_fargate_task_env_with_kwargs(): env = environments.FargateTaskEnvironment(cluster="test") schema = FargateTaskEnvironmentSchema() serialized = schema.dump(env) deserialized = schema.load(serialized) assert isinstance(deserialized, environments.FargateTaskEnvironment) assert deserialized.task_run_kwargs == {}
def test_serialize_fargate_task_environment_with_labels(): env = environments.FargateTaskEnvironment(labels=["a", "b", "c"]) schema = FargateTaskEnvironmentSchema() serialized = schema.dump(env) assert serialized assert serialized["__version__"] == prefect.__version__ assert set(serialized["labels"]) == set(["a", "b", "c"]) new = schema.load(serialized) assert new.labels == set(["a", "b", "c"])
def test_serialize_fargate_task_environment_with_labels(): env = environments.FargateTaskEnvironment(labels=["b", "c", "a"]) schema = FargateTaskEnvironmentSchema() serialized = schema.dump(env) assert serialized assert serialized["__version__"] == prefect.__version__ # labels should be sorted in the serialized obj assert serialized["labels"] == ["a", "b", "c"] new = schema.load(serialized) assert new.labels == {"a", "b", "c"}
def test_serialize_fargate_task_environment(): env = environments.FargateTaskEnvironment() schema = FargateTaskEnvironmentSchema() serialized = schema.dump(env) assert serialized assert serialized["__version__"] == prefect.__version__ assert serialized["labels"] == [] assert serialized["metadata"] == {} new = schema.load(serialized) assert new.labels == set()
def test_serialize_fargate_task_env_with_kwargs(): env = environments.FargateTaskEnvironment(cluster="test") schema = FargateTaskEnvironmentSchema() serialized = schema.dump(env) deserialized = schema.load(serialized) assert isinstance(deserialized, environments.FargateTaskEnvironment) assert deserialized.task_run_kwargs == {} def test_serialize_fargate_task_environment_with_labels(): <<<<<<< HEAD env = environments.FargateTaskEnvironment(labels=["b", "c", "a"]) ======= env = environments.FargateTaskEnvironment(labels=["a", "b", "c"]) >>>>>>> prefect clone schema = FargateTaskEnvironmentSchema() serialized = schema.dump(env) assert serialized assert serialized["__version__"] == prefect.__version__ <<<<<<< HEAD # labels should be sorted in the serialized obj assert serialized["labels"] == ["a", "b", "c"] new = schema.load(serialized) assert new.labels == {"a", "b", "c"} =======