def setUpTestData(cls): """ one-time setup """ # create user test_user = User.objects.create_user( username='******', is_staff=True, is_superuser=True, password='******', ) # change config set_csv_import_path( os.path.join( BASE_DIR, 'dfirtrack_main/tests/system_importer/system_importer_file_csv_files/' )) # change config set_csv_import_filename( 'system_importer_file_csv_testfile_01_minimal_double_quotation.csv' ) system_importer_file_csv_config_model = SystemImporterFileCsvConfigModel.objects.get( system_importer_file_csv_config_name='SystemImporterFileCsvConfig') system_importer_file_csv_config_model.csv_import_username = test_user system_importer_file_csv_config_model.save()
def test_system_importer_file_csv_messages_instant_single_system_skip( self): """ test importer view """ # create system create_system('system_csv_11_001') # change config set_csv_import_filename( 'system_importer_file_csv_testfile_11_messages_single_system.csv') # change config set_csv_skip_existing_system(True) # login testuser self.client.login( username='******', password='******') # get response response = self.client.get('/system/importer/file/csv/instant/', follow=True) # get messages messages = list(get_messages(response.wsgi_request)) # compare self.assertEqual(messages[0].message, '1 system was skipped.') self.assertEqual(messages[0].level_tag, 'success')
def test_system_importer_file_csv_complete_overwrite_instant(self): """ test importer view """ # compare before action - systems / attributes self = pre_compare_system_and_attributes_csv(self) # change config set_csv_import_filename('system_importer_file_csv_testfile_41_complete_overwrite.csv') # change config set_config_complete_attributes_csv() # change config set_config_complete_overwrite_csv() # login testuser self.client.login(username='******', password='******') # create url destination = urllib.parse.quote('/system/', safe='/') # get response response = self.client.get('/system/importer/file/csv/instant/', follow=True) # get messages messages = list(get_messages(response.wsgi_request)) # compare - meta self.assertRedirects(response, destination, status_code=302, target_status_code=200) # compare - messages self = compare_messages_csv(self, messages) # compare - systems / attributes self = compare_system_and_attributes_csv(self)
def test_system_importer_file_csv_check_attributes_instant_domain_name( self): """ test importer view """ # change config set_config_check_attributes_domain_name() # change config set_csv_import_filename( 'system_importer_file_csv_testfile_33_domain_name.csv') # login testuser self.client.login( username='******', password='******') # create url destination = urllib.parse.quote('/system/', safe='/') # get response response = self.client.get('/system/importer/file/csv/instant/', follow=True) # get messages messages = list(get_messages(response.wsgi_request)) # compare - meta self.assertRedirects(response, destination, status_code=302, target_status_code=200) # compare - messages self = compare_messages_domain_name(self, messages) # compare - systems / attributes self = compare_system_and_attributes_domain_name(self)
def test_system_importer_file_csv_check_content_file_system_instant_file_not_existing( self): """ test importer view """ # change config set_csv_import_path('/tmp') # change config set_csv_import_filename('filename_not_existing.abc') # login testuser self.client.login( username= '******', password='******') # create url destination = urllib.parse.quote('/system/', safe='/') # get response response = self.client.get('/system/importer/file/csv/instant/', follow=True) # get messages messages = list(get_messages(response.wsgi_request)) # compare self.assertRedirects(response, destination, status_code=302, target_status_code=200) self.assertEqual( messages[0].message, 'CSV import file does not exist. Check config or provide file!') self.assertEqual(messages[0].level_tag, 'error')
def test_system_importer_file_csv_messages_instant_many_systems_multiple(self): """ test importer view """ # create systems create_system('system_csv_12_001') create_system('system_csv_12_001') create_system('system_csv_12_002') create_system('system_csv_12_002') create_system('system_csv_12_003') create_system('system_csv_12_003') # change config set_csv_import_filename('system_importer_file_csv_testfile_12_messages_many_systems.csv') # change config set_csv_skip_existing_system(True) # login testuser self.client.login(username='******', password='******') # get response response = self.client.get('/system/importer/file/csv/instant/', follow=True) # get messages messages = list(get_messages(response.wsgi_request)) # compare self.assertEqual(messages[0].message, "3 systems were skipped because they existed several times. ['system_csv_12_001', 'system_csv_12_002', 'system_csv_12_003']") self.assertEqual(messages[0].level_tag, 'warning')
def test_system_importer_file_csv_status_instant_tagfree_status(self): """ test importer view """ # change config set_csv_import_filename( 'system_importer_file_csv_testfile_35_tagfree_status.csv') # change config set_config_tagfree_status() # login testuser self.client.login(username='******', password='******') # create url destination = urllib.parse.quote('/system/', safe='/') # get response response = self.client.get('/system/importer/file/csv/instant/', follow=True) # get messages messages = list(get_messages(response.wsgi_request)) # compare - meta self.assertRedirects(response, destination, status_code=302, target_status_code=200) # compare - messages self = compare_messages_tagfree_status(self, messages) # compare - systems / attributes self = compare_system_and_attributes_tagfree_status(self)
def test_system_importer_file_csv_check_content_file_system_instant_file_empty( self): """ test importer view """ # change config set_csv_import_path( os.path.join( BASE_DIR, 'dfirtrack_main/tests/system_importer/system_importer_file_csv_files/' )) # change config set_csv_import_filename( 'system_importer_file_csv_testfile_06_empty.csv') # login testuser self.client.login( username= '******', password='******') # create url destination = urllib.parse.quote('/system/', safe='/') # get response response = self.client.get('/system/importer/file/csv/instant/', follow=True) # get messages messages = list(get_messages(response.wsgi_request)) # compare self.assertRedirects(response, destination, status_code=302, target_status_code=200) self.assertEqual( messages[0].message, 'CSV import file is empty. Check config or file system!') self.assertEqual(messages[0].level_tag, 'error')
def test_system_importer_file_csv_cron_wrong_type(self): """ test importer view """ # change config set_csv_import_path(os.path.join(BASE_DIR, 'dfirtrack_main/tests/system_importer/system_importer_file_csv_files')) # change config set_csv_import_filename('system_importer_file_csv_testfile_04_wrong_type.png') # execute cron job / scheduled task 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_importer_file_csv_check_content_file_type') self.assertEqual(messages[0].message, '[Scheduled task CSV system importer] Wrong file type for CSV import. 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 CSV system importer] Wrong file type for CSV import. Check config or file system!') self.assertEqual(messages[0].level_tag, 'error')
def test_system_importer_file_csv_complete_attributes_database_instant( self): """ test importer view """ # change config set_config_complete_attributes_database() # change config set_csv_import_filename( 'system_importer_file_csv_testfile_08_complete_database.csv') # login testuser self.client.login( username= '******', password='******') # create url destination = urllib.parse.quote('/system/', safe='/') # get response response = self.client.get('/system/importer/file/csv/instant/', follow=True) # get messages messages = list(get_messages(response.wsgi_request)) # compare - meta self.assertRedirects(response, destination, status_code=302, target_status_code=200) # compare - messages self = compare_messages_database(self, messages) # compare - systems / attributes self = compare_system_and_attributes_database(self)
def test_system_importer_file_csv_instant_wrong_type(self): """ test importer view """ # change config set_csv_import_path( os.path.join( BASE_DIR, 'dfirtrack_main/tests/system_importer/system_importer_file_csv_files' )) # change config set_csv_import_filename( 'system_importer_file_csv_testfile_04_wrong_type.png') # login testuser self.client.login( username= '******', password='******') # create url destination = urllib.parse.quote('/system/', safe='/') # get response response = self.client.get('/system/importer/file/csv/instant/', follow=True) # get messages messages = list(get_messages(response.wsgi_request)) # compare self.assertRedirects(response, destination, status_code=302, target_status_code=200) self.assertEqual( messages[0].message, 'Wrong file type for CSV import. Check config or file system!') self.assertEqual(messages[0].level_tag, 'error')
def test_system_importer_file_csv_complete_preserve_cron(self): """test importer view""" # compare before action - systems / attributes self = pre_compare_system_and_attributes_csv(self) # change config set_csv_import_filename( 'system_importer_file_csv_testfile_42_complete_preserve.csv' ) # change config set_config_complete_attributes_csv() # change config set_config_complete_preserve_csv() # mock timezone.now() t_1 = datetime(2021, 3, 20, 18, 30, tzinfo=timezone.utc) with patch.object(timezone, 'now', return_value=t_1): # execute cron job / scheduled task 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 - user 1 self.assertEqual( str(response.context['user']), 'testuser_system_importer_file_csv_complete_preserve', ) self.assertEqual( messages[0].message, 'System CSV importer: created: 0 | updated: 3 | skipped: 0 | multiple: 0 [2021-03-20 18:30:00 - 2021-03-20 18:30:00]', ) self.assertEqual(messages[0].level_tag, 'success') # 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 - user 2 self.assertEqual(str(response.context['user']), 'message_user') self.assertEqual( messages[0].message, 'System CSV importer: created: 0 | updated: 3 | skipped: 0 | multiple: 0 [2021-03-20 18:30:00 - 2021-03-20 18:30:00]', ) self.assertEqual(messages[0].level_tag, 'success') # compare - systems / attributes self = compare_system_and_attributes_csv(self)
def test_system_importer_file_csv_complete_overwrite_cron(self): """test importer view""" # compare before action - systems / attributes self = pre_compare_system_and_attributes_csv(self) # change config set_csv_import_filename( 'system_importer_file_csv_testfile_41_complete_overwrite.csv') # change config set_config_complete_attributes_csv() # change config set_config_complete_overwrite_csv() # mock timezone.now() t_1 = datetime(2021, 3, 17, 19, 30, tzinfo=timezone.utc) with patch.object(timezone, 'now', return_value=t_1): # execute cron job / scheduled task 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 - user 1 self.assertEqual( str(response.context['user']), 'testuser_system_importer_file_csv_complete_overwrite', ) self.assertEqual( messages[0].message, 'System CSV importer: created: 0 | updated: 3 | skipped: 0 | multiple: 0 [2021-03-17 19:30:00 - 2021-03-17 19:30:00]', ) self.assertEqual(messages[0].level_tag, 'success') # 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 - user 2 self.assertEqual(str(response.context['user']), 'message_user') self.assertEqual( messages[0].message, 'System CSV importer: created: 0 | updated: 3 | skipped: 0 | multiple: 0 [2021-03-17 19:30:00 - 2021-03-17 19:30:00]', ) self.assertEqual(messages[0].level_tag, 'success') # compare - systems / attributes self = compare_system_and_attributes_csv(self)
def test_system_importer_file_csv_messages_cron_many_systems_skip(self): """test importer view""" # create systems create_system('system_csv_12_001') create_system('system_csv_12_002') create_system('system_csv_12_003') # change config set_csv_import_filename( 'system_importer_file_csv_testfile_12_messages_many_systems.csv' ) # change config set_csv_skip_existing_system(True) # mock timezone.now() t_7 = datetime(2021, 3, 7, 11, 35, tzinfo=timezone.utc) with patch.object(timezone, 'now', return_value=t_7): # execute cron job / scheduled task 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_importer_file_csv_messages' ) self.assertEqual( messages[0].message, 'System CSV importer: created: 0 | updated: 0 | skipped: 3 | multiple: 0 [2021-03-07 11:35:00 - 2021-03-07 11:35:00]', ) self.assertEqual(messages[0].level_tag, 'success') # 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, 'System CSV importer: created: 0 | updated: 0 | skipped: 3 | multiple: 0 [2021-03-07 11:35:00 - 2021-03-07 11:35:00]', ) self.assertEqual(messages[0].level_tag, 'success')
def test_system_importer_file_csv_check_attributes_cron_faulty_attributes( self): """test importer view""" # change config set_config_check_attributes_csv() # change config set_csv_import_filename( 'system_importer_file_csv_testfile_32_faulty_attributes.csv') # mock timezone.now() t_2 = datetime(2021, 3, 8, 18, 10, tzinfo=timezone.utc) with patch.object(timezone, 'now', return_value=t_2): # execute cron job / scheduled task 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 - user 1 self.assertEqual( str(response.context['user']), 'testuser_system_importer_file_csv_check_attributes', ) self.assertEqual( messages[0].message, 'System CSV importer: created: 4 | updated: 0 | skipped: 0 | multiple: 0 [2021-03-08 18:10:00 - 2021-03-08 18:10:00]', ) self.assertEqual(messages[0].level_tag, 'success') # 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 - user 2 self.assertEqual(str(response.context['user']), 'message_user') self.assertEqual( messages[0].message, 'System CSV importer: created: 4 | updated: 0 | skipped: 0 | multiple: 0 [2021-03-08 18:10:00 - 2021-03-08 18:10:00]', ) self.assertEqual(messages[0].level_tag, 'success') # compare - systems / attributes self = compare_system_and_attributes_csv(self)
def test_system_importer_file_csv_check_content_file_system_cron_file_no_read_permission( self, ): """test importer view""" # get timestamp string t1 = timezone.now().strftime('%Y%m%d_%H%M%S') # set file system attributes csv_import_path = '/tmp' csv_import_filename = f'{t1}_cron_no_read_permission.csv' # create file create_file_no_read_permission(csv_import_path, csv_import_filename) # change config set_csv_import_path(csv_import_path) # change config set_csv_import_filename(csv_import_filename) # execute cron job / scheduled task 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_importer_file_csv_check_content_file_system', ) self.assertEqual( messages[0].message, '[Scheduled task CSV system importer] No read permission for CSV import file. 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 CSV system importer] No read permission for CSV import file. Check config or file system!', ) self.assertEqual(messages[0].level_tag, 'error')
def test_system_importer_file_csv_complete_attributes_database_cron(self): """test importer view""" # change config set_config_complete_attributes_database() # change config set_csv_import_filename( 'system_importer_file_csv_testfile_08_complete_database.csv' ) # mock timezone.now() t_3 = datetime(2021, 3, 8, 18, 15, tzinfo=timezone.utc) with patch.object(timezone, 'now', return_value=t_3): # execute cron job / scheduled task 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 - user 1 self.assertEqual( str(response.context['user']), 'testuser_system_importer_file_csv_complete_attributes_database', ) self.assertEqual( messages[0].message, 'System CSV importer: created: 4 | updated: 0 | skipped: 0 | multiple: 0 [2021-03-08 18:15:00 - 2021-03-08 18:15:00]', ) self.assertEqual(messages[0].level_tag, 'success') # 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 - user 2 self.assertEqual(str(response.context['user']), 'message_user') self.assertEqual( messages[0].message, 'System CSV importer: created: 4 | updated: 0 | skipped: 0 | multiple: 0 [2021-03-08 18:15:00 - 2021-03-08 18:15:00]', ) self.assertEqual(messages[0].level_tag, 'success') # compare - systems / attributes self = compare_system_and_attributes_database(self)
def test_system_importer_file_csv_check_content_file_system_cron_file_empty( self): """test importer view""" # change config set_csv_import_path( os.path.join( BASE_DIR, 'dfirtrack_main/tests/system_importer/system_importer_file_csv_files/', )) # change config set_csv_import_filename( 'system_importer_file_csv_testfile_06_empty.csv') # execute cron job / scheduled task 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_importer_file_csv_check_content_file_system', ) self.assertEqual( messages[0].message, '[Scheduled task CSV system importer] CSV import file is empty. 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 CSV system importer] CSV import file is empty. Check config or file system!', ) self.assertEqual(messages[0].level_tag, 'error')
def test_system_importer_file_csv_complete_attributes_csv_cron(self): """ test importer view """ # change config set_csv_import_filename( 'system_importer_file_csv_testfile_07_complete_csv.csv') # change config set_config_complete_attributes_csv() # mock timezone.now() t_4 = datetime(2021, 3, 6, 18, 14, tzinfo=timezone.utc) with patch.object(timezone, 'now', return_value=t_4): # execute cron job / scheduled task 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 - user 1 self.assertEqual( str(response.context['user']), 'testuser_system_importer_file_csv_complete_attributes_csv') self.assertEqual( messages[0].message, 'System CSV importer: created: 3 | updated: 0 | skipped: 0 | multiple: 0 [2021-03-06 18:14:00 - 2021-03-06 18:14:00]' ) self.assertEqual(messages[0].level_tag, 'success') # 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 - user 2 self.assertEqual(str(response.context['user']), 'message_user') self.assertEqual( messages[0].message, 'System CSV importer: created: 3 | updated: 0 | skipped: 0 | multiple: 0 [2021-03-06 18:14:00 - 2021-03-06 18:14:00]' ) self.assertEqual(messages[0].level_tag, 'success') # compare - systems / attributes self = compare_system_and_attributes_csv(self)
def test_system_importer_file_csv_status_cron_tagfree_status(self): """ test importer view """ # change config set_csv_import_filename( 'system_importer_file_csv_testfile_35_tagfree_status.csv') # change config set_config_tagfree_status() # mock timezone.now() t_2 = datetime(2021, 3, 11, 19, 10, tzinfo=timezone.utc) with patch.object(timezone, 'now', return_value=t_2): # execute cron job / scheduled task 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 - user 1 self.assertEqual(str(response.context['user']), 'testuser_system_importer_file_csv_status') self.assertEqual( messages[0].message, 'System CSV importer: created: 2 | updated: 4 | skipped: 0 | multiple: 0 [2021-03-11 19:10:00 - 2021-03-11 19:10:00]' ) self.assertEqual(messages[0].level_tag, 'success') # 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 - user 2 self.assertEqual(str(response.context['user']), 'message_user') self.assertEqual( messages[0].message, 'System CSV importer: created: 2 | updated: 4 | skipped: 0 | multiple: 0 [2021-03-11 19:10:00 - 2021-03-11 19:10:00]' ) self.assertEqual(messages[0].level_tag, 'success') # compare - systems / attributes self = compare_system_and_attributes_tagfree_status(self)
def test_system_importer_file_csv_check_content_file_system_cron_file_not_existing( self, ): """test importer view""" # change config set_csv_import_path('/tmp') # change config set_csv_import_filename('filename_not_existing.abc') # execute cron job / scheduled task 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_importer_file_csv_check_content_file_system', ) self.assertEqual( messages[0].message, '[Scheduled task CSV system importer] CSV import file does not exist. Check config or provide file!', ) 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 CSV system importer] CSV import file does not exist. Check config or provide file!', ) self.assertEqual(messages[0].level_tag, 'error')
def test_system_importer_file_csv_messages_cron_single_system_multiple(self): """ test importer view """ # create system create_system('system_csv_11_001') create_system('system_csv_11_001') # change config set_csv_import_filename('system_importer_file_csv_testfile_11_messages_single_system.csv') # change config set_csv_skip_existing_system(False) # mock timezone.now() t_4 = datetime(2021, 3, 7, 11, 00, tzinfo=timezone.utc) with patch.object(timezone, 'now', return_value=t_4): # execute cron job / scheduled task 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_importer_file_csv_messages') self.assertEqual(messages[0].message, 'System CSV importer: created: 0 | updated: 0 | skipped: 0 | multiple: 1 [2021-03-07 11:00:00 - 2021-03-07 11:00:00]') self.assertEqual(messages[0].level_tag, 'success') self.assertEqual(messages[1].message, "1 system was skipped because it existed several times. ['system_csv_11_001']") self.assertEqual(messages[1].level_tag, 'warning') # 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, 'System CSV importer: created: 0 | updated: 0 | skipped: 0 | multiple: 1 [2021-03-07 11:00:00 - 2021-03-07 11:00:00]') self.assertEqual(messages[0].level_tag, 'success') self.assertEqual(messages[1].message, "1 system was skipped because it existed several times. ['system_csv_11_001']") self.assertEqual(messages[1].level_tag, 'warning')
def test_system_importer_file_csv_check_attributes_upload_post_faulty_systems( self): """test importer view""" # change config set_config_column_system(2) # change config set_csv_import_filename( 'system_importer_file_csv_testfile_31_faulty_system.csv') # login testuser self.client.login( username='******', password='******', ) # open upload file systemcsv = open( os.path.join( BASE_DIR, 'dfirtrack_main/tests/system_importer/system_importer_file_csv_files/system_importer_file_csv_testfile_31_faulty_system.csv', )) # create post data data_dict = { 'systemcsv': systemcsv, } # create url destination = urllib.parse.quote('/system/', safe='/') # get response response = self.client.post('/system/importer/file/csv/upload/', data_dict) # get messages messages = list(get_messages(response.wsgi_request)) # compare - meta self.assertRedirects(response, destination, status_code=302, target_status_code=200) # compare - messages self = compare_messages_faulty_systems(self, messages) # compare - systems / attributes self = compare_system_and_attributes_faulty_systems(self) # close file systemcsv.close()
def test_system_importer_file_csv_check_content_file_system_instant_file_no_read_permission( self, ): """test importer view""" # get timestamp string t1 = timezone.now().strftime('%Y%m%d_%H%M%S') # set file system attributes csv_import_path = '/tmp' csv_import_filename = f'{t1}_instant_no_read_permission.csv' # create file create_file_no_read_permission(csv_import_path, csv_import_filename) # change config set_csv_import_path(csv_import_path) # change config set_csv_import_filename(csv_import_filename) # login testuser self.client.login( username= '******', password='******', ) # create url destination = urllib.parse.quote('/system/', safe='/') # get response response = self.client.get('/system/importer/file/csv/instant/', follow=True) # get messages messages = list(get_messages(response.wsgi_request)) # compare self.assertRedirects(response, destination, status_code=302, target_status_code=200) self.assertEqual( messages[0].message, 'No read permission for CSV import file. Check config or file system!', ) self.assertEqual(messages[0].level_tag, 'error')
def setUpTestData(cls): """ one-time setup """ # create user test_user = User.objects.create_user( username='******', password='******') User.objects.create_user(username='******', password='******') # create objects analysisstatus_1 = Analysisstatus.objects.create( analysisstatus_name='analysisstatus_1') analysisstatus_2 = Analysisstatus.objects.create( analysisstatus_name='analysisstatus_2') systemstatus_1 = Systemstatus.objects.create( systemstatus_name='systemstatus_1') systemstatus_2 = Systemstatus.objects.create( systemstatus_name='systemstatus_2') # create systems system_csv_56_001 = System.objects.create( system_name='system_csv_56_001', systemstatus=systemstatus_1, analysisstatus=analysisstatus_1, system_modify_time=timezone.now(), system_created_by_user_id=test_user, system_modified_by_user_id=test_user, ) system_csv_56_002 = System.objects.create( system_name='system_csv_56_002', systemstatus=systemstatus_1, analysisstatus=analysisstatus_1, system_modify_time=timezone.now(), system_created_by_user_id=test_user, system_modified_by_user_id=test_user, ) system_csv_56_003 = System.objects.create( system_name='system_csv_56_003', systemstatus=systemstatus_1, analysisstatus=analysisstatus_1, system_modify_time=timezone.now(), system_created_by_user_id=test_user, system_modified_by_user_id=test_user, ) tagcolor_1 = Tagcolor.objects.create(tagcolor_name='tagcolor_1') AUTO_tag_96_1 = Tag.objects.create( tag_name='AUTO_tag_96_1', tagcolor=tagcolor_1, ) tag_97_1 = Tag.objects.create( tag_name='tag_97_1', tagcolor=tagcolor_1, ) system_csv_56_001.tag.add(AUTO_tag_96_1) system_csv_56_002.tag.add(AUTO_tag_96_1) system_csv_56_003.tag.add(AUTO_tag_96_1) system_csv_56_001.tag.add(tag_97_1) system_csv_56_002.tag.add(tag_97_1) system_csv_56_003.tag.add(tag_97_1) # set config set_csv_import_path( os.path.join( BASE_DIR, 'dfirtrack_main/tests/system_importer/system_importer_file_csv_files/' )) # set config set_csv_import_filename( 'system_importer_file_csv_testfile_56_tag_delimiter_space.csv') # set config system_importer_file_csv_config_model = SystemImporterFileCsvConfigModel.objects.get( system_importer_file_csv_config_name='SystemImporterFileCsvConfig') system_importer_file_csv_config_model.csv_column_system = 1 system_importer_file_csv_config_model.csv_skip_existing_system = False system_importer_file_csv_config_model.csv_headline = False system_importer_file_csv_config_model.csv_import_username = test_user system_importer_file_csv_config_model.csv_default_systemstatus = systemstatus_1 system_importer_file_csv_config_model.csv_default_analysisstatus = analysisstatus_1 system_importer_file_csv_config_model.csv_default_tagfree_systemstatus = systemstatus_2 system_importer_file_csv_config_model.csv_default_tagfree_analysisstatus = analysisstatus_2 system_importer_file_csv_config_model.csv_tag_lock_systemstatus = 'LOCK_SYSTEMSTATUS' system_importer_file_csv_config_model.csv_tag_lock_analysisstatus = 'LOCK_ANALYSISSTATUS' system_importer_file_csv_config_model.csv_field_delimiter = 'field_comma' system_importer_file_csv_config_model.csv_text_quote = 'text_double_quotation_marks' system_importer_file_csv_config_model.csv_ip_delimiter = 'ip_semicolon' system_importer_file_csv_config_model.csv_tag_delimiter = 'tag_space' system_importer_file_csv_config_model.csv_tag_prefix = 'AUTO' system_importer_file_csv_config_model.csv_tag_prefix_delimiter = 'tag_prefix_underscore' system_importer_file_csv_config_model.csv_choice_tag = True system_importer_file_csv_config_model.csv_column_tag = 2 system_importer_file_csv_config_model.save()