def set_shipping_chises(self, request):
cart = Cart.objects.get_or_create_from_request(request)
for form in self.form_list.values():
if form == ChoiceShippingForm:
request.session['choice_data'] = get_shipping_methods(cart)
form().set_choices(self.generate_choices(
request.session['choice_data']))
def test_get_shipping_methods(self):
"""
Test that method return currect dict
"""
cart = model_factory(Cart)
expected_dict = {'flat': {'price': 20, 'name': 'Flat Shipping'}}
settings = {'flat': 'salest.payments.shipping_modules.flat'}
with patch('django.conf.settings.SHIPPING_MODULES', settings):
current_dict = get_shipping_methods(cart)
self.assertEqual(expected_dict, current_dict)
