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'}