103 lines
3 KiB
Python
103 lines
3 KiB
Python
|
# -*- 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,
|
||
|
|
},
|
||
|
|
'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
|