def end_inner_iterator_if_required(active_iterator_handle):
itr_result_obj = iterator_utils.get_leaf_iterator()
root_result = iterator_utils.get_stm_test_result_throw()
if itr_result_obj is None or \
itr_result_obj.GetObjectHandle() == root_result.GetObjectHandle() or \
itr_result_obj.Get(Iterator.get_stc_property_Iterator()) == active_iterator_handle:
logger.debug('Either new iterator or leaf iterator.')
return
# find if new iterator or existing iterator with missing child end iterator
while itr_result_obj is not None and \
itr_result_obj.GetObjectHandle() != root_result.GetObjectHandle() and \
itr_result_obj.Get(Iterator.get_stc_property_Iterator()) != active_iterator_handle:
itr_result_obj = itr_result_obj.GetParent()
logger.debug_result_object_info(itr_result_obj)
if itr_result_obj.GetObjectHandle() == root_result.GetObjectHandle() or \
itr_result_obj.Get(Iterator.get_stc_property_Iterator()) != active_iterator_handle:
logger.debug('No existing iterator found with same handle')
return
# call end iterator for missing
leaf_iterator = iterator_utils.get_leaf_iterator()
while leaf_iterator.Get(Iterator.get_stc_property_Iterator()) != active_iterator_handle:
logger.error('Missing end iterator for child iterator')
end_iterator()
leaf_iterator = iterator_utils.get_leaf_iterator()
def complete_iteration():
logger.info('Complete iteration.')
itr_result_obj = iterator_utils.get_leaf_iterator()
if itr_result_obj is None:
logger.error('There is no iterator running to complete iteration.')
return
logger.debug_result_object_info(itr_result_obj)
iterator = Iterator(0, itr_result_obj)
iterator.complete_active_iteration()
logger.debug('Complete iteration completed.')
def add_provider_result(dict_data):
"""Add provider result to active result object.
It is leaf iterator result object in case of iterative test.
Do not throw error if result object does not exist.
Try to create one and continue.
"""
logger.info('Add provider result.')
# make sure test in created and running. If not do that first.
start_test_if_required()
result = iterator_utils.get_active_result_object()
logger.debug_result_object_info(result)
dict_data = result_utils.insert_report_group_if_not_defined(dict_data)
result_obj.add_provider_data(result, dict_data)
def end_iterator(force_stop=False):
logger.info('End iteration.')
itr_result_obj = iterator_utils.get_leaf_iterator()
if itr_result_obj is None:
logger.error('There is no iterator running to end iterator.')
return
iterator = Iterator(0, itr_result_obj)
data = iterator.complete(force_stop)
parentIterator = iterator_utils.get_leaf_iterator_no_throw()
if parentIterator is None:
logger.info('Adding iteration result to Test.')
data[pc.STATUS][pc.APPLY_VERDICT] = True
test_result = TestResult()
test_result.add_provider_data(data)
else:
logger.info('Adding iteration result to parent iterator.')
logger.debug_result_object_info(itr_result_obj)
parent_itr = Iterator(0, parentIterator)
parent_itr.add_child_iterator_data(data)
logger.debug('End iteration completed')