How to check if a specific GitHub user is sponsoring me

sponsor.py

import os
from gql import gql, Client
from gql.transport.requests import RequestsHTTPTransport

token = os.getenv("GITHUB_TOKEN")

transport = RequestsHTTPTransport(
    url="https://api.github.com/graphql",
    use_json=True,
    headers={"Authorization": f"bearer {token}"},
    verify=True,
    retries=3,
)

client = Client(
    transport=transport,
    fetch_schema_from_transport=True,
)

def is_sponsor(user) -> bool:
    query = gql(
        f"""
        query {{
            user(login: "{user}") {{
                login
                sponsorshipForViewerAsSponsorable(activeOnly: true) {{
                    createdAt
                }}
            }}
        }}
        """
    )
    result = client.execute(query)

    return result["user"].get("sponsorshipForViewerAsSponsorable") is not None

sponsor_tests.py

import unittest
from unittest.mock import patch
from sponsor import is_sponsor

class TestIsSponsor(unittest.TestCase):
    @patch('main.client.execute')
    def test_is_sponsor(self, mock_execute):
        # モックの設定
        mock_execute.return_value = {
            "user": {
                "sponsorshipForViewerAsSponsorable": {
                    "createdAt": "2022-01-01T00:00:00Z"
                }
            }
        }

        # テストの実行
        result = is_sponsor('test_user')
        self.assertTrue(result)

        # モックが呼び出されたことを確認
        mock_execute.assert_called_once()

    @patch('main.client.execute')
    def test_is_not_sponsor(self, mock_execute):
        # モックの設定
        mock_execute.return_value = {
            "user": {
                "sponsorshipForViewerAsSponsorable": None
            }
        }

        # テストの実行
        result = is_sponsor('test_user')
        self.assertFalse(result)

        # モックが呼び出されたことを確認
        mock_execute.assert_called_once()

if __name__ == '__main__':
    unittest.main()