def test_s3_serialize_with_flows(): s3 = storage.S3(bucket="bucket", key="key", secrets=["hidden", "auth"],) f = prefect.Flow("test") s3.flows["test"] = "key" serialized = S3Schema().dump(s3) assert serialized assert serialized["__version__"] == prefect.__version__ assert serialized["bucket"] == "bucket" assert serialized["key"] == "key" assert serialized["flows"] == {"test": "key"} deserialized = S3Schema().load(serialized) assert f.name in deserialized assert deserialized.secrets == ["hidden", "auth"]
def test_s3_empty_serialize(): s3 = storage.S3(bucket="bucket") serialized = S3Schema().dump(s3) assert serialized assert serialized["__version__"] == prefect.__version__ assert serialized["bucket"] assert not serialized["key"]
def test_s3_full_serialize(): s3 = storage.S3(bucket="bucket", key="key", secrets=["hidden", "auth"],) serialized = S3Schema().dump(s3) assert serialized assert serialized["__version__"] == prefect.__version__ assert serialized["bucket"] == "bucket" assert serialized["key"] == "key" assert serialized["secrets"] == ["hidden", "auth"]
def test_s3_serialize_with_flows(): s3 = storage.S3( aws_access_key_id="id", aws_secret_access_key="secret", aws_session_token="session", bucket="bucket", key="key", ) f = prefect.Flow("test") s3.flows["test"] = "key" serialized = S3Schema().dump(s3) assert serialized assert serialized["__version__"] == prefect.__version__ assert serialized["bucket"] == "bucket" assert serialized["key"] == "key" assert serialized["flows"] == {"test": "key"} deserialized = S3Schema().load(serialized) assert f.name in deserialized
def test_s3_full_serialize(): s3 = storage.S3( aws_access_key_id="id", aws_secret_access_key="secret", aws_session_token="session", bucket="bucket", key="key", ) serialized = S3Schema().dump(s3) assert serialized assert serialized["__version__"] == prefect.__version__ assert serialized["bucket"] == "bucket" assert serialized["key"] == "key"
def test_s3_aws_creds_not_serialized(): s3 = storage.S3( aws_access_key_id="id", aws_secret_access_key="secret", aws_session_token="session", bucket="bucket", key="key", ) serialized = S3Schema().dump(s3) assert serialized assert serialized["__version__"] == prefect.__version__ assert serialized["bucket"] == "bucket" assert serialized["key"] == "key" assert serialized.get("aws_access_key_id") is None assert serialized.get("aws_secret_access_key") is None assert serialized.get("aws_session_token") is None