def test_deserialize_docker_on_kubernetes_environment():
env = environments.kubernetes.DockerOnKubernetesEnvironment(
base_image="a", python_dependencies=["b", "c"], registry_url="f")
serialized = DockerOnKubernetesEnvironmentSchema().dump(env)
deserialized = DockerOnKubernetesEnvironmentSchema().load(serialized)
assert deserialized.base_image == env.base_image
assert deserialized.registry_url == env.registry_url
def test_serialize_docker_on_kubernetes_environment_no_base_image():
env = environments.kubernetes.DockerOnKubernetesEnvironment(
python_dependencies=["b", "c"], registry_url="f", image_name="g", image_tag="h"
)
serialized = DockerOnKubernetesEnvironmentSchema().dump(env)
assert serialized["base_image"] == "python:3.6"
assert serialized["registry_url"] == "f"
assert serialized["image_name"] == "g"
assert serialized["image_tag"] == "h"
assert serialized["__version__"] == prefect.__version__
def test_deserialize_minimal_docker_on_kubernetes_environment():
schema = DockerOnKubernetesEnvironmentSchema()
assert schema.load(schema.dump({"base_image": "a", "registry_url": "b"}))
def test_deserialize_empty_docker_on_kubernetes_environment():
schema = DockerOnKubernetesEnvironmentSchema()
with pytest.raises(marshmallow.ValidationError):
schema.load(schema.dump({}))