def test_upload_command_with_parameter(self, *args): with Capturing() as output: task1_main(self.files) correct_output = [ 'Done: 0, Error: 0, Total:5', 'Done: 5, Error: 0, Total:5', 'Exit' ] self.assertEqual(correct_output, output)
def test_not_supported_command(self, *args): with Capturing() as output: task1_main(self.files) self.assertEqual('Command not supported', output[0])
def test_invalid_command(self, *args): with Capturing() as output: task1_main(self.files) self.assertEqual('Invalid command', output[0])
def test_exit(self, *args): with Capturing() as output: task1_main(self.files) self.assertEqual('Exit', output[0])