aboutsummaryrefslogtreecommitdiff
path: root/setuptools/tests/test_register.py
diff options
context:
space:
mode:
Diffstat (limited to 'setuptools/tests/test_register.py')
-rw-r--r--setuptools/tests/test_register.py22
1 files changed, 22 insertions, 0 deletions
diff --git a/setuptools/tests/test_register.py b/setuptools/tests/test_register.py
new file mode 100644
index 0000000..9860580
--- /dev/null
+++ b/setuptools/tests/test_register.py
@@ -0,0 +1,22 @@
+from setuptools.command.register import register
+from setuptools.dist import Distribution
+from setuptools.errors import RemovedCommandError
+
+try:
+ from unittest import mock
+except ImportError:
+ import mock
+
+import pytest
+
+
+class TestRegister:
+ def test_register_exception(self):
+ """Ensure that the register command has been properly removed."""
+ dist = Distribution()
+ dist.dist_files = [(mock.Mock(), mock.Mock(), mock.Mock())]
+
+ cmd = register(dist)
+
+ with pytest.raises(RemovedCommandError):
+ cmd.run()