summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorÉtienne Loks <etienne.loks@iggdrasil.net>2016-12-20 18:28:47 +0100
committerÉtienne Loks <etienne.loks@iggdrasil.net>2016-12-20 18:28:47 +0100
commit44da8d8dbd14bae9c3e9d9887770698cb9152fa7 (patch)
treef585c6732edc5b22637c21d9c6e77b8e16052e0e
parentd064a7aa0c9bea7b5cdd2767df003716fe789dcd (diff)
downloadIshtar-44da8d8dbd14bae9c3e9d9887770698cb9152fa7.tar.bz2
Ishtar-44da8d8dbd14bae9c3e9d9887770698cb9152fa7.zip
Fix default parcel creation
-rw-r--r--archaeological_context_records/tests.py2
-rw-r--r--archaeological_operations/tests.py4
2 files changed, 4 insertions, 2 deletions
diff --git a/archaeological_context_records/tests.py b/archaeological_context_records/tests.py
index a242799ef..fe4bb7674 100644
--- a/archaeological_context_records/tests.py
+++ b/archaeological_context_records/tests.py
@@ -98,7 +98,7 @@ class ContextRecordInit(OperationInitTest):
if force or not data.get('operation'):
data['operation'] = self.get_default_operation(force=force)
if not data.get('parcel') or not data['parcel'].pk:
- data['parcel'] = self.get_default_parcel()
+ data['parcel'] = self.get_default_parcel(force=force)
if not data.get('history_modifier'):
data['history_modifier'] = self.get_default_user()
diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py
index 7d11ee002..316e3c897 100644
--- a/archaeological_operations/tests.py
+++ b/archaeological_operations/tests.py
@@ -482,7 +482,9 @@ class OperationInitTest(object):
self.parcels.append(models.Parcel.objects.create(**default))
return self.parcels
- def get_default_parcel(self):
+ def get_default_parcel(self, force=False):
+ if force:
+ return self.create_parcel()[-1]
return self.create_parcel()[0]
def create_operation(self, user=None, orga=None):