"""Test sts token generation."""

from botocore.client import BaseClient

from moto import mock_sts

from pricing.connectors import sts


@mock_sts
def test_connect_to_sts_success():
    """Test sts connection."""
    sts_client = sts.get_sts_client()
    assert sts_client.__class__.__name__ == 'STS'
    assert isinstance(sts_client, BaseClient)