mirror of
https://github.com/khoj-ai/khoj.git
synced 2024-11-23 15:38:55 +01:00
d2905c4be6
tests/ directory in project root is more standard. Just had to use absolute path for internal module imports to get it to work
70 lines
2.4 KiB
Python
70 lines
2.4 KiB
Python
# Standard Modules
|
|
from pathlib import Path
|
|
|
|
# Internal Packages
|
|
from src.utils.cli import cli
|
|
|
|
|
|
# Test
|
|
# ----------------------------------------------------------------------------------------------------
|
|
def test_cli_minimal_default():
|
|
# Act
|
|
actual_args = cli(['--config-file=tests/data/config.yml'])
|
|
|
|
# Assert
|
|
assert actual_args.config_file == Path('tests/data/config.yml')
|
|
assert actual_args.regenerate == False
|
|
assert actual_args.verbose == 0
|
|
|
|
# ----------------------------------------------------------------------------------------------------
|
|
def test_cli_flags():
|
|
# Act
|
|
actual_args = cli(['--config-file=tests/data/config.yml',
|
|
'--regenerate',
|
|
'-vvv'])
|
|
|
|
# Assert
|
|
assert actual_args.config_file == Path('tests/data/config.yml')
|
|
assert actual_args.regenerate == True
|
|
assert actual_args.verbose == 3
|
|
|
|
|
|
# ----------------------------------------------------------------------------------------------------
|
|
def test_cli_config_from_file():
|
|
# Act
|
|
actual_args = cli(['--config-file=tests/data/config.yml',
|
|
'--regenerate',
|
|
'-vvv'])
|
|
|
|
# Assert
|
|
assert actual_args.config_file == Path('tests/data/config.yml')
|
|
assert actual_args.regenerate == True
|
|
assert actual_args.config is not None
|
|
assert actual_args.config['content-type']['org']['input-files'] == ['~/first_from_config.org', '~/second_from_config.org']
|
|
assert actual_args.verbose == 3
|
|
|
|
|
|
# ----------------------------------------------------------------------------------------------------
|
|
def test_cli_config_from_cmd_args():
|
|
""
|
|
# Act
|
|
actual_args = cli(['--org-files=first.org'])
|
|
|
|
# Assert
|
|
assert actual_args.org_files == ['first.org']
|
|
assert actual_args.config_file is None
|
|
assert actual_args.config is not None
|
|
assert actual_args.config['content-type']['org']['input-files'] == ['first.org']
|
|
|
|
|
|
# ----------------------------------------------------------------------------------------------------
|
|
def test_cli_config_from_cmd_args_override_config_file():
|
|
# Act
|
|
actual_args = cli(['--config-file=tests/data/config.yml',
|
|
'--org-files=first.org'])
|
|
|
|
# Assert
|
|
assert actual_args.org_files == ['first.org']
|
|
assert actual_args.config_file == Path('tests/data/config.yml')
|
|
assert actual_args.config is not None
|
|
assert actual_args.config['content-type']['org']['input-files'] == ['first.org']
|