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_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()