def get_args(self, kind=None): mock_parse = get_mock_parser() if kind is not None: type(mock_parse.return_value).kind = mock.PropertyMock( return_value=kind) with mock.patch('argparse.ArgumentParser.parse_args', mock_parse): return get_arguments_parser().parse_args()
def setUp(self): super(BaseInvoicesUtilityTest, self).setUp() mock_parse = get_mock_parser() with mock.patch('argparse.ArgumentParser.parse_args', mock_parse): args = get_arguments_parser().parse_args() config = read_config(args.config) self.utility = invoices.InvoicesUtility(args.broker, args.period, config)
def setUp(self): super(BaseRefundsUtilityTest, self).setUp() mock_parse = get_mock_parser() type(mock_parse.return_value).kind = mock.PropertyMock( return_value=['general']) with mock.patch('argparse.ArgumentParser.parse_args', mock_parse): self.utility = refunds.RefundsUtility()
def setUp(self): super(BaseRefundsUtilityTest, self).setUp() mock_parse = get_mock_parser() type(mock_parse.return_value).kind = mock.PropertyMock( return_value=['general']) with mock.patch('argparse.ArgumentParser.parse_args', mock_parse): args = get_arguments_parser().parse_args() config = read_config(args.config) self.utility = refunds.RefundsUtility(args.broker, args.period, config, kind=args.kind)
def setUp(self): super(BaseInvoicesUtilityTest, self).setUp() mock_parse = get_mock_parser() with mock.patch('argparse.ArgumentParser.parse_args', mock_parse): self.utility = invoices.InvoicesUtility()