You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
2.3 KiB
60 lines
2.3 KiB
# -*- coding: utf-8 -*-
|
|
# © 2016-TODAY LasLabs, Inc. [https://laslabs.com]
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
|
|
|
|
from openerp.tests.common import TransactionCase
|
|
from openerp.addons.base_user_gravatar.models.res_users import ResUsers
|
|
import mock
|
|
import hashlib
|
|
|
|
|
|
MODULE_LOCATION = 'openerp.addons.base_user_gravatar.models.res_users'
|
|
|
|
|
|
class TestResUsers(TransactionCase):
|
|
|
|
def setUp(self, *args, **kwargs):
|
|
super(TestResUsers, self).setUp()
|
|
self.model_obj = self.env['res.users']
|
|
self.partner_vals = {
|
|
'name': 'Test',
|
|
'email': 'test@example.com',
|
|
}
|
|
self.vals = {
|
|
'name': 'Test',
|
|
'login': 'test_login',
|
|
}
|
|
self.url = 'http://www.gravatar.com/avatar/{}?s=200'
|
|
|
|
def _test_record(self, ):
|
|
partner_id = self.env['res.partner'].create(self.partner_vals)
|
|
self.vals['partner_id'] = partner_id.id
|
|
return self.env['res.users'].create(self.vals)
|
|
|
|
@mock.patch('%s.urllib2' % MODULE_LOCATION)
|
|
def test_get_gravatar_base64_opens_correct_uri(self, mk, ):
|
|
""" Test that gravatar is pinged for image """
|
|
self.model_obj._get_gravatar_base64(self.partner_vals['email'])
|
|
expect = hashlib.md5(self.partner_vals['email']).hexdigest()
|
|
mk.urlopen.assert_called_once_with(self.url.format(expect))
|
|
|
|
@mock.patch('%s.base64' % MODULE_LOCATION)
|
|
@mock.patch('%s.urllib2' % MODULE_LOCATION)
|
|
def test_get_gravatar_base64_returns_encoded_image(self, mk, b64_mk, ):
|
|
""" Test that image result is read """
|
|
expect = 'Expect'
|
|
b64_mk.encodestring.return_value = expect
|
|
result = self.model_obj._get_gravatar_base64(
|
|
self.partner_vals['email']
|
|
)
|
|
self.assertEquals(expect, result)
|
|
|
|
def test_get_gravatar_image_writes_image(self, ):
|
|
""" Test that the resulting gravatar is written to user """
|
|
with mock.patch.object(ResUsers, 'write') as write_mk:
|
|
user_id = self._test_record()
|
|
with mock.patch.object(user_id, '_get_gravatar_base64') as mk:
|
|
expect = 'Expect'
|
|
mk.side_effect = ['Fail', expect]
|
|
user_id.get_gravatar_image()
|
|
write_mk.assert_called_once_with({'image': expect})
|