def test_setting_headers(self): m = Message() m.set_header('test-header', 'test-value') self.assertEqual('test-value', m.get_header('test-header'))
def test_default_returned_for_missing_headers(self): m = Message() self.assertIsNone(m.get_header('missing')) self.assertEqual('default', m.get_header('missing', 'default'))