"""Test s3 helpers."""

from royalties.utils.aws import parse_s3_url


def test_parse_s3_url():
    """Test s3 url is parsed for the bucket name and key."""
    s3_url = 's3://bucket/key/nested-key'
    bucket, key = parse_s3_url(s3_url)
    assert bucket == 'bucket'
    assert key == 'key/nested-key'