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, [])