def test_azure_creds_not_serialized(): azure = storage.Azure( container="container", connection_string="conn", blob_name="name" ) serialized = AzureSchema().dump(azure) assert serialized assert serialized["__version__"] == prefect.__version__ assert serialized["container"] == "container" assert serialized["blob_name"] == "name" assert serialized.get("connection_string") is None
def test_azure_serialize_with_flows(): azure = storage.Azure(container="container", connection_string="conn", blob_name="name") f = prefect.Flow("test") azure.flows["test"] = "key" serialized = AzureSchema().dump(azure) assert serialized assert serialized["__version__"] == prefect.__version__ assert serialized["container"] == "container" assert serialized["blob_name"] == "name" assert serialized.get("connection_string") is None assert serialized["flows"] == {"test": "key"} deserialized = AzureSchema().load(serialized) assert f.name in deserialized