aboutsummaryrefslogtreecommitdiffstats
path: root/setuptools/tests/test_register.py
blob: 96114595db04b13b675a5d4596f71eca617fa859 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import mock
from distutils import log

import pytest

from setuptools.command.register import register
from setuptools.dist import Distribution


class TestRegisterTest:
    def test_warns_deprecation(self):
        dist = Distribution()

        cmd = register(dist)
        cmd.run_command = mock.Mock()
        cmd.send_metadata = mock.Mock()
        cmd.announce = mock.Mock()

        cmd.run()

        cmd.announce.assert_called_with(
            "WARNING: Registering is deprecated, use twine to upload instead "
            "(https://pypi.org/p/twine/)",
            log.WARN
        )

    def test_warns_deprecation_when_raising(self):
        dist = Distribution()

        cmd = register(dist)
        cmd.run_command = mock.Mock()
        cmd.send_metadata = mock.Mock()
        cmd.send_metadata.side_effect = Exception
        cmd.announce = mock.Mock()

        with pytest.raises(Exception):
            cmd.run()

        cmd.announce.assert_called_with(
            "WARNING: Registering is deprecated, use twine to upload instead "
            "(https://pypi.org/p/twine/)",
            log.WARN
        )