20 lines
507 B
Python
20 lines
507 B
Python
|
from unittest import mock
|
||
|
|
||
|
import pytest
|
||
|
|
||
|
from setuptools.command.upload import upload
|
||
|
from setuptools.dist import Distribution
|
||
|
from setuptools.errors import RemovedCommandError
|
||
|
|
||
|
|
||
|
class TestUpload:
|
||
|
def test_upload_exception(self):
|
||
|
"""Ensure that the register command has been properly removed."""
|
||
|
dist = Distribution()
|
||
|
dist.dist_files = [(mock.Mock(), mock.Mock(), mock.Mock())]
|
||
|
|
||
|
cmd = upload(dist)
|
||
|
|
||
|
with pytest.raises(RemovedCommandError):
|
||
|
cmd.run()
|