"""Tests for S3Client."""
from unittest.mock import MagicMock
from unittest.mock import patch

from src.connectors.s3_client import S3Client


class TestS3Client:
    """Test S3Client class."""

    @patch('boto3.client')
    def test_put_object(self,
                        mock_boto_client):
        """Test put_object function."""
        mock_instance = MagicMock()
        mock_boto_client.return_value = mock_instance
        mock_instance.put_object.return_value = 'test_response'
        s3_client = S3Client()
        key = 'test_key'
        data = 'test_data'
        bucket = 'qa-lambda-insights-isrc-availability-tests'
        assert s3_client.put_object(key, data) == 'test_response'
        s3_client.s3.put_object.assert_called_with(
            Bucket=bucket, Key='test_key', Body='test_data')