summaryrefslogtreecommitdiff
path: root/archaeological_context_records/tests.py
diff options
context:
space:
mode:
authorÉtienne Loks <etienne.loks@iggdrasil.net>2016-09-11 11:43:14 +0200
committerÉtienne Loks <etienne.loks@iggdrasil.net>2016-09-11 11:43:14 +0200
commit9ebf429a92a8fa0dd04df6cdd51caa2ef1c58c39 (patch)
tree3bbb30e8d51acc9d52083be6b21c2bdcfd2f5fdc /archaeological_context_records/tests.py
parent73afc7f761b29a1404cdf567e1880816abcb96aa (diff)
parent08f1fe99d11e40ca506469c1d782791fca18d49d (diff)
downloadIshtar-9ebf429a92a8fa0dd04df6cdd51caa2ef1c58c39.tar.bz2
Ishtar-9ebf429a92a8fa0dd04df6cdd51caa2ef1c58c39.zip
Merge branch 'v0.9' into wheezy
Diffstat (limited to 'archaeological_context_records/tests.py')
-rw-r--r--archaeological_context_records/tests.py12
1 files changed, 6 insertions, 6 deletions
diff --git a/archaeological_context_records/tests.py b/archaeological_context_records/tests.py
index 27ba0ab71..9b8d4f000 100644
--- a/archaeological_context_records/tests.py
+++ b/archaeological_context_records/tests.py
@@ -90,14 +90,14 @@ class ImportContextRecordTest(ImportOperationTest):
class ContextRecordInit(OperationInitTest):
test_operations = False
- def create_context_record(self, user=None, data={}):
+ def create_context_record(self, user=None, data={}, force=False):
if not getattr(self, 'context_records', None):
self.context_records = []
default = {'label': "Context record"}
- if not data.get('operation'):
- data['operation'] = self.get_default_operation()
- if not data.get('parcel'):
+ 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()
if not data.get('history_modifier'):
data['history_modifier'] = self.get_default_user()
@@ -107,8 +107,8 @@ class ContextRecordInit(OperationInitTest):
**default))
return self.context_records
- def get_default_context_record(self):
- return self.create_context_record()[0]
+ def get_default_context_record(self, force=False):
+ return self.create_context_record(force=force)[0]
def tearDown(self):
if hasattr(self, 'context_records'):