import unittest.mock

import slz_downloader.dsp.base
import slz_downloader.dsp.s3


def test_s3():
    inner_dsp = unittest.mock.Mock()
    inner_dsp.get_content_name.return_value = content_name = 'name'
    inner_dsp.get_original_content_name.return_value = origin_name = 'origin'

    client = slz_downloader.dsp.s3.Client(unittest.mock.Mock(), inner_dsp)

    job = unittest.mock.Mock()

    assert client.get_content_name(job) == content_name
    assert client.get_original_content_name(job) == origin_name