def test_marshall_generator_function_result(self):
def data():
yield self.values
yield self.values
expected = [self._get_entity() for _ in xrange(2)]
marshalled = marshall_result(data(), MockEntity)
self.assertEqual(list(marshalled), expected)
def test_marshall_result_keep_dict(self):
entity = MockEntity(**self.values)
tests = [
(self.values, self.values),
([1234, self.values, 'abcdef'], [1234, self.values, 'abcdef']),
((1234, self.values, 'abcdef'), (1234, self.values, 'abcdef')),
]
for data, expected in tests:
marshalled = marshall_result(data)
self.assertEqual(marshalled, expected)
def test_marshall_datetime_result(self):
py_dt = datetime(2014, 4, 14, 16, 27, 00)
xml_dt = DateTime('20140414T16:27:00+0200') # format used by Sklik
tests = [
(MockEntity, xml_dt, py_dt),
(MockEntity, [1234, xml_dt, 'abcdef'], [1234, py_dt, 'abcdef']),
(MockEntity, (1234, xml_dt, 'abcdef'), (1234, py_dt, 'abcdef')),
(MockEntity, [{'a': xml_dt}], [MockEntity(a=py_dt)]),
(None, {'X': [{'a': xml_dt}]}, {'X': [{'a': py_dt}]}),
]
for obj_type, data, expected in tests:
marshalled = marshall_result(data, obj_type)
self.assertEqual(marshalled, expected)
def test_marshall_generator_result(self):
data = (self.values for _ in xrange(2))
expected = [self._get_entity() for _ in xrange(2)]
marshalled = marshall_result(data, MockEntity)
self.assertEqual(list(marshalled), expected)
