cleantoots/tests/test_cli.py

73 lines
2.6 KiB
Python

import os
import unittest
from click.testing import CliRunner
from cleantoots.main import cli
class SetupConfigTestCase(unittest.TestCase):
def setUp(self) -> None:
self.runner = CliRunner()
def test_setup_config(self):
with self.runner.isolated_filesystem():
result = self.runner.invoke(cli, ["-d", ".", "config", "setup"])
assert result.exit_code == 0
assert "config.ini written" in result.output
assert os.path.isfile(os.path.join(".", "config.ini"))
def test_setup_config_twice_fails(self):
with self.runner.isolated_filesystem():
self.runner.invoke(cli, ["-d", ".", "config", "setup"])
assert os.path.isfile(os.path.join(".", "config.ini"))
result = self.runner.invoke(cli, ["-d", ".", "config", "setup"])
assert result.exit_code == 0
assert "Not touching anything" in result.output
assert "cleantoots config edit" in result.output
def test_config_output(self):
with self.runner.isolated_filesystem():
self.runner.invoke(cli, ["-d", ".", "config", "setup"])
result = self.runner.invoke(cli, ["-d", ".", "config", "list"])
expected = [
"Mastodon.social",
"api_base_url",
"app_secret_file",
"user_secret_file",
"protected_toots",
"boost_limit",
"favorite_limit",
"days_count",
"timezone",
]
for exp in expected:
assert exp in result.output
def test_config_list_no_file(self):
with self.runner.isolated_filesystem():
result = self.runner.invoke(cli, ["-d", ".", "config", "list"])
assert "cleantoots config setup" in result.output
assert result.exit_code == 0
def test_config_edit_no_file(self):
with self.runner.isolated_filesystem():
result = self.runner.invoke(cli, ["-d", ".", "config", "edit"])
assert "cleantoots config setup" in result.output
assert result.exit_code == 0
def test_login_output(self):
with self.runner.isolated_filesystem():
self.runner.invoke(cli, ["-d", ".", "config", "setup"])
result = self.runner.invoke(
cli, ["-d", ".", "config", "login"], input="\nFAKECODE"
)
assert "Enter code for" in result.output
def test_clean_exists(self):
result = self.runner.invoke(cli, ["-d", ".", "clean"])
assert result.exit_code == 0
if __name__ == "__main__":
unittest.main()