1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
from __future__ import annotations
import sys
import pytest
from kombu import compression
class test_compression:
def test_encoders__gzip(self):
assert 'application/x-gzip' in compression.encoders()
def test_encoders__bz2(self):
pytest.importorskip('bz2')
assert 'application/x-bz2' in compression.encoders()
def test_encoders__brotli(self):
pytest.importorskip('brotli')
assert 'application/x-brotli' in compression.encoders()
def test_encoders__lzma(self):
pytest.importorskip('lzma')
assert 'application/x-lzma' in compression.encoders()
def test_encoders__zstd(self):
pytest.importorskip('zstandard')
assert 'application/zstd' in compression.encoders()
def test_compress__decompress__zlib(self):
text = b'The Quick Brown Fox Jumps Over The Lazy Dog'
c, ctype = compression.compress(text, 'zlib')
assert text != c
d = compression.decompress(c, ctype)
assert d == text
def test_compress__decompress__bzip2(self):
text = b'The Brown Quick Fox Over The Lazy Dog Jumps'
c, ctype = compression.compress(text, 'bzip2')
assert text != c
d = compression.decompress(c, ctype)
assert d == text
def test_compress__decompress__brotli(self):
pytest.importorskip('brotli')
text = b'The Brown Quick Fox Over The Lazy Dog Jumps'
c, ctype = compression.compress(text, 'brotli')
assert text != c
d = compression.decompress(c, ctype)
assert d == text
def test_compress__decompress__lzma(self):
pytest.importorskip('lzma')
text = b'The Brown Quick Fox Over The Lazy Dog Jumps'
c, ctype = compression.compress(text, 'lzma')
assert text != c
d = compression.decompress(c, ctype)
assert d == text
def test_compress__decompress__zstd(self):
pytest.importorskip('zstandard')
text = b'The Brown Quick Fox Over The Lazy Dog Jumps'
c, ctype = compression.compress(text, 'zstd')
assert text != c
d = compression.decompress(c, ctype)
assert d == text
@pytest.mark.masked_modules('bz2')
def test_no_bz2(self, mask_modules):
c = sys.modules.pop('kombu.compression')
try:
import kombu.compression
assert not hasattr(kombu.compression, 'bz2')
finally:
if c is not None:
sys.modules['kombu.compression'] = c
@pytest.mark.masked_modules('lzma')
def test_no_lzma(self, mask_modules):
c = sys.modules.pop('kombu.compression')
try:
import kombu.compression
assert not hasattr(kombu.compression, 'lzma')
finally:
if c is not None:
sys.modules['kombu.compression'] = c
|
