diff options
| -rw-r--r-- | archaeological_context_records/tests.py | 2 | ||||
| -rw-r--r-- | archaeological_operations/tests.py | 4 |
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): |