import boto3
from moto import mock_sts, mock_s3


@mock_sts()
@mock_s3()
def test_put_delete_s3_object(app):
    from lemur.plugins.lemur_aws.s3 import put, delete, get

    bucket = "public-bucket"
    region = "us-east-1"
    account = "123456789012"
    path = "some-path/foo"
    data = "dummy data"

    s3_client = boto3.client('s3')
    s3_client.create_bucket(Bucket=bucket)

    put(bucket_name=bucket,
        region_name=region,
        prefix=path,
        data=data,
        encrypt=False,
        account_number=account,
        region=region)

    response = get(bucket_name=bucket, prefixed_object_name=path, account_number=account)

    # put data, and getting the same data
    assert (response == data)

    response = get(bucket_name="wrong-bucket", prefixed_object_name=path, account_number=account)

    # attempting to get thccle wrong data
    assert (response is None)

    delete(bucket_name=bucket, prefixed_object_name=path, account_number=account)
    response = get(bucket_name=bucket, prefixed_object_name=path, account_number=account)

    # delete data, and getting the same data
    assert (response is None)