# -*- coding: utf-8 -*- # (c) 2020 Matt Martz <matt@sivel.net> # GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt) from __future__ import absolute_import, division, print_function __metaclass__ = type import os from io import StringIO from email.message import Message import pytest from ansible.module_utils.urls import prepare_multipart def test_prepare_multipart(): fixture_boundary = b'===============3996062709511591449==' here = os.path.dirname(__file__) multipart = os.path.join(here, 'fixtures/multipart.txt') client_cert = os.path.join(here, 'fixtures/client.pem') client_key = os.path.join(here, 'fixtures/client.key') client_txt = os.path.join(here, 'fixtures/client.txt') fields = { 'form_field_1': 'form_value_1', 'form_field_2': { 'content': 'form_value_2', }, 'form_field_3': { 'content': '<html></html>', 'mime_type': 'text/html', }, 'form_field_4': { 'content': '{"foo": "bar"}', 'mime_type': 'application/json', }, 'file1': { 'content': 'file_content_1', 'filename': 'fake_file1.txt', }, 'file2': { 'content': '<html></html>', 'mime_type': 'text/html', 'filename': 'fake_file2.html', }, 'file3': { 'content': '{"foo": "bar"}', 'mime_type': 'application/json', 'filename': 'fake_file3.json', }, 'file4': { 'filename': client_cert, 'mime_type': 'text/plain', }, 'file5': { 'filename': client_key, 'mime_type': 'application/octet-stream' }, 'file6': { 'filename': client_txt, }, } content_type, b_data = prepare_multipart(fields) headers = Message() headers['Content-Type'] = content_type assert headers.get_content_type() == 'multipart/form-data' boundary = headers.get_boundary() assert boundary is not None with open(multipart, 'rb') as f: b_expected = f.read().replace(fixture_boundary, boundary.encode()) # Depending on Python version, there may or may not be a trailing newline assert b_data.rstrip(b'\r\n') == b_expected.rstrip(b'\r\n') def test_wrong_type(): pytest.raises(TypeError, prepare_multipart, 'foo') pytest.raises(TypeError, prepare_multipart, {'foo': None}) def test_empty(): pytest.raises(ValueError, prepare_multipart, {'foo': {}}) def test_unknown_mime(mocker): fields = {'foo': {'filename': 'foo.boom', 'content': 'foo'}} mocker.patch('mimetypes.guess_type', return_value=(None, None)) content_type, b_data = prepare_multipart(fields) assert b'Content-Type: application/octet-stream' in b_data def test_bad_mime(mocker): fields = {'foo': {'filename': 'foo.boom', 'content': 'foo'}} mocker.patch('mimetypes.guess_type', side_effect=TypeError) content_type, b_data = prepare_multipart(fields) assert b'Content-Type: application/octet-stream' in b_data