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.

64 lines
2.2 KiB

7 years ago
7 years ago
7 years ago
7 years ago
  1. # -*- coding: utf-8 -*-
  2. ###################################################################################
  3. #
  4. # Copyright (C) 2017 MuK IT GmbH
  5. #
  6. # This program is free software: you can redistribute it and/or modify
  7. # it under the terms of the GNU Affero General Public License as
  8. # published by the Free Software Foundation, either version 3 of the
  9. # License, or (at your option) any later version.
  10. #
  11. # This program is distributed in the hope that it will be useful,
  12. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14. # GNU Affero General Public License for more details.
  15. #
  16. # You should have received a copy of the GNU Affero General Public License
  17. # along with this program. If not, see <http://www.gnu.org/licenses/>.
  18. #
  19. ###################################################################################
  20. import os
  21. import base64
  22. import unittest
  23. from contextlib import closing
  24. from odoo import _
  25. from odoo.tests import common
  26. _path = os.path.dirname(os.path.dirname(__file__))
  27. class AttachmentExtensionTestCase(common.TransactionCase):
  28. at_install = False
  29. post_install = True
  30. def setUp(self):
  31. super(AttachmentExtensionTestCase, self).setUp()
  32. self.attachment_model = self.env['ir.attachment'].sudo()
  33. def tearDown(self):
  34. super(AttachmentExtensionTestCase, self).tearDown()
  35. def test_attachment_extension_filename(self):
  36. with closing(open(os.path.join(_path, 'tests/data/sample.png'), 'rb')) as file:
  37. self.sample = self.attachment_model.create({
  38. 'name': "test",
  39. 'datas_fname': "sample.png",
  40. 'datas': base64.b64encode(file.read()),
  41. })
  42. self.assertEqual(self.sample.extension, ".png")
  43. def test_attachment_extension_mimetype(self):
  44. with closing(open(os.path.join(_path, 'tests/data/sample.png'), 'rb')) as file:
  45. self.sample = self.attachment_model.create({
  46. 'name': "test",
  47. 'mimetype': "image/png",
  48. 'datas': base64.b64encode(file.read()),
  49. })
  50. self.assertEqual(self.sample.extension, ".png")