def test_build_string_valueerror(self):
class MyExtension(object):
def __init__(self):
pass
def __str__(self):
raise ValueError('test')
ext = Extensions()
ext.extensions = {'ONE': 'OK VALUE', 'TWO': MyExtension()}
expected = """the header\r\nONE OK VALUE"""
self.assertEqual(expected, ext.build_string('the header'))
def test_parse_string(self):
ext = Extensions()
header = ext.parse_string("""\
the header
EXT1
PARSETEST DATA""")
self.assertEqual('the header', header)
self.assertTrue('EXT1' in ext)
self.assertTrue('PARSETEST' in ext)
self.assertEqual(None, ext.getparam('EXT1'))
self.assertEqual('DATA', ext.getparam('PARSETEST'))
def setUp(self):
self.ext = Extensions()
self.ext.extensions = {'EMPTY': None, 'TEST': 'STUFF'}
