dotbot/tests/dotbot_plugin_directory.py

"""Test that a plugin can be loaded by directory.

This file is copied to a location with the name "directory.py",
and is then loaded from within the `test_cli.py` code.
"""

import os.path
from typing import Any

import dotbot


class Directory(dotbot.Plugin):
    def can_handle(self, directive: str) -> bool:
        return directive == "plugin_directory"

    def handle(self, directive: str, data: Any) -> bool:
        if directive != "plugin_directory":
            msg = f"Directory cannot handle directive {directive}"
            raise ValueError(msg)

        if data != "no-check-context":
            self._log.debug("Attempting to get options from Context")
            options = self._context.options()
            if len(options.plugin_dirs) != 1:
                self._log.debug(f"Context.options.plugin_dirs length is {len(options.plugin_dirs)}, expected 1")
                return False

        with open(os.path.abspath(os.path.expanduser("~/flag-directory")), "w") as file:
            file.write("directory plugin loading works")
        return True