improve tests
This commit is contained in:
parent
8c414fd87d
commit
8e14fc19e6
@ -13,7 +13,6 @@ from announce_server.decorator import (
|
||||
@pytest.mark.parametrize(
|
||||
"decorator_alias",
|
||||
[
|
||||
announce_server,
|
||||
register_block,
|
||||
register_server,
|
||||
],
|
||||
@ -35,3 +34,28 @@ def test_alias_calls_register_service(decorator_alias):
|
||||
decorator(MagicMock())
|
||||
|
||||
mock_register_service.assert_called_once_with(*test_args, **test_kwargs)
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"decorator_alias",
|
||||
[
|
||||
announce_server,
|
||||
],
|
||||
)
|
||||
def test_deprecated_alias_calls_register_service(decorator_alias):
|
||||
test_args = (None, None)
|
||||
test_kwargs = {
|
||||
"name": "test_server",
|
||||
"ip": "127.0.0.1",
|
||||
"port": 8000,
|
||||
"host_ip": "127.0.0.1",
|
||||
"host_port": 5000,
|
||||
}
|
||||
with patch("announce_server.decorator.register_service") as mock_register_service:
|
||||
mock_register_service.return_value = MagicMock()
|
||||
|
||||
with pytest.warns(DeprecationWarning):
|
||||
decorator = decorator_alias(*test_args, **test_kwargs)
|
||||
|
||||
decorator(MagicMock())
|
||||
mock_register_service.assert_called_once_with(*test_args, **test_kwargs)
|
||||
|
Loading…
Reference in New Issue
Block a user