def __run_test_handle_dist_repo(self, arguments, return_value=None, expected=''): expected = expected + "Creating dist repo for tag " + self.tag_name + "\n" with mock.patch('sys.stdout', new_callable=six.StringIO) as stdout: rv = handle_dist_repo(self.options, self.session, arguments) self.assertEqual(rv, return_value) self.assert_console_message(stdout, expected) self.activate_session.assert_called_with(self.session, self.options)
def __run_test_handle_dist_repo(self, arguments, stdout, stderr, return_value=None, expected='', expected_warn=''): expected = expected + "Creating dist repo for tag " + self.tag_name + "\n" rv = handle_dist_repo(self.options, self.session, arguments) self.assertEqual(rv, return_value) self.assert_console_message(stdout, expected) self.assert_console_message(stderr, expected_warn) self.activate_session.assert_called_with(self.session, self.options)