"""Lambda test module."""
import pytest
from moto import mock_aws

from src import app as index


@pytest.fixture()
def mock_account_id():
    """Return an AWS account id."""
    return '123456789000'


@pytest.fixture()
def mock_role():
    """Return an AWS role name."""
    return 'test-role'


@mock_aws
def test_assume_source_account_role(mock_account_id, mock_role, monkeypatch):
    """Test assume_source_account_role function."""
    aws_credentials = ['AccessKeyId', 'SecretAccessKey', 'SessionToken']

    monkeypatch.setattr(index.config, 'EXTERNAL_ID', 'abc1234')
    result = index.assume_source_account_role(mock_account_id, mock_role)
    for credential in aws_credentials:
        assert result[credential]