def test_writing_excel(self):
""" This is a pretty crap test. It at least exercises the code.
If anyone wants to go through the brain damage of actually building
an 'expected' xlsx output and comparing it
(see https://github.com/jmcnamara/XlsxWriter/blob/master/xlsxwriter/test/helperfunctions.py for reference)
, by all means submit a pull request!
"""
res = QueryResult(
SimpleQueryFactory(
sql='select 1 as "a", 2 as ""',
title='\\/*[]:?this title is longer than 32 characters').sql,
connections[CONN])
res.execute_query()
res.process()
d = datetime.now()
d = timezone.make_aware(d, timezone.get_current_timezone())
res._data = [[1, None], [u"Jenét", d]]
res = ExcelExporter(
query=SimpleQueryFactory())._get_output(res).getvalue()
expected = b('PK')
self.assertEqual(res[:2], expected)
def test_writing_unicode(self):
res = QueryResult(SimpleQueryFactory(sql='select 1 as "a", 2 as ""').sql, connections[CONN])
res.execute_query()
res.process()
res._data = [[1, None], [u"Jenét", '1']]
res = CSVExporter(query=None)._get_output(res).getvalue()
self.assertEqual(res, 'a,\r\n1,\r\nJenét,1\r\n')
def test_writing_unicode(self):
res = QueryResult(SimpleQueryFactory(sql='select 1 as "a", 2 as ""').sql, connections[CONN])
res.execute_query()
res.process()
res._data = [[1, None], [u"Jenét", '1']]
res = CSVExporter(query=None)._get_output(res).getvalue()
self.assertEqual(res, 'a,\r\n1,\r\nJenét,1\r\n')
def test_writing_datetimes(self):
res = QueryResult(SimpleQueryFactory(sql='select 1 as "a", 2 as "b"').sql, connections[CONN])
res.execute_query()
res.process()
res._data = [[1, date.today()]]
res = JSONExporter(query=None)._get_output(res).getvalue()
expected = [{'a': 1, 'b': date.today()}]
self.assertEqual(res, json.dumps(expected, cls=DjangoJSONEncoder))
def test_writing_json(self):
res = QueryResult(SimpleQueryFactory(sql='select 1 as "a", 2 as ""').sql, connections[CONN])
res.execute_query()
res.process()
res._data = [[1, None], [u"Jenét", '1']]
res = JSONExporter(query=None)._get_output(res).getvalue()
expected = [{'a': 1, '': None}, {'a': 'Jenét', '': '1'}]
self.assertEqual(res, json.dumps(expected))
def test_writing_json(self):
res = QueryResult(SimpleQueryFactory(sql='select 1 as "a", 2 as ""').sql)
res.execute_query()
res.process()
res._data = [[1, None], [u"Jenét", "1"]]
res = JSONExporter(query=None)._get_output(res).getvalue()
expected = [{"a": 1, "": None}, {"a": "Jenét", "": "1"}]
self.assertEqual(res, json.dumps(expected))
def test_writing_datetimes(self):
res = QueryResult(SimpleQueryFactory(sql='select 1 as "a", 2 as "b"').sql, connections[CONN])
res.execute_query()
res.process()
res._data = [[1, date.today()]]
res = JSONExporter(query=None)._get_output(res).getvalue()
expected = [{'a': 1, 'b': date.today()}]
self.assertEqual(res, json.dumps(expected, cls=DjangoJSONEncoder))
def test_writing_json(self):
res = QueryResult(SimpleQueryFactory(sql='select 1 as "a", 2 as ""').sql, connections[CONN])
res.execute_query()
res.process()
res._data = [[1, None], [u"Jenét", '1']]
res = JSONExporter(query=None)._get_output(res).getvalue()
expected = [{'a': 1, '': None}, {'a': 'Jenét', '': '1'}]
self.assertEqual(res, json.dumps(expected))
def test_writing_dict_fields(self):
res = QueryResult(SimpleQueryFactory(sql='select 1 as "a", 2 as ""',
title='\\/*[]:?this title is longer than 32 characters').sql, connections[CONN])
res.execute_query()
res.process()
res._data = [[1, ['foo', 'bar']], [2, {'foo': 'bar'}]]
res = ExcelExporter(query=SimpleQueryFactory())._get_output(res).getvalue()
expected = b('PK')
self.assertEqual(res[:2], expected)
def test_writing_dict_fields(self):
res = QueryResult(SimpleQueryFactory(sql='select 1 as "a", 2 as ""',
title='\\/*[]:?this title is longer than 32 characters').sql, connections[CONN])
res.execute_query()
res.process()
res._data = [[1, ['foo', 'bar']], [2, {'foo': 'bar'}]]
res = ExcelExporter(query=SimpleQueryFactory())._get_output(res).getvalue()
expected = b('PK')
self.assertEqual(res[:2], expected)
def test_writing_excel_with_invalid_characters_in_worksheet_name(self):
res = QueryResult(
SimpleQueryFactory(
sql='select 1 as "a", 2 as ""',
title='[]:*?/\\this title is longer than 32 characters').sql,
connections[CONN])
res.execute_query()
res.process()
d = datetime.now()
d = timezone.make_aware(d, timezone.get_current_timezone())
res._data = [[1, None], [u"Jenét", d]]
res = ExcelExporter(
query=SimpleQueryFactory())._get_output(res).getvalue()
expected = b('PK')
self.assertEqual(res[:2], expected)
def test_writing_pdf(self):
""" Use same logic as with excel
"""
res = QueryResult(
SimpleQueryFactory(
sql='select 1 as "a", 2 as ""',
title='this title is longer than 32 characters').sql)
res.execute_query()
res.process()
d = datetime.now()
d = timezone.make_aware(d, timezone.get_current_timezone())
res._data = [[1, None], ["Jenét", d]]
res = PdfExporter(
query=SimpleQueryFactory())._get_output(res).getvalue()
expected = b('%PDF')
self.assertEqual(res[:4], expected)
def test_writing_excel(self):
""" This is a pretty crap test. It at least exercises the code.
If anyone wants to go through the brain damage of actually building
an 'expected' xlsx output and comparing it
(see https://github.com/jmcnamara/XlsxWriter/blob/master/xlsxwriter/test/helperfunctions.py for reference)
, by all means submit a pull request!
"""
res = QueryResult(SimpleQueryFactory(sql='select 1 as "a", 2 as ""',
title='this title is longer than 32 characters').sql)
res.execute_query()
res.process()
d = datetime.now()
d = timezone.make_aware(d, timezone.get_current_timezone())
res._data = [[1, None], [u"Jenét", d]]
res = ExcelExporter(query=SimpleQueryFactory())._get_output(res).getvalue()
expected = b('PK')
self.assertEqual(res[:2], expected)
