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 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189
|
#
# This file is part of pyasn1 software.
#
# Copyright (c) 2005-2019, Ilya Etingof <etingof@gmail.com>
# License: http://snmplabs.com/pyasn1/license.html
#
import pickle
import sys
try:
import unittest2 as unittest
except ImportError:
import unittest
from tests.base import BaseTestCase
from pyasn1.type import char
from pyasn1.type import univ
from pyasn1.type import constraint
from pyasn1.error import PyAsn1Error
class AbstractStringTestCase(object):
initializer = ()
encoding = "us-ascii"
asn1Type = None
def setUp(self):
BaseTestCase.setUp(self)
self.asn1String = self.asn1Type(
bytes(self.initializer), encoding=self.encoding
)
self.pythonString = bytes(self.initializer).decode(self.encoding)
def testUnicode(self):
assert self.asn1String == self.pythonString, "unicode init fails"
def testLength(self):
assert len(self.asn1String) == len(self.pythonString), "unicode len() fails"
def testSizeConstraint(self):
asn1Spec = self.asn1Type(subtypeSpec=constraint.ValueSizeConstraint(1, 1))
try:
asn1Spec.clone(self.pythonString)
except PyAsn1Error:
pass
else:
assert False, "Size constraint tolerated"
try:
asn1Spec.clone(self.pythonString[0])
except PyAsn1Error:
assert False, "Size constraint failed"
def testSerialised(self):
assert bytes(self.asn1String) == self.pythonString.encode(
self.encoding
), "__str__() fails"
def testPrintable(self):
assert str(self.asn1String) == self.pythonString, "__str__() fails"
def testInit(self):
assert self.asn1Type(self.pythonString) == self.pythonString
assert (
self.asn1Type(self.pythonString.encode(self.encoding)) == self.pythonString
)
assert (
self.asn1Type(univ.OctetString(self.pythonString.encode(self.encoding)))
== self.pythonString
)
assert self.asn1Type(self.asn1Type(self.pythonString)) == self.pythonString
assert (
self.asn1Type(self.initializer, encoding=self.encoding) == self.pythonString
)
def testInitFromAsn1(self):
assert self.asn1Type(self.asn1Type(self.pythonString)) == self.pythonString
assert (
self.asn1Type(
univ.OctetString(
self.pythonString.encode(self.encoding), encoding=self.encoding
)
)
== self.pythonString
)
def testAsOctets(self):
assert self.asn1String.asOctets() == self.pythonString.encode(
self.encoding
), "testAsOctets() fails"
def testAsNumbers(self):
assert self.asn1String.asNumbers() == self.initializer, "testAsNumbers() fails"
def testSeq(self):
assert self.asn1String[0] == self.pythonString[0], "__getitem__() fails"
def testEmpty(self):
try:
str(self.asn1Type())
except PyAsn1Error:
pass
else:
assert 0, "Value operation on ASN1 type tolerated"
def testAdd(self):
assert (
self.asn1String + self.pythonString.encode(self.encoding)
== self.pythonString + self.pythonString
), "__add__() fails"
def testRadd(self):
assert (
self.pythonString.encode(self.encoding) + self.asn1String
== self.pythonString + self.pythonString
), "__radd__() fails"
def testMul(self):
assert self.asn1String * 2 == self.pythonString * 2, "__mul__() fails"
def testRmul(self):
assert 2 * self.asn1String == 2 * self.pythonString, "__rmul__() fails"
def testContains(self):
assert self.pythonString in self.asn1String
assert self.pythonString + self.pythonString not in self.asn1String
def testReverse(self):
assert list(reversed(self.asn1String)) == list(reversed(self.pythonString))
def testSchemaPickling(self):
old_asn1 = self.asn1Type()
serialised = pickle.dumps(old_asn1)
assert serialised
new_asn1 = pickle.loads(serialised)
assert type(new_asn1) == self.asn1Type
assert old_asn1.isSameTypeWith(new_asn1)
def testValuePickling(self):
old_asn1 = self.asn1String
serialised = pickle.dumps(old_asn1)
assert serialised
new_asn1 = pickle.loads(serialised)
assert new_asn1 == self.asn1String
class VisibleStringTestCase(AbstractStringTestCase, BaseTestCase):
initializer = (97, 102)
encoding = "us-ascii"
asn1Type = char.VisibleString
class GeneralStringTestCase(AbstractStringTestCase, BaseTestCase):
initializer = (169, 174)
encoding = "iso-8859-1"
asn1Type = char.GeneralString
class UTF8StringTestCase(AbstractStringTestCase, BaseTestCase):
initializer = (209, 132, 208, 176)
encoding = "utf-8"
asn1Type = char.UTF8String
class BMPStringTestCase(AbstractStringTestCase, BaseTestCase):
initializer = (4, 48, 4, 68)
encoding = "utf-16-be"
asn1Type = char.BMPString
class UniversalStringTestCase(AbstractStringTestCase, BaseTestCase):
initializer = (0, 0, 4, 48, 0, 0, 4, 68)
encoding = "utf-32-be"
asn1Type = char.UniversalString
suite = unittest.TestLoader().loadTestsFromModule(sys.modules[__name__])
if __name__ == "__main__":
unittest.TextTestRunner(verbosity=2).run(suite)
|