def test_valid_str_data(self):
"""Test valid number data.
This is testing if str data is being parsed to number in json.
"""
response = responses.NumberResponse('12.3', 'postfix')
resp = json.loads(response.to_http_response().content.decode('utf-8'))
expected_resp = {'data': {'value': 12.3}, 'postfix': 'postfix'}
self.assertEqual(resp, expected_resp)
def test_invalid_data(self):
"""Test valid number data.
This is testing if str data is being parsed to number in json.
"""
with self.assertRaises(exceptions.ResponseException) as e:
responses.NumberResponse('invalid', 'postfix')
expected_args = ('Cannot parse answer.', {
'data': {
'value': 'invalid',
'postfix': 'postfix'
},
'errors': {
'value': ['Enter a number.']
}
})
self.assertEqual(e.exception.args, expected_args)
def number_endpoint(self, user):
"""Mock number endpoint."""
return responses.NumberResponse(3, 'postfix_number')
def test_valid_no_postfix(self):
"""Test valid data with no postfix."""
response = responses.NumberResponse(12.3)
resp = json.loads(response.to_http_response().content.decode('utf-8'))
expected_resp = {'data': {'value': 12.3}, 'postfix': ''}
self.assertEqual(resp, expected_resp)