173 lines
4.4 KiB
Python
173 lines
4.4 KiB
Python
from unittest.mock import Mock, patch
|
|
|
|
import pytest
|
|
from click.testing import CliRunner
|
|
|
|
from main import cli, get_configuration
|
|
|
|
|
|
@pytest.fixture(name="with_sample_config")
|
|
def fixture_with_sample_config(tmp_path):
|
|
config_path = tmp_path / "config.yml"
|
|
config_path.write_text(
|
|
"""
|
|
token: abc
|
|
presets:
|
|
test:
|
|
text: abc
|
|
emoji: ":tada:"
|
|
"""
|
|
)
|
|
|
|
|
|
def test_get_configuration_raises_if_noexist():
|
|
with pytest.raises(RuntimeError):
|
|
get_configuration("not/a/path.yml")
|
|
|
|
|
|
def test_get_configuration_returns_configuration_obj(tmp_path):
|
|
config_path = tmp_path / "config.yml"
|
|
config_path.write_text(
|
|
"""
|
|
token: abc
|
|
presets:
|
|
test:
|
|
text: abc
|
|
emoji: ":tada:"
|
|
"""
|
|
)
|
|
|
|
conf = get_configuration(config_path)
|
|
|
|
assert conf.token == "abc"
|
|
assert "test" in conf.presets
|
|
assert conf.presets["test"].text == "abc"
|
|
assert conf.presets["test"].emoji == ":tada:"
|
|
|
|
|
|
def test_cli_requires_text_or_preset_input(tmp_path, with_sample_config):
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli, ["--config", tmp_path / "config.yml"])
|
|
|
|
assert isinstance(result.exception, RuntimeError)
|
|
assert (
|
|
str(result.exception)
|
|
== "Must specify either status text via --text/-t or a preset via --preset/-p."
|
|
)
|
|
|
|
|
|
def test_cli_overrides_preset_with_text_input(with_sample_config, tmp_path):
|
|
runner = CliRunner()
|
|
|
|
mock_client = Mock()
|
|
mock_client.users_profile_set = Mock()
|
|
with patch(
|
|
"main.get_client", autospec=True, return_value=mock_client
|
|
):
|
|
result = runner.invoke(
|
|
cli,
|
|
[
|
|
"--text",
|
|
"testtext",
|
|
"--preset",
|
|
"test",
|
|
"--config",
|
|
tmp_path / "config.yml",
|
|
],
|
|
)
|
|
|
|
call_args = mock_client.users_profile_set.call_args
|
|
|
|
assert result.exit_code == 0
|
|
assert call_args.kwargs["profile"]["status_text"] == "testtext"
|
|
assert call_args.kwargs["profile"]["status_emoji"] == ":tada:"
|
|
|
|
|
|
def test_cli_overrides_preset_with_emoji_input(with_sample_config, tmp_path):
|
|
runner = CliRunner()
|
|
|
|
mock_client = Mock()
|
|
mock_client.users_profile_set = Mock()
|
|
with patch(
|
|
"main.get_client", autospec=True, return_value=mock_client
|
|
):
|
|
result = runner.invoke(
|
|
cli,
|
|
[
|
|
"--emoji",
|
|
":skull:",
|
|
"--preset",
|
|
"test",
|
|
"--config",
|
|
tmp_path / "config.yml",
|
|
],
|
|
)
|
|
|
|
call_args = mock_client.users_profile_set.call_args
|
|
|
|
assert result.exit_code == 0
|
|
assert call_args.kwargs["profile"]["status_text"] == "abc"
|
|
assert call_args.kwargs["profile"]["status_emoji"] == ":skull:"
|
|
|
|
|
|
def test_cli_raises_if_noexist_preset(tmp_path, with_sample_config):
|
|
runner = CliRunner()
|
|
|
|
mock_client = Mock()
|
|
mock_client.users_profile_set = Mock()
|
|
with patch(
|
|
"main.get_client", autospec=True, return_value=mock_client
|
|
):
|
|
result = runner.invoke(
|
|
cli, ["--preset", "not-a-preset", "--config", tmp_path / "config.yml"]
|
|
)
|
|
|
|
assert result.exit_code == 1
|
|
assert str(result.exception) == "Unknown preset: not-a-preset"
|
|
|
|
|
|
def test_cli_sends_request_to_slack(tmp_path, with_sample_config):
|
|
runner = CliRunner()
|
|
|
|
mock_client = Mock()
|
|
mock_client.users_profile_set = Mock()
|
|
with patch(
|
|
"main.get_client", autospec=True, return_value=mock_client
|
|
):
|
|
runner.invoke(
|
|
cli,
|
|
[
|
|
"--emoji",
|
|
":skull:",
|
|
"--preset",
|
|
"test",
|
|
"--config",
|
|
tmp_path / "config.yml",
|
|
],
|
|
)
|
|
|
|
mock_client.users_profile_set.assert_called()
|
|
|
|
|
|
def test_cli_raises_if_api_error(tmp_path, with_sample_config):
|
|
runner = CliRunner()
|
|
|
|
mock_client = Mock()
|
|
mock_client.users_profile_set = Mock(return_value={"ok": False})
|
|
with patch(
|
|
"main.get_client", autospec=True, return_value=mock_client
|
|
):
|
|
result = runner.invoke(
|
|
cli,
|
|
[
|
|
"--emoji",
|
|
":skull:",
|
|
"--preset",
|
|
"test",
|
|
"--config",
|
|
tmp_path / "config.yml",
|
|
],
|
|
)
|
|
|
|
assert result.exit_code == 1
|
|
assert str(result.exception) == "Failed to set status!"
|