forked from openkylin/flit
171 lines
4.5 KiB
Python
171 lines
4.5 KiB
Python
from contextlib import contextmanager
|
|
from tempfile import NamedTemporaryFile
|
|
import os
|
|
import io
|
|
import pathlib
|
|
import sys
|
|
|
|
import pytest
|
|
import responses
|
|
from testpath import modified_env
|
|
from unittest.mock import patch
|
|
|
|
from flit import upload
|
|
from flit.build import ALL_FORMATS
|
|
|
|
samples_dir = pathlib.Path(__file__).parent / 'samples'
|
|
|
|
repo_settings = {'url': upload.PYPI,
|
|
'username': 'user',
|
|
'password': 'pw',
|
|
'is_warehouse': True,
|
|
}
|
|
|
|
pypirc1 = """
|
|
[distutils]
|
|
index-servers =
|
|
pypi
|
|
|
|
[pypi]
|
|
username: fred
|
|
password: s3cret
|
|
"""
|
|
# That's not a real password. Well, hopefully not.
|
|
|
|
@contextmanager
|
|
def temp_pypirc(content):
|
|
try:
|
|
temp_file = NamedTemporaryFile("w+", delete=False)
|
|
temp_file.write(content)
|
|
temp_file.close()
|
|
yield temp_file.name
|
|
finally:
|
|
os.unlink(temp_file.name)
|
|
|
|
|
|
@responses.activate
|
|
def test_upload(copy_sample):
|
|
responses.add(responses.POST, upload.PYPI, status=200)
|
|
td = copy_sample('module1_toml')
|
|
|
|
with temp_pypirc(pypirc1) as pypirc, \
|
|
patch('flit.upload.get_repository', return_value=repo_settings):
|
|
upload.main(td / 'pyproject.toml', repo_name='pypi', pypirc_path=pypirc)
|
|
|
|
assert len(responses.calls) == 2
|
|
|
|
def test_get_repository():
|
|
with temp_pypirc(pypirc1) as pypirc:
|
|
repo = upload.get_repository(pypirc_path=pypirc)
|
|
assert repo['url'] == upload.PYPI
|
|
assert repo['username'] == 'fred'
|
|
assert repo['password'] == 's3cret'
|
|
|
|
def test_get_repository_env():
|
|
with temp_pypirc(pypirc1) as pypirc, \
|
|
modified_env({
|
|
'FLIT_INDEX_URL': 'https://pypi.example.com',
|
|
'FLIT_USERNAME': 'alice',
|
|
'FLIT_PASSWORD': 'p4ssword', # Also not a real password
|
|
}):
|
|
repo = upload.get_repository(pypirc_path=pypirc)
|
|
# Because we haven't specified a repo name, environment variables should
|
|
# have higher priority than the config file.
|
|
assert repo['url'] == 'https://pypi.example.com'
|
|
assert repo['username'] == 'alice'
|
|
assert repo['password'] == 'p4ssword'
|
|
|
|
@contextmanager
|
|
def _fake_keyring(pw):
|
|
real_keyring = sys.modules.get('keyring', None)
|
|
class FakeKeyring:
|
|
@staticmethod
|
|
def get_password(service_name, username):
|
|
return pw
|
|
|
|
sys.modules['keyring'] = FakeKeyring()
|
|
|
|
try:
|
|
yield
|
|
finally:
|
|
if real_keyring is None:
|
|
del sys.modules['keyring']
|
|
else:
|
|
sys.modules['keyring'] = real_keyring
|
|
|
|
pypirc2 = """
|
|
[distutils]
|
|
index-servers =
|
|
pypi
|
|
|
|
[pypi]
|
|
username: fred
|
|
"""
|
|
|
|
def test_get_repository_keyring():
|
|
with modified_env({'FLIT_PASSWORD': None}), \
|
|
_fake_keyring('tops3cret'):
|
|
repo = upload.get_repository(pypirc_path=io.StringIO(pypirc2))
|
|
|
|
assert repo['username'] == 'fred'
|
|
assert repo['password'] == 'tops3cret'
|
|
|
|
|
|
pypirc3_repo = "https://invalid-repo.inv"
|
|
pypirc3_user = "test"
|
|
pypirc3_pass = "not_a_real_password"
|
|
pypirc3 = f"""
|
|
[distutils] =
|
|
index-servers =
|
|
test123
|
|
|
|
[test123]
|
|
repository: {pypirc3_repo}
|
|
username: {pypirc3_user}
|
|
password: {pypirc3_pass}
|
|
"""
|
|
|
|
|
|
def test_upload_pypirc_file(copy_sample):
|
|
with temp_pypirc(pypirc3) as pypirc, \
|
|
patch("flit.upload.upload_file") as upload_file:
|
|
td = copy_sample("module1_toml")
|
|
formats = list(ALL_FORMATS)[:1]
|
|
upload.main(
|
|
td / "pyproject.toml",
|
|
formats=set(formats),
|
|
repo_name="test123",
|
|
pypirc_path=pypirc,
|
|
)
|
|
_, _, repo = upload_file.call_args[0]
|
|
|
|
assert repo["url"] == pypirc3_repo
|
|
assert repo["username"] == pypirc3_user
|
|
assert repo["password"] == pypirc3_pass
|
|
|
|
|
|
def test_upload_invalid_pypirc_file(copy_sample):
|
|
with patch("flit.upload.upload_file"):
|
|
td = copy_sample("module1_toml")
|
|
formats = list(ALL_FORMATS)[:1]
|
|
with pytest.raises(FileNotFoundError):
|
|
upload.main(
|
|
td / "pyproject.toml",
|
|
formats=set(formats),
|
|
repo_name="test123",
|
|
pypirc_path="./file.invalid",
|
|
)
|
|
|
|
def test_upload_default_pypirc_file(copy_sample):
|
|
with patch("flit.upload.do_upload") as do_upload:
|
|
td = copy_sample("module1_toml")
|
|
formats = list(ALL_FORMATS)[:1]
|
|
upload.main(
|
|
td / "pyproject.toml",
|
|
formats=set(formats),
|
|
repo_name="test123",
|
|
)
|
|
|
|
file = do_upload.call_args[0][2]
|
|
assert file == "~/.pypirc"
|