from src import aws_utils


def test_split_s3():
    s3_path = 's3://abucket/path/to/file.txt'

    result = aws_utils.split_s3_path(s3_path)

    assert result == ('abucket', 'path/to/file.txt')


def test_parse_arn():
    arn = ('arn:aws:dynamodb:us-east-1:103233932089:table/'
           'dev-smart-downloader-tasks/stream/2023-11-10T10:48:34.578')

    result = aws_utils.parse_arn(arn)

    assert result == {
        'account': '103233932089',
        'arn': 'arn',
        'partition': 'aws',
        'region': 'us-east-1',
        'resource': 'dev-smart-downloader-tasks/stream/'
                    '2023-11-10T10:48:34.578',
        'resource_type': 'table',
        'service': 'dynamodb'}