def test_on_failure(self):
subscriber = OnDoneFilteredRecordingSubscriber()
exception = Exception('my exception')
future = FakeTransferFuture(exception=exception)
subscriber.on_done(future)
self.assertEqual(subscriber.on_failure_calls, [(future, exception)])
self.assertEqual(subscriber.on_success_calls, [])
def setUp(self):
super(TestDirectoryCreatorSubscriber, self).setUp()
self.directory_to_create = os.path.join(self.file_creator.rootdir,
'new-directory')
self.filename = os.path.join(self.directory_to_create, 'myfile')
call_args = FakeTransferFutureCallArgs(fileobj=self.filename)
meta = FakeTransferFutureMeta(call_args=call_args)
self.future = FakeTransferFuture(meta=meta)
self.subscriber = DirectoryCreatorSubscriber()
def get_transfer_future(self, size=0):
return FakeTransferFuture(meta=FakeTransferFutureMeta(
call_args=FakeTransferFutureCallArgs(copy_source={
'Bucket': self.source_bucket,
'Key': self.source_key,
},
bucket=self.bucket,
key=self.key,
extra_args={}),
size=size,
user_context={},
))
def setUp(self):
super(TestProvideLastModifiedTimeSubscriber, self).setUp()
self.filename = self.file_creator.create_file('myfile', 'my contents')
self.desired_utime = datetime.datetime(
2016, 1, 18, 7, 0, 0, tzinfo=tzlocal())
self.result_queue = queue.Queue()
self.subscriber = ProvideLastModifiedTimeSubscriber(
self.desired_utime, self.result_queue)
call_args = FakeTransferFutureCallArgs(fileobj=self.filename)
meta = FakeTransferFutureMeta(call_args=call_args)
self.future = FakeTransferFuture(meta=meta)
def set_future(self):
copy_source = {'Bucket': 'mybucket', 'Key': self.filename}
call_args = FakeTransferFutureCallArgs(copy_source=copy_source,
extra_args=self.extra_args)
meta = FakeTransferFutureMeta(call_args=call_args)
return FakeTransferFuture(meta=meta)
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_success(self):
subscriber = OnDoneFilteredRecordingSubscriber()
future = FakeTransferFuture('return-value')
subscriber.on_done(future)
self.assertEqual(subscriber.on_success_calls, [future])
self.assertEqual(subscriber.on_failure_calls, [])