from __future__ import annotations

import unittest
from unittest.mock import patch, mock_open

from pre_commit_hooks.detect_gcp_credentials import detect_gcp_credentials_in_file


class TestDetect(unittest.TestCase):

    @patch('builtins.open', mock_open(read_data="Some content with GCP credentials: 'AIzaSyCzQ2rBq4dd8uf4Gh9J7G1IiCvC5q8oCvk'"))
    def test_detect_gcp_api_key(self):
        result = detect_gcp_credentials_in_file('')
        self.assertIn("AIzaSyCzQ2rBq4dd8uf4Gh9J7G1IiCvC5q8oCvk", result)

    @patch('builtins.open', mock_open(read_data="Some content with GCP credentials: '-----BEGIN PRIVATE KEY-----\nAbCdEf1234567890\n-----END PRIVATE KEY-----'"))
    def test_detect_gcp_service_account_key(self):
        result = detect_gcp_credentials_in_file('')
        self.assertIn("-----BEGIN PRIVATE KEY-----\nAbCdEf1234567890\n-----END PRIVATE KEY-----", result)

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