def test_write_key__to_stdout(isatty_cleanup):
"""When sys.stdout is being piped/redicrected, print contents to it."""
download.sys.stdout.isatty = mock.Mock(return_value=False)
key = mock.Mock()
download.write_key(key)
key.get_contents_to_file.assert_called_once_with(download.sys.stdout)
def test_write_key__generate_url(flag, capfd, argv_cleanup):
"""Ensure that we print a url when asked."""
key = mock.Mock()
download.sys.argv = ["download", flag]
download.write_key(key)
key.generate_url.assert_called_once_with(300) # 5 minute URLs
out, err = capfd.readouterr()
assert not err
assert str(key.generate_url(300)) in out
def test_write_key__to_file(capfd, isatty_cleanup):
"""Verify the calls made to write a S3 key to file."""
download.sys.stdout.isatty = mock.Mock(return_value=True)
key = mock.Mock()
key.name = "mock_pkg/mock_pkg-1.0.2-dev1.tar.gz"
with mock.patch.object(builtins, "open") as open_patch:
download.write_key(key)
key.get_contents_to_file.assert_called_once_with(open_patch().__enter__())
out, err = capfd.readouterr()
assert not err
assert "mock_pkg-1.0.2-dev1.tar.gz" in out
