def setUp(self):
self.result_queue = queue.Queue()
self.tempdir = tempfile.mkdtemp()
self.filename = mock.Mock()
self.filename.src = 'bucket/key'
self.filename.dest = os.path.join(self.tempdir, 'local', 'file')
self.filename.operation_name = 'download'
self.context = mock.Mock()
self.task = CreateLocalFileTask(self.context, self.filename,
self.result_queue)
def create_task(self):
# We don't actually care about the arguments, we just want to test
# the ordering of the tasks.
return CreateLocalFileTask(None, None)