From 5f77a4ab463e711121484a17bb5de522400b2dc9 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Sat, 29 Apr 2017 13:20:21 +0200 Subject: Extra security to prevent recursion on bulk update --- archaeological_operations/tests.py | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'archaeological_operations') diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py index a5b83ff1c..0b466c48c 100644 --- a/archaeological_operations/tests.py +++ b/archaeological_operations/tests.py @@ -25,6 +25,7 @@ from django.core.files.uploadedfile import SimpleUploadedFile from django.core.urlresolvers import reverse from django.test.client import Client +from django.contrib.auth.models import User from django.contrib.auth.models import Permission import models @@ -593,8 +594,12 @@ def create_operation(user, orga=None, values={}): class OperationInitTest(object): def create_user(self): username, password, self.user = create_user() + return self.user def get_default_user(self): + q = User.objects.filter(is_superuser=False) + if q.count(): + return q.all()[0] return self.create_user() def create_orgas(self, user=None): -- cgit v1.2.3