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.

88 lines
3.2 KiB

7 years ago
7 years ago
7 years ago
7 years ago
7 years ago
7 years ago
  1. # -*- coding: utf-8 -*-
  2. ###################################################################################
  3. #
  4. # MuK Document Management System
  5. #
  6. # Copyright (C) 2018 MuK IT GmbH
  7. #
  8. # This program is free software: you can redistribute it and/or modify
  9. # it under the terms of the GNU Affero General Public License as
  10. # published by the Free Software Foundation, either version 3 of the
  11. # License, or (at your option) any later version.
  12. #
  13. # This program is distributed in the hope that it will be useful,
  14. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  16. # GNU Affero General Public License for more details.
  17. #
  18. # You should have received a copy of the GNU Affero General Public License
  19. # along with this program. If not, see <http://www.gnu.org/licenses/>.
  20. #
  21. ###################################################################################
  22. import os
  23. import json
  24. import base64
  25. import logging
  26. import unittest
  27. import requests
  28. import odoo
  29. from odoo import _
  30. from odoo.tests import common
  31. _logger = logging.getLogger(__name__)
  32. class AttachmentTestCase(common.HttpCase):
  33. at_install = False
  34. post_install = True
  35. def setUp(self):
  36. super(AttachmentTestCase, self).setUp()
  37. self.attachment = self.env['ir.attachment'].sudo()
  38. self.param = self.env['ir.config_parameter'].sudo()
  39. def tearDown(self):
  40. super(AttachmentTestCase, self).tearDown()
  41. def test_attachment(self):
  42. self.param.set_param('ir_attachment.location', 'lobject')
  43. attach = self.attachment.create({
  44. 'name': "Test",
  45. 'datas': base64.b64encode(b"\xff data")})
  46. self.assertTrue(attach.datas)
  47. self.assertTrue(attach.with_context({'bin_size': True}).datas)
  48. oid = attach.with_context({'oid': True}).store_lobject
  49. self.assertTrue(oid)
  50. attach.write({'datas': base64.b64encode(b"\xff data")})
  51. self.assertTrue(oid != attach.with_context({'oid': True}).store_lobject)
  52. self.assertTrue(attach.export_data(['datas']))
  53. self.assertTrue(attach.export_data(['datas'], raw_data=True))
  54. attach.unlink()
  55. def test_lobject(self):
  56. self.param.set_param('ir_attachment.location', 'lobject')
  57. attach = self.attachment.create({
  58. 'name': "Test",
  59. 'datas': base64.b64encode(b"\xff data")})
  60. human_size = attach.with_context({'human_size': True}).store_lobject
  61. self.assertTrue(human_size)
  62. stream = attach.with_context({'stream': True}).store_lobject
  63. self.assertTrue(stream.read())
  64. def test_download(self):
  65. self.param.set_param('ir_attachment.location', 'lobject')
  66. attach = self.attachment.create({
  67. 'name': "Test",
  68. 'datas': base64.b64encode(b"\xff data")})
  69. self.authenticate('admin', 'admin')
  70. url = "/web/lobject"
  71. params = {'id': attach.id}
  72. url_parts = list(urlparse(url))
  73. query = dict(parse_qsl(url_parts[4]))
  74. query.update(params)
  75. url_parts[4] = urlencode(query)
  76. url = urlunparse(url_parts)
  77. self.assertTrue(self.url_open(url))