summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorÉtienne Loks <etienne.loks@iggdrasil.net>2016-12-20 18:23:34 +0100
committerÉtienne Loks <etienne.loks@iggdrasil.net>2016-12-20 18:23:34 +0100
commitd064a7aa0c9bea7b5cdd2767df003716fe789dcd (patch)
tree1d13c5afb20715ae44ba17bc97dff61749bb1e20
parent2ea51d570a9728d604ec136185f9ad5c9d8f792b (diff)
downloadIshtar-d064a7aa0c9bea7b5cdd2767df003716fe789dcd.tar.bz2
Ishtar-d064a7aa0c9bea7b5cdd2767df003716fe789dcd.zip
Fix default test
-rw-r--r--archaeological_context_records/tests.py2
-rw-r--r--archaeological_operations/tests.py2
2 files changed, 4 insertions, 0 deletions
diff --git a/archaeological_context_records/tests.py b/archaeological_context_records/tests.py
index 8ce2fd9d5..a242799ef 100644
--- a/archaeological_context_records/tests.py
+++ b/archaeological_context_records/tests.py
@@ -108,6 +108,8 @@ class ContextRecordInit(OperationInitTest):
return self.context_records
def get_default_context_record(self, force=False):
+ if force:
+ return self.create_context_record(force=force)[-1]
return self.create_context_record(force=force)[0]
def tearDown(self):
diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py
index 93c4fbe92..7d11ee002 100644
--- a/archaeological_operations/tests.py
+++ b/archaeological_operations/tests.py
@@ -496,6 +496,8 @@ class OperationInitTest(object):
return self.operations
def get_default_operation(self, force=False):
+ if force:
+ return self.create_operation()[-1]
return self.create_operation()[0]
def tearDown(self):