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', )