def test_show_current_transfers(self): """Test the --current_transfers output with transfers in progress.""" # create a download fake_download = FakeDownload('share_id', 'down_node_id') fake_download.deflated_size = 10 fake_download.n_bytes_read = 1 fake_download.path = "down_path" self.action_q.queue.waiting.append(fake_download) # create an upload fake_upload = FakeUpload('share_id', 'node_id') fake_upload.deflated_size = 100 fake_upload.n_bytes_written = 10 fake_upload.path = "up_path" self.action_q.queue.waiting.append(fake_upload) out = StringIO() expected = ( u"Current uploads:\n path: up_path\n deflated size: 100\n " u"bytes written: 10\nCurrent downloads:\n path: down_path\n " u"deflated size: 10\n bytes read: 1\n") result = yield self.tool.get_current_uploads() show_uploads(result, out) result = yield self.tool.get_current_downloads() show_downloads(result, out) self.assertEqual(out.getvalue(), expected)
def __init__(self, *args): FakeUpload.__init__(self, *args) self.path = 'other_path'