diff --git a/poetry/publishing/publisher.py b/poetry/publishing/publisher.py index d30a9a47119..67515f77f64 100644 --- a/poetry/publishing/publisher.py +++ b/poetry/publishing/publisher.py @@ -86,7 +86,7 @@ def publish( "to {}".format( self._package.pretty_name, self._package.pretty_version, - {"pypi": "PyPI"}.get(repository_name, "PyPI"), + "PyPI" if repository_name == "pypi" else repository_name, ) ) diff --git a/tests/publishing/test_publisher.py b/tests/publishing/test_publisher.py index da376120645..a86c482226e 100644 --- a/tests/publishing/test_publisher.py +++ b/tests/publishing/test_publisher.py @@ -2,6 +2,8 @@ import pytest +from cleo.io import BufferedIO + from poetry.factory import Factory from poetry.io.null_io import NullIO from poetry.publishing.publisher import Publisher @@ -38,7 +40,8 @@ def test_publish_can_publish_to_given_repository(fixture_dir, mocker, config): "http-basic": {"my-repo": {"username": "foo", "password": "bar"}}, } ) - publisher = Publisher(poetry, NullIO()) + io = BufferedIO() + publisher = Publisher(poetry, io) publisher.publish("my-repo", None, None) @@ -47,6 +50,7 @@ def test_publish_can_publish_to_given_repository(fixture_dir, mocker, config): ("http://foo.bar",), {"cert": None, "client_cert": None, "dry_run": False}, ] == uploader_upload.call_args + assert "Publishing my-package (1.2.3) to my-repo" in io.fetch_output() def test_publish_raises_error_for_undefined_repository(fixture_dir, mocker, config):