Skip to content
Snippets Groups Projects
Commit 181a2f74 authored by jakubs's avatar jakubs
Browse files

SP-136 BIS-21 refactor registration context

SVN: 25936
parent 37e8fc6c
No related branches found
No related tags found
No related merge requests found
......@@ -4,7 +4,7 @@ jythonHookTestTool = JythonHookTestTool.createInTest()
execfile("sourceTest/java/ch/systemsx/cisd/etlserver/registrator/simple-transaction.py")
transaction.getRegistrationContext().put("body","1")
transaction.getRegistrationContext().getPersistentMap().put("body","1")
def rollback_transaction(service, transaction, algorithmRunner, throwable):
jythonHookTestTool.log("rollback_transaction")
......@@ -15,7 +15,7 @@ def pre_metadata_registration(context):
assert_context_content(context, "pre_metadata_registration", "pre_metadata_registration", None);
assert_context_content(context, "pre_metadata_registration", "post_storage", None);
context.put("pre_metadata_registration", "2")
context.getPersistentMap().put("pre_metadata_registration", "2")
jythonHookTestTool.log("pre_metadata_registration")
def post_metadata_registration(context):
......@@ -24,7 +24,7 @@ def post_metadata_registration(context):
assert_context_content(context, "post_metadata_registration", "pre_metadata_registration", "2");
assert_context_content(context, "post_metadata_registration", "post_storage", None);
context.put("post_metadata_registration", "3")
context.getPersistentMap().put("post_metadata_registration", "3")
jythonHookTestTool.log("post_metadata_registration")
def post_storage(context):
......@@ -33,11 +33,11 @@ def post_storage(context):
assert_context_content(context, "post_storage", "pre_metadata_registration", "2");
assert_context_content(context, "post_storage", "post_metadata_registration", "3");
context.put("post_storage", "4")
context.getPersistentMap().put("post_storage", "4")
jythonHookTestTool.log("post_storage")
def assert_context_content(context, caller, name, expected):
value = context.get(name)
value = context.getPersistentMap().get(name)
if (value != expected):
if (value != None and expected != None and type(value) != type(expected)):
value = "%s:%s" % (type(value), value)
......
execfile("sourceTest/java/ch/systemsx/cisd/etlserver/registrator/all-hooks.py")
def should_retry_processing(map, error):
lastResult = map.get("RETRY_COUNT")
def should_retry_processing(context, error):
lastResult = context.getPersistentMap().get("RETRY_COUNT")
if (lastResult == None):
lastResult = 0
map.put("RETRY_COUNT", lastResult + 1)
context.getPersistentMap().put("RETRY_COUNT", lastResult + 1)
print "RETRY TRANSACTION PUTS THE SCORE %d" % (lastResult + 1)
return True
def process(transaction):
incoming = transaction.getIncoming()
key = transaction.getRegistrationContext().get("RETRY_COUNT");
key = transaction.getRegistrationContext().getPersistentMap().get("RETRY_COUNT");
if (key == None):
key = 1
dataSet = transaction.createNewDataSet()
......
......@@ -7,7 +7,7 @@ def process(transaction):
transaction.moveFile(transaction.getIncoming().getPath() + '/sub_data_set_1', dataSet)
dataSet.setDataSetType('O1')
dataSet.setExperiment(transaction.getExperiment('/SPACE/PROJECT/EXP'))
transaction.getRegistrationContext().put("body","1")
transaction.getRegistrationContext().getPersistentMap().put("body","1")
def rollback_transaction(service, transaction, algorithmRunner, throwable):
jythonHookTestTool.log("rollback_transaction")
......@@ -18,7 +18,7 @@ def pre_metadata_registration(context):
assert_context_content(context, "pre_metadata_registration", "pre_metadata_registration", None);
assert_context_content(context, "pre_metadata_registration", "post_storage", None);
context.put("pre_metadata_registration", "2")
context.getPersistentMap().put("pre_metadata_registration", "2")
jythonHookTestTool.log("pre_metadata_registration")
def post_metadata_registration(context):
......@@ -27,7 +27,7 @@ def post_metadata_registration(context):
assert_context_content(context, "post_metadata_registration", "pre_metadata_registration", "2");
assert_context_content(context, "post_metadata_registration", "post_storage", None);
context.put("post_metadata_registration", "3")
context.getPersistentMap().put("post_metadata_registration", "3")
jythonHookTestTool.log("post_metadata_registration")
def post_storage(context):
......@@ -36,11 +36,11 @@ def post_storage(context):
assert_context_content(context, "post_storage", "pre_metadata_registration", "2");
assert_context_content(context, "post_storage", "post_metadata_registration", "3");
context.put("post_storage", "4")
context.getPersistentMap().put("post_storage", "4")
jythonHookTestTool.log("post_storage")
def assert_context_content(context, caller, name, expected):
value = context.get(name)
value = context.getPersistentMap().get(name)
if (value != expected):
if (value != None and expected != None and type(value) != type(expected)):
value = "%s:%s" % (type(value), value)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment