def test_create_common_parser_passed_usage_string(self, mock_func): mock_parser = Mock() mock_parser.parse_args.return_value = lambda: Mock(), [] mock_func.return_value = mock_parser restorewal_parse_args(args=[]) kwargs = mock_func.call_args_list[0][1] expected_usage_suffix = ' '.join([ '[options]', '<name_of_wal_file_to_restore>', '<path_to_write_restored_file>', ]) self.assertTrue(kwargs['usage'].endswith(expected_usage_suffix))
def restorewal_handle_args(): parser, options, args = restorewal_parse_args() try: restorewal_validate_options_and_args(options, args) except Exception, e: stdout.write(str(e) + '\n') parser.print_help() raise e
def setUp(self): self.tempdir = TempDirectory() self.config_dict = { 'General': { 'pgsql_data_directory': self.tempdir.path, }, } self.config_file = os.path.join(self.tempdir.path, 'config_file') write_config_to_filename(self.config_dict, self.config_file) parser, self.options, self.args = restorewal_parse_args(['-c', self.config_file])
def test_restorewal_parse_args_returns_args(self): item1, item2, args = restorewal_parse_args(args=None) self.assertEqual(type(args), type([]))
def test_restorewal_parse_args_returns_options(self): item1, options, item3 = restorewal_parse_args(args=None) self.assertTrue(isinstance(options, object))
def test_restorewal_parse_args_returns_parser(self): parser, item2, item3 = restorewal_parse_args(args=None) self.assertTrue(isinstance(parser, OptionParser))
def test_restorewal_parse_args_returns_three_items(self): item1, item2, item3 = restorewal_parse_args() self.assertNotEqual(type(item1), type(None)) self.assertNotEqual(type(item2), type(None)) self.assertNotEqual(type(item3), type(None))
def test_create_common_parser_used(self, mock_func): mock_parser = Mock() mock_parser.parse_args.return_value = lambda: Mock(), [] mock_func.return_value = mock_parser restorewal_parse_args(args=[]) self.assertTrue(mock_func.called)