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()
