def test_init_user_group(self, mock_superuser):
Group.objects.all().delete()
self.assertEqual(0, Group.objects.count())
self.assertFalse(mock_superuser.called)
init_user_group('w3af_webui')
self.assertTrue(mock_superuser.called)
self.assertEqual(6, Group.objects.count())
def setUp(self):
# Every test needs a client.
User.objects.all().delete()
self.user_smb = User.objects.create_user('somebody__else', '*****@*****.**',
'somebody__else')
self.user_smb.is_staff = True
self.user_smb.save()
self.user = User.objects.create_user('new_unittest', '*****@*****.**',
'new_test_password')
init_user_group('w3af_webui')
call_command('set_role_for_user',
'new_unittest')
self.user.is_staff = True
self.user.is_superuser = True
self.user.save()
self.client = Client()
self.client.login(username=self.user.username,
password='******')
self.profile = any_model(ScanProfile, user=self.user)
self.target = any_model(Target, user=self.user)
self.not_mine_target = any_model(Target, user=self.user_smb)
self.scan_task = any_model(ScanTask,
user=self.user,
status=settings.TASK_STATUS['free'],
target=self.target,
last_updated='0',
)
self.scan = Scan.objects.create(scan_task=self.scan_task,
data='test',
)
def setUp(self):
# Every test needs a client.
User.objects.all().delete()
self.user_smb = User.objects.create_user('somebody__else',
'*****@*****.**',
'somebody__else')
self.user_smb.is_staff = True
self.user_smb.save()
self.user = User.objects.create_user('new_unittest',
'*****@*****.**',
'new_test_password')
init_user_group('w3af_webui')
call_command('set_role_for_user', 'new_unittest')
self.user.is_staff = True
self.user.is_superuser = True
self.user.save()
self.client = Client()
self.client.login(username=self.user.username,
password='******')
self.profile = any_model(ScanProfile, user=self.user)
self.target = any_model(Target, user=self.user)
self.not_mine_target = any_model(Target, user=self.user_smb)
self.scan_task = any_model(
ScanTask,
user=self.user,
status=settings.TASK_STATUS['free'],
target=self.target,
last_updated='0',
)
self.scan = Scan.objects.create(
scan_task=self.scan_task,
data='test',
)
