docstr: Test we return the right value -- that will get put in to XCom by the execution engine
docstr_tok: ["Test","we","return","the","right","value","--","that","will","get","put","in","to","XCom","by","the","execution","engine"]
code: def test_return_value(
      self, mock_conn, mock_run_query, mock_check_query_status
  ):
      """Test we return the right value -- that will get put in to XCom by the execution engine"""
      dag_run = DagRun(
          dag_id=self.dag.dag_id,
          execution_date=timezone.utcnow(),
          run_id="test",
      )
      ti = TaskInstance(task=self.athena)
      ti.dag_run = dag_run
  
      assert (
          self.athena.execute(ti.get_template_context())
          == ATHENA_QUERY_ID
      )
  
imports: from __future__ import annotations
  import json
  from unittest import mock
  import pytest
  from openlineage.client.facet import SchemaDatasetFacet, SchemaField, SqlJobFacet, SymlinksDatasetFacet, SymlinksDatasetFacetIdentifiers
  from openlineage.client.run import Dataset
  from airflow.exceptions import TaskDeferred
  from airflow.models import DAG, DagRun, TaskInstance
  from airflow.providers.amazon.aws.hooks.athena import AthenaHook
  from airflow.providers.amazon.aws.operators.athena import AthenaOperator
  from airflow.providers.amazon.aws.triggers.athena import AthenaTrigger
  from airflow.providers.openlineage.extractors import OperatorLineage
  from airflow.utils import timezone
  from airflow.utils.timezone import datetime
label_desc: Use image node_id 10 for calling the TestAthenaOperator obj's underlying member method code with example usage: obj.test_return_value(mock_conn, mock_run_query, mock_check_query_status) without return types