MockMessage test cleanup

This commit is contained in:
Jakub Warmuz 2015-03-18 22:08:01 +00:00
parent 77eb5f7625
commit 8208a7f4d5
No known key found for this signature in database
GPG key ID: 2A7BAD3A489B52EA

View file

@ -45,15 +45,8 @@ class MessageTest(unittest.TestCase):
'name': {'type': 'string'},
},
}
__slots__ = ('price', 'name')
@classmethod
def from_valid_json(cls, jobj):
return cls(price=jobj.get('price'), name=jobj.get('name'))
def _fields_to_json(self):
# pylint: disable=no-member
return {'price': self.price, 'name': self.name}
price = jose.Field('price')
name = jose.Field('name')
self.parent_cls = MockParentMessage
self.msg = MockMessage(price=123, name='foo')