def test_common_func_get_claimant(self):
_first = MagicMock(first=MagicMock(return_value="claimant"))
_values_list = MagicMock(return_value=_first)
_filter = MagicMock(values_list=_values_list)
class FunctionTask:
def __init__(self):
self.filter = MagicMock(return_value=_filter)
def _function_task(*args):
return FunctionTask()
def _mock_manager(*args):
return _function_task
def mock_manager(*args):
return _mock_manager()
with patch("django.db.models.fields.related_descriptors.create_reverse_many_to_one_manager", mock_manager):
taskflow = TaskFlowInstance()
taskflow.flow_type = "common_func"
self.assertEqual(taskflow.function_task_claimant, "claimant")
def test_not_common_func(self):
taskflow = TaskFlowInstance()
taskflow.flow_type = "common"
self.assertEqual(taskflow.function_task_claimant, None)
