mirror of
https://github.com/Basilisk-Development-Team/UXP-mirror.git
synced 2026-11-05 17:42:42 -05:00
Jeremy Andrews
0c61ee391c
reduce is moved to functools, urllib reogranized. 2to3 . --fix urllib -wpn 2to3 . --fix reduce -wpn 2to3 . --fix import -wpn 2to3 . --fix imports -wpn 2to3 . --fix imports2 -wpn Ref: PEP 3108, PEP 3100 https://peps.python.org/pep-3108/ https://peps.python.org/pep-3100/
83 lines
2.0 KiB
Python
83 lines
2.0 KiB
Python
'''Tests varblock operations.'''
|
|
|
|
|
|
try:
|
|
from io import StringIO as BytesIO
|
|
except ImportError:
|
|
from io import BytesIO
|
|
import unittest
|
|
|
|
import rsa
|
|
from rsa._compat import b
|
|
from rsa import varblock
|
|
|
|
class VarintTest(unittest.TestCase):
|
|
|
|
def test_read_varint(self):
|
|
|
|
encoded = b('\xac\x02crummy')
|
|
infile = BytesIO(encoded)
|
|
|
|
(decoded, read) = varblock.read_varint(infile)
|
|
|
|
# Test the returned values
|
|
self.assertEqual(300, decoded)
|
|
self.assertEqual(2, read)
|
|
|
|
# The rest of the file should be untouched
|
|
self.assertEqual(b('crummy'), infile.read())
|
|
|
|
def test_read_zero(self):
|
|
|
|
encoded = b('\x00crummy')
|
|
infile = BytesIO(encoded)
|
|
|
|
(decoded, read) = varblock.read_varint(infile)
|
|
|
|
# Test the returned values
|
|
self.assertEqual(0, decoded)
|
|
self.assertEqual(1, read)
|
|
|
|
# The rest of the file should be untouched
|
|
self.assertEqual(b('crummy'), infile.read())
|
|
|
|
def test_write_varint(self):
|
|
|
|
expected = b('\xac\x02')
|
|
outfile = BytesIO()
|
|
|
|
written = varblock.write_varint(outfile, 300)
|
|
|
|
# Test the returned values
|
|
self.assertEqual(expected, outfile.getvalue())
|
|
self.assertEqual(2, written)
|
|
|
|
|
|
def test_write_zero(self):
|
|
|
|
outfile = BytesIO()
|
|
written = varblock.write_varint(outfile, 0)
|
|
|
|
# Test the returned values
|
|
self.assertEqual(b('\x00'), outfile.getvalue())
|
|
self.assertEqual(1, written)
|
|
|
|
|
|
class VarblockTest(unittest.TestCase):
|
|
|
|
def test_yield_varblock(self):
|
|
infile = BytesIO(b('\x01\x0512345\x06Sybren'))
|
|
|
|
varblocks = list(varblock.yield_varblocks(infile))
|
|
self.assertEqual([b('12345'), b('Sybren')], varblocks)
|
|
|
|
class FixedblockTest(unittest.TestCase):
|
|
|
|
def test_yield_fixedblock(self):
|
|
|
|
infile = BytesIO(b('123456Sybren'))
|
|
|
|
fixedblocks = list(varblock.yield_fixedblocks(infile, 6))
|
|
self.assertEqual([b('123456'), b('Sybren')], fixedblocks)
|
|
|