def test_serialize_local_environment():
env = environments.LocalEnvironment(encryption_key=FERNET_KEY)
serialized = LocalEnvironmentSchema().dump(env)
assert (
serialized["encryption_key"]
== "MWNyZGVyVEhWSjd2dlZKajc5Wm5zODFfMW9wYVRJRDBIUlpvT3pxSXBPQT0="
)
assert serialized["__version__"] == prefect.__version__
def test_serialize_local_environment_with_labels():
env = environments.LocalEnvironment(labels=["bob", "alice"])
schema = RemoteEnvironmentSchema()
serialized = schema.dump(env)
assert serialized
assert serialized["__version__"] == prefect.__version__
assert set(serialized["labels"]) == set(["bob", "alice"])
new = schema.load(serialized)
assert new.labels == set(["bob", "alice"])
def test_serialize_local_environment_with_labels():
env = environments.LocalEnvironment(labels=["b", "c", "a"])
schema = LocalEnvironmentSchema()
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 == {"b", "c", "a"}
def test_environment_schema_with_local_environment():
env = environments.LocalEnvironment(encryption_key=FERNET_KEY)
serialized = EnvironmentSchema().dump(env)
deserialized = EnvironmentSchema().load(serialized)
assert isinstance(deserialized, environments.LocalEnvironment)
assert deserialized.encryption_key == FERNET_KEY