def test_start_waiting_debounces(self, mock_progress_bar): progress_printer = ProgressPrinter(total=10, msg_verb='uploading') progress_printer.start_waiting() self.assertEqual(1, progress_printer.progress_bar.show_waiting.call_count) progress_printer.progress_bar.show_waiting.assert_called_with( 'Waiting for project to become ready for uploading') progress_printer.start_waiting() progress_printer.start_waiting() self.assertEqual(1, progress_printer.progress_bar.show_waiting.call_count)
def test_done_waiting(self, mock_progress_bar): progress_printer = ProgressPrinter(total=10, msg_verb='downloading') progress_printer.start_waiting() self.assertEqual(True, progress_printer.waiting) self.assertEqual(1, progress_printer.progress_bar.show_waiting.call_count) progress_printer.progress_bar.show_waiting.assert_called_with( 'Waiting for project to become ready for downloading') progress_printer.done_waiting() self.assertEqual(False, progress_printer.waiting) self.assertEqual(1, progress_printer.progress_bar.show_running.call_count)
def test_stuff(self, mock_progress_bar): progress_printer = ProgressPrinter(total=10, msg_verb='sending') # pretend we just created a project mock_project = Mock(kind=KindType.project_str, path='') progress_printer.transferring_item(item=mock_project, increment_amt=1) mock_progress_bar.return_value.update.assert_called_with( 10, 'sending project') mock_progress_bar.return_value.show.assert_called() mock_progress_bar.reset_mock() # pretend we just created a folder mock_project = Mock(kind=KindType.folder_str, path='/data') progress_printer.transferring_item(item=mock_project, increment_amt=2) mock_progress_bar.return_value.update.assert_called_with( 30, 'sending data') mock_progress_bar.return_value.show.assert_called() mock_progress_bar.reset_mock() # pretend we get stuck waiting for project uploads to be ready progress_printer.start_waiting() mock_progress_bar.return_value.show_waiting.assert_called_with( 'Waiting for project to become ready for sending') mock_progress_bar.reset_mock() # pretend project uploads are ready progress_printer.done_waiting() mock_progress_bar.return_value.show_running.assert_called() mock_progress_bar.reset_mock() # pretend we uploaded a file mock_project = Mock(kind=KindType.file_str, path='/data/log.txt') progress_printer.transferring_item(item=mock_project, increment_amt=2) mock_progress_bar.return_value.update.assert_called_with( 50, 'sending log.txt') mock_progress_bar.return_value.show.assert_called() mock_progress_bar.reset_mock() # pretend we are finished uploading progress_printer.finished() mock_progress_bar.return_value.set_state.assert_called_with( mock_progress_bar.STATE_DONE) mock_progress_bar.return_value.show.assert_called() mock_progress_bar.reset_mock()
def test_stuff(self, mock_progress_bar): progress_printer = ProgressPrinter(total=10, msg_verb='sending') # pretend we just created a project mock_project = Mock(kind=KindType.project_str, path='') progress_printer.transferring_item(item=mock_project, increment_amt=1) mock_progress_bar.return_value.update.assert_called_with(10, 'sending project') mock_progress_bar.return_value.show.assert_called() mock_progress_bar.reset_mock() # pretend we just created a folder mock_project = Mock(kind=KindType.folder_str, path='/data') progress_printer.transferring_item(item=mock_project, increment_amt=2) mock_progress_bar.return_value.update.assert_called_with(30, 'sending data') mock_progress_bar.return_value.show.assert_called() mock_progress_bar.reset_mock() # pretend we get stuck waiting for project uploads to be ready progress_printer.start_waiting() mock_progress_bar.return_value.show_waiting.assert_called_with('Waiting for project to become ready for sending') mock_progress_bar.reset_mock() # pretend project uploads are ready progress_printer.done_waiting() mock_progress_bar.return_value.show_running.assert_called() mock_progress_bar.reset_mock() # pretend we uploaded a file mock_project = Mock(kind=KindType.file_str, path='/data/log.txt') progress_printer.transferring_item(item=mock_project, increment_amt=2) mock_progress_bar.return_value.update.assert_called_with(50, 'sending log.txt') mock_progress_bar.return_value.show.assert_called() mock_progress_bar.reset_mock() # pretend we are finished uploading progress_printer.finished() mock_progress_bar.return_value.set_state.assert_called_with(mock_progress_bar.STATE_DONE) mock_progress_bar.return_value.show.assert_called() mock_progress_bar.reset_mock()