partner-contact/partner_external_map/tests/test_partner_external_map.py

# -*- coding: utf-8 -*-
# © 2016 Pedro M. Baeza <pedro.baeza@tecnativa.com>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

from openerp.tests import common
from openerp.exceptions import Warning as UserError
from ..hooks import set_default_map_settings


class TestPartnerExternalMap(common.TransactionCase):
    def setUp(self):
        super(TestPartnerExternalMap, self).setUp()
        self.user = self.env['res.users'].create({
            'name': 'Test user',
            'login': 'test_login',
            'context_map_website_id': self.ref(
                'partner_external_map.google_maps'),
            'context_route_map_website_id': self.ref(
                'partner_external_map.google_maps'),
        })
        self.user.partner_id.city = 'Tomelloso'
        self.partner = self.env['res.partner'].create({
            'name': 'Test partner',
            'city': 'Madrid',
        })

    def test_post_init_hook(self):
        # Call this again for coverage purposes, but it has been already run
        set_default_map_settings(self.cr, self.registry)
        self.assertTrue(self.env.user.context_map_website_id)
        self.assertTrue(self.env.user.context_route_map_website_id)
        self.assertEqual(self.env.user.partner_id,
                         self.env.user.context_route_start_partner_id)

    def test_create_user(self):
        self.assertEqual(
            self.user.partner_id, self.user.context_route_start_partner_id)

    def test_open_map(self):
        action = self.partner.sudo(self.user.id).open_map()
        self.assertEqual(
            action['url'], "https://www.google.com/maps?ie=UTF8&q=Madrid")

    def test_open_route_map(self):
        action = self.partner.sudo(self.user.id).open_route_map()
        self.assertEqual(
            action['url'], "https://www.google.com/maps?saddr=Tomelloso&daddr="
                           "Madrid&directionsmode=driving")

    def test_open_map_with_coordinates(self):
        # Simulate that we have the base_geolocalize module installed creating
        # by hand the variables - This can't be done with routes
        partner = self.partner.sudo(self.user.id)
        partner.partner_latitude = 39.15837
        partner.partner_longitude = -3.02145
        action = partner.open_map()
        self.assertEqual(
            action['url'],
            "https://www.google.com/maps?z=15&q=39.15837,-3.02145")

    def test_exception_no_map_website(self):
        self.user.context_map_website_id = False
        with self.assertRaises(UserError):
            self.partner.sudo(self.user.id).open_map()

    def test_exception_no_map_route_website(self):
        self.user.context_route_start_partner_id = False
        with self.assertRaises(UserError):
            self.partner.sudo(self.user.id).open_route_map()

    def test_exception_no_starting_partner(self):
        self.user.context_route_map_website_id = False
        with self.assertRaises(UserError):
            self.partner.sudo(self.user.id).open_route_map()

    def test_exception_no_address_url(self):
        self.user.context_map_website_id.address_url = False
        with self.assertRaises(UserError):
            self.partner.sudo(self.user.id).open_map()

    def test_exception_no_route_address_url(self):
        self.user.context_map_website_id.route_address_url = False
        with self.assertRaises(UserError):
            self.partner.sudo(self.user.id).open_route_map()