"""Test stepfunctions connector."""

import boto3
from daemon_asset_copy import config
from daemon_asset_copy.connectors import stepfunctions


def test_get_stepfunctions_client(mocker):
    """Test get_stepfunctions_client."""
    mocker.patch.object(boto3.session, 'Session', autospec=True)
    boto3.session.Session.return_value.client.return_value = 'abc'
    mocker.patch.object(boto3.session, 'Config', autospec=True)

    assert 'abc' == stepfunctions.get_stepfunctions_client()
    boto3.session.Config.assert_called_once_with(read_timeout=70)
    boto3.session.Session.return_value.client.assert_called_once_with(
        'stepfunctions',
        region_name=config.AWS_REGION,
        config=boto3.session.Config.return_value,
    )