khoj/tests/test_cli.py

# 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']