def process_payment_method(self, form): payment_method = form.cleaned_data['payment_method'] payment_forms = get_payment_forms(payment_method, self.payment_forms) if len(payment_forms): self.form_list.insert(4, 'Payment', payment_forms[0]) elif 'Payment' in self.form_list: del self.form_list['Payment']
def test_get_payments_form_custom(self): """ Test that method can return custom payment forms """ custom_forms = [Mock()] * 3 with patch('django.conf.settings.PAYMENT_MODULES', {'dummy': 'salest.payments.modules.dummy'}): forms_from_module = get_payment_forms('dummy', custom_forms) self.assertEqual(custom_forms, forms_from_module)
def test_get_payment_forms_method(self): """ Test that method can return forms from module """ with patch('django.conf.settings.PAYMENT_MODULES', {'dummy': 'salest.payments.modules.dummy'}): forms_from_module = get_payment_forms('dummy') self.assertEquals(forms_from_module, dummy_default_forms)
def add_paymet_form(self): """ Add payment for to form wizard """ module_name = self.get_payment_method_name_from_request() if module_name: self.request.session['module_name'] = module_name payment_forms = get_payment_forms(module_name, self.payment_forms) forms_count = self.steps.count forms_list = {} for counter, payment_form in enumerate(payment_forms): forms_list.update( {unicode(forms_count + counter): payment_form}) self.form_list.update(forms_list)