File: test_controller.py

package info (click to toggle)
odoo 18.0.0%2Bdfsg-2
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 878,716 kB
  • sloc: javascript: 927,937; python: 685,670; xml: 388,524; sh: 1,033; sql: 415; makefile: 26
file content (207 lines) | stat: -rw-r--r-- 9,130 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207

# Part of Odoo. See LICENSE file for full copyright and licensing details.

import binascii
import json

import odoo.tests
from odoo.tests.common import HttpCase, new_test_user
from odoo.tools.json import scriptsafe as json_safe


@odoo.tests.tagged('-at_install', 'post_install')
class TestController(HttpCase):

    @classmethod
    def setUpClass(cls):
        super().setUpClass()
        portal_user = new_test_user(cls.env, login='portal_user', groups='base.group_portal')
        cls.portal_user = portal_user
        cls.portal = portal_user.login
        admin_user = new_test_user(cls.env, login='admin_user', groups='base.group_user,base.group_system')
        cls.admin = admin_user.login
        cls.headers = {"Content-Type": "application/json"}
        cls.pixel = 'R0lGODlhAQABAIAAAP///wAAACwAAAAAAQABAAACAkQBADs='

    def _build_payload(self, params=None):
        """
        Helper to properly build jsonrpc payload
        """
        return {
            "jsonrpc": "2.0",
            "method": "call",
            "id": 0,
            "params": params or {},
        }

    def test_01_upload_document(self):
        self.authenticate('admin', 'admin')
        # Upload document.
        response = self.url_open(
            '/html_editor/attachment/add_data',
            headers={'Content-Type': 'application/json'},
            data=json_safe.dumps({'params': {
                'name': 'test.txt',
                'data': 'SGVsbG8gd29ybGQ=',  # base64 Hello world
                'is_image': False,
            }})
        ).json()
        self.assertFalse('error' in response, 'Upload failed: %s' % response.get('error', {}).get('message'))
        attachment_id = response['result']['id']
        checksum = response['result']['checksum']
        # Download document and check content.
        response = self.url_open(
            '/web/content/%s?unique=%s&download=true' % (attachment_id, checksum)
        )
        self.assertEqual(200, response.status_code, 'Expect response')
        self.assertEqual(b'Hello world', response.content, 'Expect raw content')

        def test_02_illustration_shape(self):
            self.authenticate('admin', 'admin')
            # SVG with all replaceable colors.
            svg = b"""
    <svg viewBox="0 0 400 400">
      <rect width="300" height="300" style="fill:#3AADAA;" />
      <rect x="20" y="20" width="300" height="300" style="fill:#7C6576;" />
      <rect x="40" y="40" width="300" height="300" style="fill:#F6F6F6;" />
      <rect x="60" y="60" width="300" height="300" style="fill:#FFFFFF;" />
      <rect x="80" y="80" width="300" height="300" style="fill:#383E45;" />
    </svg>
            """
            attachment = self.env['ir.attachment'].create({
                'name': 'test.svg',
                'mimetype': 'image/svg+xml',
                'datas': binascii.b2a_base64(svg, newline=False),
                'public': True,
                'res_model': 'ir.ui.view',
                'res_id': 0,
            })
            # Shape illustration with slug.
            slug = self.env['ir.http']._slug
            url = '/html_editor/shape/illustration/%s' % slug(attachment)
            palette = 'c1=%233AADAA&c2=%237C6576&&c3=%23F6F6F6&&c4=%23FFFFFF&&c5=%23383E45'
            attachment['url'] = '%s?%s' % (url, palette)

            response = self.url_open(url)
            self.assertEqual(200, response.status_code, 'Expect response')
            self.assertEqual(svg, response.content, 'Expect unchanged SVG')

            response = self.url_open(url + '?c1=%23ABCDEF')
            self.assertEqual(200, response.status_code, 'Expect response')
            self.assertEqual(len(svg), len(response.content), 'Expect same length as original')
            self.assertTrue('ABCDEF' in str(response.content), 'Expect patched c1')
            self.assertTrue('3AADAA' not in str(response.content), 'Old c1 should not be there anymore')

            # Shape illustration without slug.
            url = '/html_editor/shape/illustration/noslug'
            attachment['url'] = url

            response = self.url_open(url)
            self.assertEqual(200, response.status_code, 'Expect response')
            self.assertEqual(svg, response.content, 'Expect unchanged SVG')

            response = self.url_open(url + '?c1=%23ABCDEF')
            self.assertEqual(200, response.status_code, 'Expect response')
            self.assertEqual(len(svg), len(response.content), 'Expect same length as original')
            self.assertTrue('ABCDEF' in str(response.content), 'Expect patched c1')
            self.assertTrue('3AADAA' not in str(response.content), 'Old c1 should not be there anymore')

    def test_03_get_image_info(self):
        gif_base64 = "R0lGODdhAQABAIAAAP///////ywAAAAAAQABAAACAkQBADs="
        self.authenticate('admin', 'admin')
        # Upload document.
        response = self.url_open(
            '/html_editor/attachment/add_data',
            headers={'Content-Type': 'application/json'},
            data=json_safe.dumps({'params': {
                'name': 'test.gif',
                'data': gif_base64,
                'is_image': True,
            }})
        )
        response = response.json()
        self.assertFalse('error' in response, 'Upload failed: %s' % response.get('error', {}).get('message'))
        attachment_id = response['result']['id']
        image_src = response['result']['image_src']
        mimetype = response['result']['mimetype']
        self.assertEqual('image/gif', mimetype, "Wrong mimetype")
        # Ensure image info can be retrieved.
        response = self.url_open('/html_editor/get_image_info',
                                 headers={'Content-Type': 'application/json'},
                                 data=json_safe.dumps({
                                     "params": {
                                         "src": image_src,
                                     }
                                 }),
                                 )
        response = response.json()
        self.assertEqual(attachment_id, response['result']['original']['id'], "Wrong id")
        self.assertEqual(image_src, response['result']['original']['image_src'], "Wrong image_src")
        self.assertEqual(mimetype, response['result']['original']['mimetype'], "Wrong mimetype")

    def test_04_admin_attachment(self):
        self.authenticate(self.admin, self.admin)
        payload = self._build_payload({"name": "pixel", "data": self.pixel, "is_image": True})
        response = self.url_open('/html_editor/attachment/add_data', data=json.dumps(payload), headers=self.headers)
        self.assertEqual(200, response.status_code)
        attachment = self.env['ir.attachment'].search([('name', '=', 'pixel')])
        self.assertTrue(attachment)

        domain = [('name', '=', 'pixel')]
        result = attachment.search(domain)
        self.assertTrue(len(result), "No attachment fetched")
        self.assertEqual(result, attachment)

    def test_05_internal_link_preview(self):
        self.authenticate(self.admin, self.admin)
        # retrieve metadata of an record without customerized link_preview_name but with display_name
        response_without_preview_name = self.url_open(
            '/html_editor/link_preview_internal',
            data=json_safe.dumps({
                "params": {
                    "preview_url": f"/odoo/users/{self.portal_user.id}",
                }
            }),
            headers=self.headers
        )
        self.assertEqual(200, response_without_preview_name.status_code)
        self.assertTrue('display_name' in response_without_preview_name.text)

        # retrieve metadata of a url with wrong action name
        response_wrong_action = self.url_open(
            '/html_editor/link_preview_internal',
            data=json_safe.dumps({
                "params": {
                    "preview_url": "/odoo/actionInvalid/1",
                }
            }),
            headers=self.headers
        )
        self.assertEqual(200, response_wrong_action.status_code)
        self.assertTrue('error_msg' in response_wrong_action.text)

        # retrieve metadata of a url with wrong record id
        response_wrong_record = self.url_open(
            '/html_editor/link_preview_internal',
            data=json_safe.dumps({
                "params": {
                    "preview_url": "/odoo/users/9999",
                }
            }),
            headers=self.headers
        )
        self.assertEqual(200, response_wrong_record.status_code)
        self.assertTrue('error_msg' in response_wrong_record.text)

        # retrieve metadata of a url not directing to a record
        response_not_record = self.url_open(
            '/html_editor/link_preview_internal',
            data=json_safe.dumps({
                "params": {
                    "preview_url": "/odoo/users",
                }
            }),
            headers=self.headers
        )
        self.assertEqual(200, response_not_record.status_code)
        self.assertTrue('other_error_msg' in response_not_record.text)