def setUp(self): super().setUp() self.renderer = PaginatedCsvRenderer() self.data = { 'results': [ { 'string': 'ab,c', 'list': ['a', 'b', 'c'], 'dict': { 'a': 1, 'b': 2, 'c': 3 }, }, { 'string': 'def', 'string2': 'ghi', 'list': ['d', 'e', 'f', 'g'], 'dict': { 'd': 4, 'b': 5, 'c': 6 }, }, ] } self.context = {}
def setUp(self): super(PaginatedCsvRendererTests, self).setUp() self.renderer = PaginatedCsvRenderer() self.data = {'results': [ { 'string': 'ab,c', 'list': ['a', 'b', 'c'], 'dict': {'a': 1, 'b': 2, 'c': 3}, }, { 'string': 'def', 'string2': 'ghi', 'list': ['d', 'e', 'f', 'g'], 'dict': {'d': 4, 'b': 5, 'c': 6}, }, ]} self.context = {}
class PaginatedCsvRendererTests(TestCase): def setUp(self): super().setUp() self.renderer = PaginatedCsvRenderer() self.data = { 'results': [ { 'string': 'ab,c', 'list': ['a', 'b', 'c'], 'dict': { 'a': 1, 'b': 2, 'c': 3 }, }, { 'string': 'def', 'string2': 'ghi', 'list': ['d', 'e', 'f', 'g'], 'dict': { 'd': 4, 'b': 5, 'c': 6 }, }, ] } self.context = {} def set_request(self, params=None): request = MagicMock() mock_params = PropertyMock(return_value=params) type(request).query_params = mock_params self.context['request'] = request def test_csv_media_type(self): self.assertEqual(self.renderer.media_type, 'text/csv') def test_render(self): rendered_data = self.renderer.render(self.data, renderer_context=self.context) self.assertEqual( rendered_data, b'dict.a,dict.b,dict.c,dict.d,list,string,string2\r\n' b'1,2,3,,"a, b, c","ab,c",\r\n' b',5,6,4,"d, e, f, g",def,ghi\r\n') def test_render_fields(self): self.set_request( dict(fields='string2,invalid,dict.b,list,dict.a,string')) rendered_data = self.renderer.render(self.data, renderer_context=self.context) self.assertEqual( rendered_data, b'string2,dict.b,list,dict.a,string\r\n' b',2,"a, b, c",1,"ab,c"\r\n' b'ghi,5,"d, e, f, g",,def\r\n') def test_render_flatten_lists(self): self.renderer.concatenate_lists_sep = None rendered_data = self.renderer.render(self.data, renderer_context=self.context) self.assertEqual( rendered_data, b'dict.a,dict.b,dict.c,dict.d,list.0,list.1,list.2,list.3,string,string2\r\n' b'1,2,3,,a,b,c,,"ab,c",\r\n' b',5,6,4,d,e,f,g,def,ghi\r\n') def test_render_fields_flatten_lists(self): self.renderer.concatenate_lists_sep = None self.set_request( dict(fields='string2,invalid,list.2,dict.a,list.1,string')) rendered_data = self.renderer.render(self.data, renderer_context=self.context) self.assertEqual( rendered_data, b'string2,list.2,dict.a,list.1,string\r\n' b',c,1,b,"ab,c"\r\n' b'ghi,f,,e,def\r\n') def test_render_fields_limit_headers(self): self.renderer.header = ('string2', 'invalid', 'dict.a') self.set_request( dict(fields='string2,invalid,dict.b,list,dict.a,string')) rendered_data = self.renderer.render(self.data, renderer_context=self.context) self.assertEqual(rendered_data, b'string2,invalid,dict.a\r\n' b',,1\r\n' b'ghi,,\r\n')
class PaginatedCsvRendererTests(TestCase): def setUp(self): super(PaginatedCsvRendererTests, self).setUp() self.renderer = PaginatedCsvRenderer() self.data = {'results': [ { 'string': 'ab,c', 'list': ['a', 'b', 'c'], 'dict': {'a': 1, 'b': 2, 'c': 3}, }, { 'string': 'def', 'string2': 'ghi', 'list': ['d', 'e', 'f', 'g'], 'dict': {'d': 4, 'b': 5, 'c': 6}, }, ]} self.context = {} def set_request(self, params=None): request = MagicMock() mock_params = PropertyMock(return_value=params) type(request).query_params = mock_params self.context['request'] = request def test_csv_media_type(self): self.assertEqual(self.renderer.media_type, 'text/csv') def test_render(self): rendered_data = self.renderer.render(self.data, renderer_context=self.context) self.assertEquals(rendered_data, 'dict.a,dict.b,dict.c,dict.d,list,string,string2\r\n' '1,2,3,,"a, b, c","ab,c",\r\n' ',5,6,4,"d, e, f, g",def,ghi\r\n') def test_render_fields(self): self.set_request(dict(fields='string2,invalid,dict.b,list,dict.a,string')) rendered_data = self.renderer.render(self.data, renderer_context=self.context) self.assertEquals(rendered_data, 'string2,dict.b,list,dict.a,string\r\n' ',2,"a, b, c",1,"ab,c"\r\n' 'ghi,5,"d, e, f, g",,def\r\n') def test_render_flatten_lists(self): self.renderer.concatenate_lists_sep = None rendered_data = self.renderer.render(self.data, renderer_context=self.context) self.assertEquals(rendered_data, 'dict.a,dict.b,dict.c,dict.d,list.0,list.1,list.2,list.3,string,string2\r\n' '1,2,3,,a,b,c,,"ab,c",\r\n' ',5,6,4,d,e,f,g,def,ghi\r\n') def test_render_fields_flatten_lists(self): self.renderer.concatenate_lists_sep = None self.set_request(dict(fields='string2,invalid,list.2,dict.a,list.1,string')) rendered_data = self.renderer.render(self.data, renderer_context=self.context) self.assertEquals(rendered_data, 'string2,list.2,dict.a,list.1,string\r\n' ',c,1,b,"ab,c"\r\n' 'ghi,f,,e,def\r\n') def test_render_fields_limit_headers(self): self.renderer.header = ('string2', 'invalid', 'dict.a') self.set_request(dict(fields='string2,invalid,dict.b,list,dict.a,string')) rendered_data = self.renderer.render(self.data, renderer_context=self.context) self.assertEquals(rendered_data, 'string2,invalid,dict.a\r\n' ',,1\r\n' 'ghi,,\r\n')