# Part of Odoo. See LICENSE file for full copyright and licensing details. import json from uuid import uuid4 from werkzeug import urls from odoo import Command from odoo.http import root from odoo.tests import tagged from odoo.addons.base.tests.common import HttpCaseWithUserDemo from odoo.addons.website_sale.controllers.main import WebsiteSale as WebsiteSaleController @tagged('at_install') class TestWebsiteSaleExpressCheckoutFlows(HttpCaseWithUserDemo): """ The goal of this method class is to test the address management on express checkout. """ @classmethod def setUpClass(cls): super().setUpClass() cls.website = cls.env.ref('website.default_website') cls.country_id = cls.env.ref('base.be').id cls.sale_order = cls.env['sale.order'].create({ 'partner_id': cls.website.user_id.partner_id.id, 'website_id': cls.website.id, 'order_line': [Command.create({ 'product_id': cls.env['product.product'].create({ 'name': 'Product A', 'list_price': 100, 'website_published': True, 'sale_ok': True}).id, 'name': 'Product A', })] }) cls.express_checkout_billing_values = { 'name': 'Express Checkout Partner', 'email': 'express@check.out', 'phone': '0000000000', 'street': 'ooo', 'street2': 'ppp', 'city': 'ooo', 'zip': '1200', 'country': 'US', 'state': 'AL', } def test_express_checkout_public_user(self): """ Test that when using express checkout as a public user, a new partner is created. """ session = self.authenticate(None, None) session['sale_order_id'] = self.sale_order.id root.session_store.save(session) self.make_jsonrpc_request( urls.url_join( self.base_url(), WebsiteSaleController._express_checkout_route ), params={ 'billing_address': dict(self.express_checkout_billing_values) } ) new_partner = self.sale_order.partner_id self.assertNotEqual(new_partner, self.website.user_id.partner_id) for k in self.express_checkout_billing_values: if k in ['state', 'country']: # State and country are stored as ids in `new_partner` and therefore cannot be # compared. continue self.assertEqual(new_partner[k], self.express_checkout_billing_values[k]) def test_express_checkout_registered_user(self): """ Test that when you use express checkout as a registered user and the address sent by the express checkout form exactly matches the one registered in odoo, we do not create a new partner and reuse the existing one. """ self.sale_order.partner_id = self.user_demo.partner_id.id session = self.authenticate(self.user_demo.login, self.user_demo.login) session['sale_order_id'] = self.sale_order.id root.session_store.save(session) self.make_jsonrpc_request( urls.url_join( self.base_url(), WebsiteSaleController._express_checkout_route ), params={ 'billing_address': { 'name': self.user_demo.partner_id.name, 'email': self.user_demo.partner_id.email, 'phone': self.user_demo.partner_id.phone, 'street': self.user_demo.partner_id.street, 'street2': self.user_demo.partner_id.street2, 'city': self.user_demo.partner_id.city, 'zip': self.user_demo.partner_id.zip, 'country': self.user_demo.partner_id.country_id.code, 'state': self.user_demo.partner_id.state_id.code, } } ) self.assertEqual(self.sale_order.partner_id.id, self.user_demo.partner_id.id) self.assertEqual(self.sale_order.partner_invoice_id.id, self.user_demo.partner_id.id) def test_express_checkout_registered_user_existing_address(self): """ Test that when you use the express checkout as a registered user and the address sent by the express checkout form exactly matches to one of the addresses linked to this user in odoo, we do not create a new partner and reuse the existing one. """ # Create a child partner for the demo partner child_partner_address = dict(self.express_checkout_billing_values) child_partner_country = self.env['res.country'].search([ ('code', '=', child_partner_address.pop('country')), ], limit=1) child_partner_state = self.env['res.country.state'].search([ ('code', '=', child_partner_address.pop('state')), ], limit=1) child_partner = self.env['res.partner'].create(dict( **child_partner_address, parent_id=self.user_demo.partner_id.id, type='invoice', country_id=child_partner_country.id, state_id=child_partner_state.id, )) self.sale_order.partner_id = self.user_demo.partner_id.id session = self.authenticate(self.user_demo.login, self.user_demo.login) session['sale_order_id'] = self.sale_order.id root.session_store.save(session) self.make_jsonrpc_request( urls.url_join( self.base_url(), WebsiteSaleController._express_checkout_route ), params={ 'billing_address': dict(self.express_checkout_billing_values) } ) self.assertEqual(self.sale_order.partner_id.id, self.user_demo.partner_id.id) self.assertEqual(self.sale_order.partner_invoice_id.id, child_partner.id) def test_express_checkout_registered_user_new_address(self): """ Test that when you use the express checkout as a registered user and the address sent by the express checkout form doesn't match to one of the addresses linked to this user in odoo, we create a new partner. """ self.sale_order.partner_id = self.user_demo.partner_id.id session = self.authenticate(self.user_demo.login, self.user_demo.login) session['sale_order_id'] = self.sale_order.id root.session_store.save(session) self.make_jsonrpc_request( urls.url_join( self.base_url(), WebsiteSaleController._express_checkout_route ), params={ 'billing_address': dict(self.express_checkout_billing_values) } ) self.assertEqual(self.sale_order.partner_id.id, self.user_demo.partner_id.id) new_partner = self.sale_order.partner_invoice_id self.assertNotEqual(new_partner, self.website.user_id.partner_id) for k in self.express_checkout_billing_values: if k in ['state', 'country']: # State and country are stored as ids in `new_partner` and therefore cannot be # compared. continue self.assertEqual(new_partner[k], self.express_checkout_billing_values[k])