164 lines
5.1 KiB
Python
164 lines
5.1 KiB
Python
from tempfile import NamedTemporaryFile
|
|
|
|
import pytest
|
|
|
|
from src.config import (
|
|
Config,
|
|
FetchConfig,
|
|
LoggingConfig,
|
|
read_toml_config,
|
|
)
|
|
from src.models import LoggingOutputConfig
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"config_data, expected",
|
|
[
|
|
(
|
|
'[main]\napi_token = "something"\n'
|
|
'format = "{owner}/{name}"\n'
|
|
'endpoint = "https://example.com"\n'
|
|
'out_dir = "/home/user/repositories"\n'
|
|
"\n"
|
|
"[fetch]\n"
|
|
'ssh_key = "/tmp/no_key"',
|
|
Config(
|
|
api_token="something",
|
|
repository_format="{owner}/{name}",
|
|
out_dir="/home/user/repositories",
|
|
endpoint="https://example.com",
|
|
fetch=FetchConfig(ssh_key="/tmp/no_key"),
|
|
logging=LoggingConfig(),
|
|
),
|
|
),
|
|
(
|
|
'[main]\napi_token = "something"\n'
|
|
'format = "{owner}/{name}"\n'
|
|
'endpoint = "https://example.com"\n'
|
|
'out_dir = "/home/user/repositories"\n'
|
|
"\n"
|
|
"[fetch]\n"
|
|
'user = "alex"\n'
|
|
'token = "repo-token"',
|
|
Config(
|
|
api_token="something",
|
|
repository_format="{owner}/{name}",
|
|
out_dir="/home/user/repositories",
|
|
endpoint="https://example.com",
|
|
fetch=FetchConfig(user="alex", token="repo-token"),
|
|
logging=LoggingConfig(),
|
|
),
|
|
),
|
|
(
|
|
'[main]\napi_token = "something"\n'
|
|
'format = "{owner}/{name}"\n'
|
|
'endpoint = "https://example.com"\n'
|
|
'out_dir = "/home/user/repositories"\n'
|
|
"\n"
|
|
"[fetch]\n"
|
|
'ssh_key = "/tmp/no_key"\n'
|
|
"\n"
|
|
"[logging]\n"
|
|
'level = "DEBUG"\n'
|
|
"\n"
|
|
"[[logging.outputs]]\n"
|
|
'type = "console"\n'
|
|
'format = "plain"\n'
|
|
'stream = "stdout"\n'
|
|
"\n"
|
|
"[[logging.outputs]]\n"
|
|
'type = "console"\n'
|
|
'format = "json"\n'
|
|
'stream = "stderr"\n'
|
|
"\n"
|
|
"[[logging.outputs]]\n"
|
|
'type = "file"\n'
|
|
'path = "/tmp/gitea-mirror.log"\n'
|
|
'format = "json"\n'
|
|
"max_size = 1048576\n"
|
|
"backup_count = 3",
|
|
Config(
|
|
api_token="something",
|
|
repository_format="{owner}/{name}",
|
|
out_dir="/home/user/repositories",
|
|
endpoint="https://example.com",
|
|
fetch=FetchConfig(ssh_key="/tmp/no_key"),
|
|
logging=LoggingConfig(
|
|
level="DEBUG",
|
|
outputs=[
|
|
LoggingOutputConfig(
|
|
type="console",
|
|
format="plain",
|
|
stream="stdout",
|
|
),
|
|
LoggingOutputConfig(
|
|
type="console",
|
|
format="json",
|
|
stream="stderr",
|
|
),
|
|
LoggingOutputConfig(
|
|
type="file",
|
|
path="/tmp/gitea-mirror.log",
|
|
format="json",
|
|
max_size=1048576,
|
|
backup_count=3,
|
|
),
|
|
],
|
|
),
|
|
),
|
|
),
|
|
("[main]", None),
|
|
(
|
|
'[main]\napi_token = "something"\n'
|
|
'format = "{owner}/{name}"\n'
|
|
'endpoint = "https://example.com"\n'
|
|
'out_dir = "/home/user/repositories"\n'
|
|
"\n"
|
|
"[fetch]\n"
|
|
'ssh_key = "/tmp/no_key"\n'
|
|
'user = "alex"\n'
|
|
'token = "repo-token"',
|
|
None,
|
|
),
|
|
(
|
|
'[main]\napi_token = "something"\n'
|
|
'format = "{unknown}"\n'
|
|
'endpoint = "https://example.com"\n'
|
|
'out_dir = "/home/user/repositories"\n'
|
|
"\n"
|
|
"[fetch]\n"
|
|
'ssh_key = "/tmp/no_key"',
|
|
None,
|
|
),
|
|
(
|
|
'[main]\napi_token = "something"\n'
|
|
'format = "{owner}/{name}"\n'
|
|
'endpoint = "https://example.com"\n'
|
|
'out_dir = "/home/user/repositories"\n'
|
|
"\n"
|
|
"[fetch]\n"
|
|
'ssh_key = "/tmp/no_key"\n'
|
|
"\n"
|
|
"[logging]\n"
|
|
'level = "INF"',
|
|
None,
|
|
),
|
|
("not valid toml", None),
|
|
],
|
|
)
|
|
def test_toml_config(config_data, expected):
|
|
with NamedTemporaryFile() as tf:
|
|
if config_data:
|
|
tf.write(config_data.encode("utf-8"))
|
|
tf.flush()
|
|
if expected:
|
|
assert read_toml_config(tf.name) == expected
|
|
else:
|
|
with pytest.raises(RuntimeError):
|
|
read_toml_config(tf.name)
|
|
|
|
|
|
def test_toml_config_not_exists():
|
|
with pytest.raises(RuntimeError):
|
|
read_toml_config("not_existing_file")
|