105 lines
4.8 KiB
Python
105 lines
4.8 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
||
|
|
import io
|
||
|
|
import logging
|
||
|
|
import unittest
|
||
|
|
import zipfile
|
||
|
|
from odoo.fields import Command
|
||
|
|
|
||
|
|
from odoo.tests.common import HttpCase, tagged
|
||
|
|
from base64 import b64decode
|
||
|
|
|
||
|
|
from odoo.tools import mute_logger
|
||
|
|
_logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
try:
|
||
|
|
import vobject
|
||
|
|
except ImportError:
|
||
|
|
_logger.warning("`vobject` Python module not found, vcard file generation disabled. Consider installing this module if you want to generate vcard files")
|
||
|
|
vobject = None
|
||
|
|
|
||
|
|
|
||
|
|
@tagged('-at_install', 'post_install')
|
||
|
|
class TestPartnerVCard(HttpCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
super().setUp()
|
||
|
|
|
||
|
|
if not vobject:
|
||
|
|
raise unittest.SkipTest("Skip tests when `vobject` Python module is not found.")
|
||
|
|
|
||
|
|
self.partners = self.env['res.partner'].create([{
|
||
|
|
'name': 'John Doe',
|
||
|
|
'email': 'john.doe@test.example.com',
|
||
|
|
'mobile': '+1 202 555 0888',
|
||
|
|
'phone': '+1 202 555 0122',
|
||
|
|
'function': 'Painter',
|
||
|
|
'street': 'Cookieville Minimum-Security Orphanarium',
|
||
|
|
'city': 'New York',
|
||
|
|
'country_id': self.env.ref('base.us').id,
|
||
|
|
'zip': '97648',
|
||
|
|
'website': 'https://test.exemple.com',
|
||
|
|
}, {
|
||
|
|
'name': 'shut',
|
||
|
|
'email': 'shut@test.example.com',
|
||
|
|
'mobile': '+1 202 555 0999',
|
||
|
|
'phone': '+1 202 555 0123',
|
||
|
|
'function': 'Developer',
|
||
|
|
'street': 'Donutville Maximum-Security Orphanarium',
|
||
|
|
'city': 'Washington DC',
|
||
|
|
'country_id': self.env.ref('base.us').id,
|
||
|
|
'zip': '97649',
|
||
|
|
'website': 'https://test.example.com',
|
||
|
|
}])
|
||
|
|
self.authenticate("admin", "admin")
|
||
|
|
|
||
|
|
def check_vcard_contents(self, vcard, partner):
|
||
|
|
self.assertEqual(vcard.contents["n"][0].value.family, partner.name, "Vcard should have the same name")
|
||
|
|
self.assertEqual(vcard.contents["adr"][0].value.street, partner.street, "Vcard should have the same street")
|
||
|
|
self.assertEqual(vcard.contents["adr"][0].value.city, partner.city, "Vcard should have the same city")
|
||
|
|
self.assertEqual(vcard.contents["adr"][0].value.code, partner.zip, "Vcard should have the same zip")
|
||
|
|
self.assertEqual(vcard.contents["adr"][0].value.country, self.env.ref('base.us').name, "Vcard should have the same country")
|
||
|
|
self.assertEqual(vcard.contents["email"][0].value, partner.email, "Vcard should have the same email")
|
||
|
|
self.assertEqual(vcard.contents["url"][0].value, partner.website, "Vcard should have the same website")
|
||
|
|
self.assertEqual(vcard.contents["tel"][0].params['TYPE'], ["work"], "Vcard should have the same phone")
|
||
|
|
self.assertEqual(vcard.contents["tel"][0].value, partner.phone, "Vcard should have the same phone")
|
||
|
|
self.assertEqual(vcard.contents["tel"][1].params['TYPE'], ["cell"], "Vcard should have the same mobile")
|
||
|
|
self.assertEqual(vcard.contents["tel"][1].value, partner.mobile, "Vcard should have the same mobile")
|
||
|
|
self.assertEqual(vcard.contents["title"][0].value, partner.function, "Vcard should have the same function")
|
||
|
|
self.assertEqual(len(vcard.contents['photo'][0].value), len(b64decode(partner.avatar_512)), "Vcard should have the same photo")
|
||
|
|
|
||
|
|
def test_fetch_single_partner_vcard(self):
|
||
|
|
res = self.url_open('/web_enterprise/partner/%d/vcard' % self.partners[0].id)
|
||
|
|
vcard = vobject.readOne(res.text)
|
||
|
|
self.check_vcard_contents(vcard, self.partners[0])
|
||
|
|
|
||
|
|
def test_fetch_multiple_partners_vcard(self):
|
||
|
|
res = self.url_open('/web/partner/vcard?partner_ids=%s,%s'
|
||
|
|
% (self.partners[0].id, self.partners[1].id))
|
||
|
|
with io.BytesIO(res.content) as buffer:
|
||
|
|
with zipfile.ZipFile(buffer, 'r') as zipf:
|
||
|
|
vcfFileList = zipf.namelist()
|
||
|
|
for i, vcfFile in enumerate(vcfFileList):
|
||
|
|
vcardFile = zipf.read(vcfFile).decode()
|
||
|
|
self.check_vcard_contents(vobject.readOne(vcardFile), self.partners[i])
|
||
|
|
|
||
|
|
@unittest.skip
|
||
|
|
def test_not_exist_partner_vcard(self):
|
||
|
|
partner_id = self.partner.id
|
||
|
|
self.partner.unlink()
|
||
|
|
res = self.url_open('/web/partner/%d/vcard' % partner_id)
|
||
|
|
self.assertEqual(res.status_code, 404)
|
||
|
|
|
||
|
|
def test_check_partner_access_for_user(self):
|
||
|
|
self.env['res.users'].create({
|
||
|
|
'groups_id': [Command.set([self.env.ref('base.group_public').id])],
|
||
|
|
'name': 'Test User',
|
||
|
|
'login': 'testuser',
|
||
|
|
'password': 'testuser',
|
||
|
|
})
|
||
|
|
self.authenticate('testuser', 'testuser')
|
||
|
|
with mute_logger('odoo.http'): # mute 403 warning
|
||
|
|
res = self.url_open('/web/partner/vcard?partner_ids=%s,%s' %
|
||
|
|
(self.partners[0].id, self.partners[1].id))
|
||
|
|
self.assertEqual(res.status_code, 403)
|