Python XHeaderの例

コード例 #1

ファイル: test_common.py プロジェクト: pjcunningham/aiospamc

def test_headers_items():
    headers = [XHeader(name='A', value='a'), XHeader(name='B', value='b')]
    h = SpamcHeaders(headers=headers)
    header_tuples = [(header.field_name(), header) for header in headers]

    for test_result in h.items():
        assert test_result in header_tuples

コード例 #2

ファイル: test_common.py プロジェクト: pjcunningham/aiospamc

def test_headers_keys():
    headers = [XHeader(name='A', value='a'), XHeader(name='B', value='b')]
    h = SpamcHeaders(headers=headers)
    header_fields = [header.field_name() for header in headers]

    for test_result in h.keys():
        assert test_result in header_fields

コード例 #3

ファイル: test_common.py プロジェクト: pjcunningham/aiospamc

def test_headers_bytes():
    headers = [XHeader(name='A', value='a'), XHeader(name='B', value='b')]
    h = SpamcHeaders(headers=headers)
    result = bytes(h)

    header_bytes = [bytes(header) for header in headers]
    for header in header_bytes:
        assert header in result

コード例 #4

ファイル: test_xheader.py プロジェクト: pjcunningham/aiospamc

def test_bytes():
    x_header = XHeader('XHeader', 'head-value')

    assert bytes(x_header) == b'XHeader: head-value\r\n'

コード例 #5

ファイル: test_xheader.py プロジェクト: pjcunningham/aiospamc

def test_field_name():
    x_header = XHeader('head-name', 'head-value')

    assert x_header.field_name() == 'head-name'

コード例 #6

ファイル: test_xheader.py プロジェクト: pjcunningham/aiospamc

def test_user_value():
    x_header = XHeader('head-name', 'head-value')

    assert x_header.name == 'head-name'
    assert x_header.value == 'head-value'

コード例 #7

ファイル: test_xheader.py プロジェクト: pjcunningham/aiospamc

def test_repr():
    x_header = XHeader('head-name', 'head-value')

    assert repr(
        x_header) == 'XHeader(name=\'head-name\', value=\'head-value\')'

コード例 #8

ファイル: test_common.py プロジェクト: pjcunningham/aiospamc

def test_headers_values():
    headers = [XHeader(name='A', value='a'), XHeader(name='B', value='b')]
    h = SpamcHeaders(headers=headers)

    for test_result in h.values():
        assert test_result in headers

コード例 #9

ファイル: test_headers.py プロジェクト: wevsty/aiospamc

 def test_instantiates(self):
     x_header = XHeader('head-name', 'head-value')
     assert 'x_header' in locals()

コード例 #10

ファイル: conftest.py プロジェクト: pjcunningham/aiospamc

def x_headers():
    from aiospamc.headers import XHeader

    return [XHeader(name='A', value='a'), XHeader(name='B', value='b')]

コード例 #11

import pytest

from aiospamc.headers import Compress, ContentLength, XHeader
from aiospamc.requests import Request


def test_request_instantiates():
    request = Request('TEST')

    assert 'request' in locals()


@pytest.mark.parametrize(
    'verb,body,headers',
    [('TEST', None, []),
     ('TEST', None, [XHeader('X-Tests-Head', 'Tests value')]),
     ('TEST', 'Test body\n', [ContentLength(length=10)]),
     ('TEST', 'Test body\n', [ContentLength(length=10),
                              Compress()])])
def test_request_bytes(verb, body, headers):
    request = Request(verb=verb, body=body, headers=headers)

    assert bytes(request).startswith(verb.encode())
    assert bytes(b'SPAMC/1.5\r\n') in bytes(request)
    assert all(bytes(header) in bytes(request) for header in headers)
    if body:
        if any(isinstance(header, Compress) for header in headers):
            assert bytes(request).endswith(zlib.compress(body.encode()))
        else:
            assert bytes(request).endswith(body.encode())