mirror of
https://github.com/pre-commit/pre-commit-hooks.git
synced 2026-04-05 11:36:54 +00:00
24 lines
958 B
Python
24 lines
958 B
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
from unittest.mock import mock_open
|
|
from unittest.mock import patch
|
|
|
|
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()
|