def test_multiple_items(self):
j = JsonBuilder()
j.put({'a': 'b'})
j.put({'c': 'd'})
self.assertJsonEqual(j.to_dict(), {'a': 'b', 'c': 'd'})
j.put({'a': 'x'}, {'e': 'f'})
self.assertJsonEqual(j.to_dict(), {'a': 'x', 'c': 'd', 'e': 'f'})
j.put(e='x')
self.assertJsonEqual(j.to_dict(), {'a': 'x', 'c': 'd', 'e': 'x'})
def test_multiple_items(self):
j = JsonBuilder()
j.put({'a': 'b'})
j.put({'c': 'd'})
self.assertEqual(j.to_dict(), {'a': 'b', 'c': 'd'})
j.put({'a': 'x'}, {'e': 'f'})
self.assertEqual(j.to_dict(), {'a': 'x', 'c': 'd', 'e': 'f'})
j.put(e='x')
self.assertEqual(j.to_dict(), {'a': 'x', 'c': 'd', 'e': 'x'})
def test_nested_future_error_node(self):
j = JsonBuilder()
f1 = Future()
f2 = Future()
f1.set_result({'nested': f2})
j.put(f1)
self.assertEqual(j.to_string(), """{"nested": null}""")
result = RequestResult()
result.set_exception(
FailedRequestException(reason='error', code='code'))
f2.set_result({'a': result})
self.assertEqual(
j.to_dict(),
{'nested': {
'a': {
'error': {
'reason': 'error',
'code': 'code'
}
}
}})
def test_failed_future(self):
j = JsonBuilder()
f = Future()
result = RequestResult()
result.set_exception(FailedRequestException(reason='error', code='code'))
f.set_result(result)
j.put(f)
self.assertJsonEqual(j.to_dict(), {'error': {'reason': 'error', 'code': 'code'}})
def test_failed_future(self):
j = JsonBuilder()
f = Future()
result = TestDoc.get_test_request_result()
result._data_parse_error = DataParseError(reason='error', code='code')
f.set_result(result)
j.put(f)
self.assertEqual(j.to_dict(), {'error': {'reason': 'error', 'code': 'code'}})
def test_nested_json_builder(self):
j1 = JsonBuilder()
j1.put(k1='v1')
j2 = JsonBuilder()
j2.put(k2='v2')
j1.put(j2)
self.assertEqual(j1.to_dict(), {'k2': 'v2', 'k1': 'v1'})
def test_future_string_value(self):
j = JsonBuilder()
f = Future()
result = TestDoc.get_test_request_result()
result._content_type = 'xml'
result._data = '<test>test</test>'
f.set_result(result)
j.put(f)
self.assertEqual(j.to_dict(), {})
def test_future_string_value(self):
j = JsonBuilder()
f = Future()
result = TestDoc.get_test_request_result()
result._content_type = 'xml'
result._data = '<test>test</test>'
f.set_result(result)
j.put(f)
self.assertEqual(j.to_dict(), {})
def test_nested_json_builder(self):
j1 = JsonBuilder()
j1.put(k1='v1')
j2 = JsonBuilder()
j2.put(k2='v2')
j1.put(j2)
self.assertJsonEqual(
j1.to_dict(), {'k2': 'v2', 'k1': 'v1'}
)
def test_future(self):
j = JsonBuilder()
f = Future()
j.put(f)
self.assertFalse(j.is_empty())
self.assertEqual(j.to_string(), """{}""")
f.set_result({'a': 'b'})
self.assertEqual(j.to_dict()['a'], 'b')
self.assertEqual(j.to_string(), """{"a": "b"}""")
def test_future(self):
j = JsonBuilder()
f = Future()
j.put(f)
self.assertFalse(j.is_empty())
self.assertEqual(j.to_string(), """{}""")
f.set_result({'a': 'b'})
self.assertEqual(j.to_dict()['a'], 'b')
self.assertEqual(j.to_string(), """{"a": "b"}""")
def test_to_dict(self):
class Serializable(object):
def __init__(self, name, values):
self.name = name
self.values = values
def to_dict(self):
return {self.name: self.values}
j = JsonBuilder()
j.put(Serializable('some', ['test1', 'test2', 'test3']))
self.assertEqual(j.to_dict(), {'some': ['test1', 'test2', 'test3']})
def test_failed_future(self):
j = JsonBuilder()
f = Future()
result = TestDoc.get_test_request_result()
result._data_parse_error = DataParseError(reason='error', code='code')
f.set_result(result)
j.put(f)
self.assertEqual(j.to_dict(),
{'error': {
'reason': 'error',
'code': 'code'
}})
def test_failed_future(self):
j = JsonBuilder()
f = Future()
result = RequestResult()
result.set_exception(
FailedRequestException(reason='error', code='code'))
f.set_result(result)
j.put(f)
self.assertEqual(j.to_dict(),
{'error': {
'reason': 'error',
'code': 'code'
}})
def test_serializable(self):
class Serializable(object):
def __init__(self, name, values):
self.name = name
self.values = values
def to_dict(self):
return {self.name: self.values}
j = JsonBuilder()
j.put(Serializable('some', ['test1', 'test2', 'test3']))
self.assertJsonEqual(
j.to_dict(), {'some': ['test1', 'test2', 'test3']}
)
def test_nested_future_error_node(self):
j = JsonBuilder()
f1 = Future()
f2 = Future()
f1.set_result({'nested': f2})
j.put(f1)
self.assertEqual(j.to_string(), """{"nested": null}""")
result = TestDoc.get_test_request_result()
result._data_parse_error = DataParseError(reason='error', code='code')
f2.set_result(
{'a': result}
)
self.assertEqual(
j.to_dict(), {'nested': {'a': {'error': {'reason': 'error', 'code': 'code'}}}}
)
def test_nested_future_error_node(self):
j = JsonBuilder()
f1 = Future()
f2 = Future()
f1.set_result({'nested': f2})
j.put(f1)
self.assertEqual(j.to_string(), """{"nested": null}""")
result = RequestResult()
result.set_exception(FailedRequestException(reason='error', code='code'))
f2.set_result(
{'a': result}
)
self.assertJsonEqual(
j.to_dict(), {'nested': {'a': {'error': {'reason': 'error', 'code': 'code'}}}}
)
def test_nested_future_error_node(self):
j = JsonBuilder()
f1 = Future()
f2 = Future()
f1.set_result({'nested': f2})
j.put(f1)
self.assertEqual(j.to_string(), """{"nested": null}""")
result = TestDoc.get_test_request_result()
result._data_parse_error = DataParseError(reason='error', code='code')
f2.set_result({'a': result})
self.assertEqual(
j.to_dict(),
{'nested': {
'a': {
'error': {
'reason': 'error',
'code': 'code'
}
}
}})
def test_frozenset(self):
j = JsonBuilder()
j.put({'a': {'b': frozenset([1, 2, 3])}})
self.assertSetEqual(set(j.to_dict()['a']['b']), {1, 2, 3})
def test_frozenset(self):
j = JsonBuilder()
j.put({'a': {'b': frozenset([1, 2, 3])}})
self.assertSetEqual(set(j.to_dict()['a']['b']), {1, 2, 3})
