def setUp(self):
createFivePerson()
queryset = Person.objects.all()
columns = ['First Name', 'Last Name', 'Age']
self.instance = QuerysetToWorkbook(queryset,
columns,
filename='report')
class TestQuerysetToWorkbookMethods(TestCase):
def setUp(self):
createFivePerson()
queryset = Person.objects.all()
columns = ['First Name', 'Last Name', 'Age']
self.instance = QuerysetToWorkbook(queryset,
columns,
filename='report')
def tearDown(self):
if hasattr(self, 'zip_saved_file'):
os.remove(self.instance.get_filename(extension='zip'))
if hasattr(self, 'saved_file'):
os.remove(self.instance.get_filename())
def test_queryset_to_workbook(self):
workbook = self.instance.queryset_to_workbook()
rows = workbook.active.rows[0]
self.assertEqual(len(rows), 3)
def test_get_first(self):
self.instance.queryset_to_workbook()
first_row = self.instance.get_first()
self.assertEqual(first_row, 'A1:C1')
def test_set_colums(self):
dict_colums = {'test_column': 'Test Column', 'new_test': 'New Test'}
self.instance.columns = dict_colums.values()
self.assertEqual(self.instance.columns, dict_colums)
def test_get_data(self):
class LazyDict(object):
test_column = 'Test Column'
new_test = 'New Test',
age = 1
first_name = 'John'
last_name = 'Constantine'
data_cleaned = {
'age': 1,
'first_name': 'John',
'last_name': 'Constantine'
}
removed = self.instance.get_data(self.instance.columns, LazyDict())
self.assertEqual(removed, data_cleaned)
def test_get_filename(self):
file_name = self.instance.filename + '_' + timezone.now().strftime(
'%Y%m%d') + '.' + 'xlsx'
self.assertEqual(self.instance.get_filename(), file_name)
def test_get_content(self):
string_io = self.instance.get_content()
self.assertIsInstance(string_io, cStringIO.OutputType)
def test_get_compressed_file(self):
string_io = self.instance.get_compressed_file()
self.assertIsInstance(string_io, cStringIO.OutputType)
def test_compress_django_file(self):
django_file = self.instance.compress_django_file()
self.assertIsInstance(django_file, File)
def test_save_compress(self):
self.zip_saved_file = self.instance.save(compress=True)
self.assertTrue(
os.path.exists(self.instance.get_filename(extension='zip')))
def test_save_file(self):
self.saved_file = self.instance.save(compress=False)
self.assertTrue(os.path.exists(self.instance.get_filename()))
def test_response_compressed(self):
response = self.instance.response(compress=True)
self.assertIsInstance(response, HttpResponse)
self.assertTrue(response['Content-Disposition'])
def test_response(self):
response = self.instance.response(compress=False)
self.assertIsInstance(response, HttpResponse)
self.assertTrue(response['Content-Disposition'])
