def test_read_secret_different_versions_from_secret_manager(self): hook = SecretsManagerHook() secret = hook.get_secret(secret_id=TEST_SECRET_ID) assert TEST_SECRET_VALUE_UPDATED == secret secret = hook.get_secret(secret_id=TEST_SECRET_ID, secret_version='1') assert TEST_SECRET_VALUE == secret secret = hook.get_secret(secret_id=TEST_SECRET_ID, secret_version='2') assert TEST_SECRET_VALUE_UPDATED == secret
def test_get_missing_key(self, mock_get_credentials, mock_client): mock_client.secret_version_path.return_value = "full-path" mock_client.access_secret_version.side_effect = NotFound('test-msg') secrets_manager_hook = SecretsManagerHook(gcp_conn_id='test') mock_get_credentials.assert_called_once_with() secret = secrets_manager_hook.get_secret(secret_id="secret") mock_client.secret_version_path.assert_called_once_with('example-project', 'secret', 'latest') mock_client.access_secret_version.assert_called_once_with("full-path") self.assertIsNone(secret)
def test_get_existing_key(self, mock_get_credentials, mock_client): mock_client.secret_version_path.return_value = "full-path" test_response = AccessSecretVersionResponse() test_response.payload.data = "result".encode("UTF-8") mock_client.access_secret_version.return_value = test_response secrets_manager_hook = SecretsManagerHook(gcp_conn_id='test') mock_get_credentials.assert_called_once_with() secret = secrets_manager_hook.get_secret(secret_id="secret") mock_client.secret_version_path.assert_called_once_with('example-project', 'secret', 'latest') mock_client.access_secret_version.assert_called_once_with("full-path") self.assertEqual("result", secret)
def test_read_missing_secret_from_secret_manager(self): hook = SecretsManagerHook() secret = hook.get_secret(secret_id=TEST_MISSING_SECRET_ID) assert secret is None
def test_read_secret_from_secret_manager(self): hook = SecretsManagerHook() secret = hook.get_secret(secret_id=TEST_SECRET_ID) assert TEST_SECRET_VALUE == secret
def test_read_secret_from_secret_manager(self): hook = SecretsManagerHook() secret = hook.get_secret(secret_id=TEST_SECRET_ID) self.assertEqual(TEST_SECRET_VALUE, secret)