def test_system_exporter_spreadsheet_csv_cron_path_no_write_permission(
self):
"""test spreadsheet export via scheduled task to server file system"""
# get and modify main config
main_config_model = MainConfigModel.objects.get(
main_config_name='MainConfig')
main_config_model.cron_export_path = '/root'
main_config_model.cron_username = '******'
main_config_model.save()
# create spreadsheet without GET by directly calling the function
system_cron()
# login testuser
self.client.login(
username='******',
password='******',
)
# get response
response = self.client.get('/system/')
# get messages
messages = list(get_messages(response.wsgi_request))
# compare
self.assertEqual(str(response.context['user']),
'testuser_system_exporter_spreadsheet_csv')
self.assertEqual(
messages[0].message,
'[Scheduled task spreadsheet exporter] SYSTEM_CSV: No write permission for export path. Check config or file system!',
)
self.assertEqual(messages[0].level_tag, 'error')
# switch user context
self.client.logout()
self.client.login(username='******',
password='******')
# get response
response = self.client.get('/system/')
# get messages
messages = list(get_messages(response.wsgi_request))
# compare
self.assertEqual(str(response.context['user']), 'message_user')
self.assertEqual(
messages[0].message,
'[Scheduled task spreadsheet exporter] SYSTEM_CSV: No write permission for export path. Check config or file system!',
)
self.assertEqual(messages[0].level_tag, 'error')
def test_system_exporter_spreadsheet_csv_cron_complete_spreadsheet(self):
""" test spreadsheet export via scheduled task to server file system """
""" modify config section """
# get and modify main config
main_config_model = MainConfigModel.objects.get(
main_config_name='MainConfig')
main_config_model.cron_export_path = '/tmp'
main_config_model.cron_username = '******'
main_config_model.save()
# get and modify config to show all columns
system_exporter_spreadsheet_csv_config_model = SystemExporterSpreadsheetCsvConfigModel(
system_exporter_spreadsheet_csv_config_name=
'SystemExporterSpreadsheetCsvConfig')
system_exporter_spreadsheet_csv_config_model.spread_csv_system_id = True
system_exporter_spreadsheet_csv_config_model.spread_csv_dnsname = True
system_exporter_spreadsheet_csv_config_model.spread_csv_domain = True
system_exporter_spreadsheet_csv_config_model.spread_csv_systemstatus = True
system_exporter_spreadsheet_csv_config_model.spread_csv_analysisstatus = True
system_exporter_spreadsheet_csv_config_model.spread_csv_reason = True
system_exporter_spreadsheet_csv_config_model.spread_csv_recommendation = True
system_exporter_spreadsheet_csv_config_model.spread_csv_systemtype = True
system_exporter_spreadsheet_csv_config_model.spread_csv_ip = True
system_exporter_spreadsheet_csv_config_model.spread_csv_os = True
system_exporter_spreadsheet_csv_config_model.spread_csv_company = True
system_exporter_spreadsheet_csv_config_model.spread_csv_location = True
system_exporter_spreadsheet_csv_config_model.spread_csv_serviceprovider = True
system_exporter_spreadsheet_csv_config_model.spread_csv_tag = True
system_exporter_spreadsheet_csv_config_model.spread_csv_case = True
system_exporter_spreadsheet_csv_config_model.spread_csv_system_create_time = True
system_exporter_spreadsheet_csv_config_model.spread_csv_system_modify_time = True
system_exporter_spreadsheet_csv_config_model.save()
""" call view section """
# mock timezone.now()
t3_now = timezone.now()
with patch.object(timezone, 'now', return_value=t3_now):
# create spreadsheet without GET by directly calling the function
system_cron()
""" get file section """
# refresh config
main_config_model.refresh_from_db()
# get time for output file
filetime = t3_now.strftime('%Y%m%d_%H%M')
# prepare output file path
output_file_path = main_config_model.cron_export_path + '/' + filetime + '_systems.csv'
# open file from temp folder
csv_disk = open(output_file_path)
# open file as csv object
csv_reader = csv.reader(csv_disk, delimiter=',')
""" prepare objects section """
# get objects
system_1 = System.objects.get(system_name='system_1_all_attributes')
system_2 = System.objects.get(system_name='system_2_no_attributes')
""" compare values section """
# TODO: there must be a more convenient way to random access csv cells directly than iterating over lines and switch for line numbers
# TODO: like with 'xlrd' for xls files for example
# set counter
i = 1
# compare lines
for csv_line in csv_reader:
if csv_line:
if i == 1:
self.assertEqual(csv_line[0], 'ID')
self.assertEqual(csv_line[1], 'System')
self.assertEqual(csv_line[2], 'DNS name')
self.assertEqual(csv_line[3], 'Domain')
self.assertEqual(csv_line[4], 'Systemstatus')
self.assertEqual(csv_line[5], 'Analysisstatus')
self.assertEqual(csv_line[6], 'Reason')
self.assertEqual(csv_line[7], 'Recommendation')
self.assertEqual(csv_line[8], 'Systemtype')
self.assertEqual(csv_line[9], 'IP')
self.assertEqual(csv_line[10], 'OS')
self.assertEqual(csv_line[11], 'Company')
self.assertEqual(csv_line[12], 'Location')
self.assertEqual(csv_line[13], 'Serviceprovider')
self.assertEqual(csv_line[14], 'Tag')
self.assertEqual(csv_line[15], 'Case')
self.assertEqual(csv_line[16], 'Created')
self.assertEqual(csv_line[17], 'Modified')
elif i == 2:
self.assertEqual(csv_line[0], str(system_1.system_id))
self.assertEqual(csv_line[1], 'system_1_all_attributes')
self.assertEqual(csv_line[2], 'dnsname_1')
self.assertEqual(csv_line[3], 'domain_1')
self.assertEqual(csv_line[4], 'systemstatus_1')
self.assertEqual(csv_line[5], 'analysisstatus_1')
self.assertEqual(csv_line[6], 'reason_1')
self.assertEqual(csv_line[7], 'recommendation_1')
self.assertEqual(csv_line[8], 'systemtype_1')
self.assertEqual(csv_line[9],
'127.0.0.1,127.0.0.2,127.0.0.3')
self.assertEqual(csv_line[10], 'os_1')
self.assertEqual(csv_line[11],
'company_1,company_2,company_3')
self.assertEqual(csv_line[12], 'location_1')
self.assertEqual(csv_line[13], 'serviceprovider_1')
self.assertEqual(csv_line[14], 'tag_1,tag_2,tag_3')
self.assertEqual(csv_line[15], 'case_1,case_2,case_3')
self.assertEqual(csv_line[16], '2011-12-13 14:15')
self.assertEqual(csv_line[17], '2011-12-13 14:15')
elif i == 3:
self.assertEqual(csv_line[0], str(system_2.system_id))
self.assertEqual(csv_line[1], 'system_2_no_attributes')
self.assertEqual(csv_line[2], '')
self.assertEqual(csv_line[3], '')
self.assertEqual(csv_line[4], 'systemstatus_1')
self.assertEqual(csv_line[5], '')
self.assertEqual(csv_line[6], '')
self.assertEqual(csv_line[7], '')
self.assertEqual(csv_line[8], '')
self.assertEqual(csv_line[9], '')
self.assertEqual(csv_line[10], '')
self.assertEqual(csv_line[11], '')
self.assertEqual(csv_line[12], '')
self.assertEqual(csv_line[13], '')
self.assertEqual(csv_line[14], '')
self.assertEqual(csv_line[15], '')
self.assertEqual(csv_line[16], '2009-08-17 16:15')
self.assertEqual(csv_line[17], '2009-08-17 16:15')
elif i == 5:
self.assertEqual(csv_line[0], 'Created:')
self.assertEqual(csv_line[1],
t3_now.strftime('%Y-%m-%d %H:%M'))
elif i == 6:
self.assertEqual(csv_line[0], 'Created by:')
self.assertEqual(csv_line[1], 'cron')
# increase counter
i += 1
# compare number of rows (needs to be at the end because 'line_num' is some kind of pointer)
self.assertEqual(csv_reader.line_num, 6)
# close file
csv_disk.close()