"""Test s3 connector."""

import boto3
from freezegun import freeze_time

from video.connectors import s3 as s3_connector


def test_get_s3_resource(mocker):
    """Test get_s3_client."""
    mocker.patch.object(boto3, 'resource', autospec=True)
    connection = s3_connector.get_s3_client()

    assert connection


@freeze_time('2018-08-10')
def test_get_url_for_s3_object(mocker):
    """Test get_url_for_s3_object."""
    mocker.patch.object(boto3, 'client', autospec=True)
    test_bucket = 'test_bucket'
    test_key = 'test_key'
    url = s3_connector.get_url_for_s3_object(
        test_bucket,
        test_key
    )
    assert url