"""Tests for the  AWS Secret Manager connector."""

import json
from unittest.mock import MagicMock

from boto3.session import Session

from users.connectors import aws_secrets_manager


def test_get_secret(monkeypatch):
    """Test get secret."""

    class Client:
        def get_secret_value(self, SecretId):
            return {'SecretString': json.dumps({'token': SecretId})}

    monkeypatch.setattr(aws_secrets_manager, 'TOKEN_TYPES', {'feature-fm': 'token'})
    monkeypatch.setattr(Session, 'client', MagicMock(return_value=Client()))
    expected_result = {'token': 'token'}
    result = aws_secrets_manager.get_secret('feature-fm')

    assert result == expected_result