code: def sagemaker_session_mock():
      session_mock = Mock()
      session_mock.boto_session.client = Mock()
      return session_mock
  
file_path: sagemaker-python-sdk/tests/unit/sagemaker/workflow/test_callback_step.py
imports: from __future__ import absolute_import
  import json
  import pytest
  from mock import Mock
  from sagemaker.workflow.parameters import ParameterInteger, ParameterString
  from sagemaker.workflow.pipeline import Pipeline, PipelineGraph
  from sagemaker.workflow.callback_step import CallbackStep, CallbackOutput, CallbackOutputTypeEnum
  from tests.unit.sagemaker.workflow.helpers import CustomStep, ordered
label_desc: Use image node_id 1 for calling a global function with example usage: sagemaker_session_mock() and returns: session_mock