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)