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()