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)