from unittest.mock import Mock


def boto3_client(firehose_client):
    """A mock for boto3.client."""
    def produce_client(name):
        return {"firehose": firehose_client}[name]

    mocked = Mock(side_effect=produce_client)
    return mocked