class TestProvideUploadContentTypeSubscriber(unittest.TestCase):
def setUp(self):
self.filename = "myfile.txt"
self.extra_args = {}
self.future = self.set_future()
self.subscriber = ProvideUploadContentTypeSubscriber()
def set_future(self):
call_args = FakeTransferFutureCallArgs(fileobj=self.filename, extra_args=self.extra_args)
meta = FakeTransferFutureMeta(call_args=call_args)
return FakeTransferFuture(meta=meta)
def test_on_queued_provides_content_type(self):
self.subscriber.on_queued(self.future)
self.assertEqual(self.extra_args, {"ContentType": "text/plain"})
def test_on_queued_does_not_provide_content_type_when_unknown(self):
self.filename = "file-with-no-extension"
self.future = self.set_future()
self.subscriber.on_queued(self.future)
self.assertEqual(self.extra_args, {})
class TestProvideUploadContentTypeSubscriber(unittest.TestCase):
def setUp(self):
self.filename = 'myfile.txt'
self.extra_args = {}
self.future = self.set_future()
self.subscriber = ProvideUploadContentTypeSubscriber()
def set_future(self):
call_args = FakeTransferFutureCallArgs(fileobj=self.filename,
extra_args=self.extra_args)
meta = FakeTransferFutureMeta(call_args=call_args)
return FakeTransferFuture(meta=meta)
def test_on_queued_provides_content_type(self):
self.subscriber.on_queued(self.future)
self.assertEqual(self.extra_args, {'ContentType': 'text/plain'})
def test_on_queued_does_not_provide_content_type_when_unknown(self):
self.filename = 'file-with-no-extension'
self.future = self.set_future()
self.subscriber.on_queued(self.future)
self.assertEqual(self.extra_args, {})
def setUp(self):
self.filename = "myfile.txt"
self.extra_args = {}
self.future = self.set_future()
self.subscriber = ProvideUploadContentTypeSubscriber()
def setUp(self):
self.filename = 'myfile.txt'
self.extra_args = {}
self.future = self.set_future()
self.subscriber = ProvideUploadContentTypeSubscriber()
def _add_additional_subscribers(self, subscribers, fileinfo):
subscribers.append(ProvideSizeSubscriber(fileinfo.size))
if self._should_inject_content_type():
subscribers.append(ProvideUploadContentTypeSubscriber())
if self._cli_params.get('is_move', False):
subscribers.append(DeleteSourceFileSubscriber())
def _add_additional_subscribers(self, subscribers, fileinfo):
subscribers.append(ProvideSizeSubscriber(fileinfo.size))
if self._should_inject_content_type():
subscribers.append(ProvideUploadContentTypeSubscriber())
