Wizard tests improvements. Test inappropriate parcel deletion.

1 files changed, 10 insertions, 2 deletions

diff --git a/archaeological_context_records/tests.py b/archaeological_context_records/tests.py
index 83e013ee1..0018aa1f3 100644
--- a/archaeological_context_records/tests.py
+++ b/archaeological_context_records/tests.py
@@ -148,14 +148,22 @@ class ContextRecordInit(OperationInitTest):
         if not getattr(self, 'context_records', None):
             self.context_records = []
         default = {'label': "Context record"}
-        if force or not data.get('operation'):
+        if force or not data.get('operation') \
+                or not models.Operation.objects.filter(
+                    pk=data['operation'].pk).count():
             data['operation'] = self.get_default_operation(force=force)
-        if not data.get('parcel') or not data['parcel'].pk:
+        if not data.get('parcel') or not data['parcel'].pk \
+            or not models.Parcel.objects.filter(
+                pk=data['parcel'].pk).count():
             data['parcel'] = self.get_default_parcel(force=force)
         if not data.get('history_modifier'):
             data['history_modifier'] = self.get_default_user()
 
         default.update(data)
+        data['operation'] = models.Operation.objects.get(
+            pk=data['operation'].pk)
+        data['parcel'] = models.Parcel.objects.get(
+            pk=data['parcel'].pk)
         self.context_records.append(models.ContextRecord.objects.create(
             **default))
         return self.context_records