diff --git a/tests/unit/publication/gitlab/tokenizer/test_client.py b/tests/unit/publication/gitlab/tokenizer/test_client.py index 0e81b0ce..8e07c6c4 100644 --- a/tests/unit/publication/gitlab/tokenizer/test_client.py +++ b/tests/unit/publication/gitlab/tokenizer/test_client.py @@ -1,40 +1,42 @@ -from unittest.mock import MagicMock, patch - -import pytest - -from overhave.publication.gitlab.tokenizer import TokenizerClient -from overhave.publication.gitlab.tokenizer.client import InvalidUrlException - - -class TestTokenizerClient: - """Tests for :class:`TokenizerClient`.""" - - @pytest.mark.parametrize( - "test_tokenizer_client", - [{"initiator": "peka", "remote_key": "pepe", "remote_key_name": "sad-pepe", "url": "https://ya.ru"}], - indirect=True, - ) - def test_tokenizer_client_get_token_works(self, test_tokenizer_client) -> None: - token_mock = "magic_token" - draft_id_mock = 4 - - client = test_tokenizer_client - - request_mock = MagicMock() - request_mock.json = MagicMock(return_value={"token": token_mock}) - - with patch.object(TokenizerClient, "_make_request", return_value=request_mock) as make_request: - tokenizer_client = client.get_token(draft_id_mock) - assert tokenizer_client.token == token_mock - make_request.assert_called_once() - - @pytest.mark.parametrize( - "test_tokenizer_client", - [{"initiator": "peka", "remote_key": "pepe", "remote_key_name": "sad-pepe"}], - indirect=True, - ) - def test_tokenizer_client_get_token_url_validation_raises_error(self, test_tokenizer_client) -> None: - draft_id_mock = 4 - - with pytest.raises(InvalidUrlException): - test_tokenizer_client.get_token(draft_id_mock) +from unittest.mock import MagicMock, patch + +import pytest + +from overhave.publication.gitlab.tokenizer import TokenizerClient +from overhave.publication.gitlab.tokenizer.client import InvalidUrlException + + +class TestTokenizerClient: + """Tests for :class:`TokenizerClient`.""" + + @pytest.mark.parametrize( + "test_tokenizer_client", + [{"initiator": "peka", "remote_key": "pepe", "remote_key_name": "sad-pepe", "url": "https://ya.ru"}], + indirect=True, + ) + def test_tokenizer_client_get_token_works(self, test_tokenizer_client: TokenizerClient) -> None: + token_mock = "magic_token" + draft_id_mock = 4 + + client = test_tokenizer_client + + request_mock = MagicMock() + request_mock.json = MagicMock(return_value={"token": token_mock}) + + with patch.object(TokenizerClient, "_make_request", return_value=request_mock) as make_request: + tokenizer_client = client.get_token(draft_id_mock) + assert tokenizer_client.token == token_mock + make_request.assert_called_once() + + @pytest.mark.parametrize( + "test_tokenizer_client", + [{"initiator": "peka", "remote_key": "pepe", "remote_key_name": "sad-pepe"}], + indirect=True, + ) + def test_tokenizer_client_get_token_url_validation_raises_error( + self, test_tokenizer_client: TokenizerClient + ) -> None: + draft_id_mock = 4 + + with pytest.raises(InvalidUrlException): + test_tokenizer_client.get_token(draft_id_mock)