diff --git a/pyatlan/model/assets/s3_object.py b/pyatlan/model/assets/s3_object.py index 68f232c4..a0971ccb 100644 --- a/pyatlan/model/assets/s3_object.py +++ b/pyatlan/model/assets/s3_object.py @@ -44,7 +44,13 @@ def creator( "s3_bucket_name", "s3_bucket_qualified_name", ], - [name, connection_qualified_name, aws_arn,s3_bucket_name,s3_bucket_qualified_name], + [ + name, + connection_qualified_name, + aws_arn, + s3_bucket_name, + s3_bucket_qualified_name, + ], ) attributes = S3Object.Attributes.create( name=name, @@ -101,7 +107,13 @@ def create_with_prefix( "s3_bucket_name", "s3_bucket_qualified_name", ], - [name, connection_qualified_name, prefix, s3_bucket_name, s3_bucket_qualified_name], + [ + name, + connection_qualified_name, + prefix, + s3_bucket_name, + s3_bucket_qualified_name, + ], ) attributes = S3Object.Attributes.create_with_prefix( name=name, @@ -351,7 +363,13 @@ def create( "s3_bucket_name", "s3_bucket_qualified_name", ], - [name, connection_qualified_name, aws_arn, s3_bucket_name, s3_bucket_qualified_name], + [ + name, + connection_qualified_name, + aws_arn, + s3_bucket_name, + s3_bucket_qualified_name, + ], ) fields = connection_qualified_name.split("/") if len(fields) != 3: @@ -370,7 +388,7 @@ def create( connection_qualified_name=connection_qualified_name, qualified_name=f"{connection_qualified_name}/{aws_arn}", connector_name=connector_type.value, - s3_bucket_name= s3_bucket_name, + s3_bucket_name=s3_bucket_name, s3_bucket_qualified_name=s3_bucket_qualified_name, bucket=S3Bucket.ref_by_qualified_name(s3_bucket_qualified_name), ) @@ -394,7 +412,13 @@ def create_with_prefix( "s3_bucket_name", "s3_bucket_qualified_name", ], - [name, connection_qualified_name, prefix, s3_bucket_name, s3_bucket_qualified_name], + [ + name, + connection_qualified_name, + prefix, + s3_bucket_name, + s3_bucket_qualified_name, + ], ) fields = connection_qualified_name.split("/") if len(fields) != 3: diff --git a/tests/integration/s3_asset_test.py b/tests/integration/s3_asset_test.py index 9405b7fa..843064f5 100644 --- a/tests/integration/s3_asset_test.py +++ b/tests/integration/s3_asset_test.py @@ -138,7 +138,7 @@ def _assert_object(s3object, bucket, with_name=False): assert s3object.qualified_name assert s3object.name == OBJECT_NAME assert s3object.connector_name == AtlanConnectorType.S3.value - assert s3object.s3_bucket_name == bucket.name + assert s3object.s3_bucket_name == bucket.name assert s3object.s3_bucket_qualified_name == bucket.qualified_name if with_name: assert s3object.aws_arn is None diff --git a/tests/unit/model/s3object_test.py b/tests/unit/model/s3object_test.py index 209e0623..ab5f7519 100644 --- a/tests/unit/model/s3object_test.py +++ b/tests/unit/model/s3object_test.py @@ -138,7 +138,12 @@ ], ) def test_create_without_required_parameters_raises_validation_error( - name, connection_qualified_name, aws_arn, s3_bucket_name, s3_bucket_qualified_name, msg + name, + connection_qualified_name, + aws_arn, + s3_bucket_name, + s3_bucket_qualified_name, + msg, ): with pytest.raises(ValueError, match=msg): S3Object.create( @@ -276,7 +281,12 @@ def test_create_without_required_parameters_raises_validation_error( ], ) def test_create_with_prefix_without_required_parameters_raises_validation_error( - name, connection_qualified_name, prefix, s3_bucket_name, s3_bucket_qualified_name, msg + name, + connection_qualified_name, + prefix, + s3_bucket_name, + s3_bucket_qualified_name, + msg, ): with pytest.raises(ValueError, match=msg): S3Object.create_with_prefix(