def test_gcs_empty_serialize(): gcs = storage.GCS(bucket="bucket") serialized = GCSSchema().dump(gcs) assert serialized assert serialized["__version__"] == prefect.__version__ assert serialized["bucket"] assert not serialized["key"] assert serialized["secrets"] == []
def test_gcs_full_serialize(): gcs = storage.GCS( bucket="bucket", key="key", project="project", secrets=["CREDS"], labels=["foo", "bar"], add_default_labels=False, )
def test_gcs_serialize_with_flows(): gcs = storage.GCS(project="project", bucket="bucket", key="key", secrets=["CREDS"]) f = prefect.Flow("test") gcs.flows["test"] = "key" serialized = GCSSchema().dump(gcs) assert serialized assert serialized["__version__"] == prefect.__version__ assert serialized["bucket"] == "bucket" assert serialized["key"] == "key" assert serialized["project"] == "project" assert serialized["flows"] == {"test": "key"} deserialized = GCSSchema().load(serialized) assert f.name in deserialized assert deserialized.secrets == ["CREDS"]
def test_gcs_full_serialize(): gcs = storage.GCS( bucket="bucket", key="key", project="project", secrets=["CREDS"], labels=["foo", "bar"], add_default_labels=False, ) serialized = GCSSchema().dump(gcs) assert serialized assert serialized["__version__"] == prefect.__version__ assert serialized["bucket"] == "bucket" assert serialized["key"] == "key" assert serialized["project"] == "project" assert serialized["secrets"] == ["CREDS"]
def test_gcs_serialize_with_flows(): gcs = storage.GCS(project="project", bucket="bucket", key="key", secrets=["CREDS"])
def test_gcs_empty_serialize(): gcs = storage.GCS(bucket="bucket")