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
|
# coding=utf-8
""" test types
"""
import pytest
from pgpy.types import PGPObject
text = {
# some basic utf-8 test strings - these should all pass
'english': u'The quick brown fox jumped over the lazy dog',
# this hiragana pangram comes from http://www.columbia.edu/~fdc/utf8/
'hiragana': u'いろはにほへど ちりぬるを\n'
u'わがよたれぞ つねならむ\n'
u'うゐのおくやま けふこえて\n'
u'あさきゆめみじ ゑひもせず',
'poo': u'Hello, \U0001F4A9!',
}
# some alternate encodings to try
# these should fail
encoded_text = {
# try some alternate encodings as well
# 'crunch the granite of science'
'cyrillic': u'грызть гранит науки'.encode('iso8859_5'),
# 'My hovercraft is full of eels'
'cp865': u'Mit luftpudefartøj er fyldt med ål'.encode('cp865'),
}
# test harness for pgpy.types.PGPObject, since it defines a couple of abstract methods
class FakePGPObject(PGPObject):
@classmethod
def new(cls, text):
obj = FakePGPObject()
obj.data = cls.text_to_bytes(text)
return obj
def __init__(self):
self.data = bytearray()
def __bytearray__(self):
return bytearray(b'_fake_') + self.data
def parse(self, packet):
self.data = packet
class TestPGPObject(object):
@pytest.mark.regression(issue=154)
@pytest.mark.parametrize('text', [v for _, v in sorted(text.items())], ids=sorted(text.keys()))
def test_text_to_bytes(self, text):
pgpo = FakePGPObject.new(text)
assert pgpo.__bytearray__() == bytearray(b'_fake_') + bytearray(text, 'utf-8')
@pytest.mark.regression(issue=154)
@pytest.mark.parametrize('encoded_text', [v for _, v in sorted(encoded_text.items())], ids=sorted(encoded_text.keys()))
def test_text_to_bytes_encodings(self, encoded_text):
pgpo = FakePGPObject.new(encoded_text)
# this should fail
with pytest.raises(UnicodeDecodeError):
pgpo.data.decode('utf-8')
def test_text_to_bytes_none(self):
assert PGPObject.text_to_bytes(None) is None
def test_bytes_to_text_none(self):
assert PGPObject.bytes_to_text(None) is None
def test_bytes_to_text_text(self):
assert PGPObject.bytes_to_text('asdf') == 'asdf'
|
