code: def test_hook_poll_query_when_final(self, mock_conn):
      mock_conn.return_value.get_query_execution.return_value = (
          MOCK_SUCCEEDED_QUERY_EXECUTION
      )
      result = self.athena.poll_query_status(
          query_execution_id=MOCK_DATA["query_execution_id"],
          sleep_time=0,
      )
      mock_conn.return_value.get_query_execution.assert_called_once()
      assert result == "SUCCEEDED"
  
imports: from __future__ import annotations
  from unittest import mock
  import pytest
  from airflow.providers.amazon.aws.hooks.athena import AthenaHook
label_desc: Use image node_id 13 for calling the TestAthenaHook obj's underlying member method code with example usage: obj.test_hook_poll_query_when_final(mock_conn) without return types