from octopus.tasks import aws_cli_tool
from unittest.mock import MagicMock
from unittest.mock import patch
import subprocess


@patch('subprocess.Popen')
def test_upload_to_s3_from_stdin(mocker):
    mocker.return_value = MagicMock()
    mock_pipe = MagicMock(stdout=subprocess.PIPE)

    resp = aws_cli_tool.upload_to_s3_from_stdin(MagicMock(),
                                                'dim_artist',
                                                mock_pipe,
                                                'some-bucket')
    assert isinstance(resp, dict)