Class TestAtFileExpansion

tests/test_manager.py:2499–2626

Tests that @file arguments are expanded through the CLI.

class TestAtFileExpansion(unittest.TestCase):
    """Tests that @file arguments are expanded through the CLI."""

    @patch("explainshell.extraction.common.gz_sha256", side_effect=lambda p: p)
    @patch("os.path.islink", return_value=False)
    @patch(
        "explainshell.manager.config.source_from_path",
        side_effect=lambda p: f"fake/release/1/{os.path.basename(p)}",
    )
    def test_extract_expands_at_file(
        self,
        _mock_source: MagicMock,
        _mock_link: MagicMock,
        _mock_sha: MagicMock,
    ) -> None:
        """@file arg is expanded to the file's contents and reaches the classifier."""
        with tempfile.NamedTemporaryFile(mode="w", suffix=".txt", delete=False) as f:
            f.write("/fake/echo.1.gz\n")
            f.flush()
            list_path = f.name

        try:
            with _temp_db() as db_path:
                Store.create(db_path).close()
                runner = CliRunner()
                with self.assertLogs("explainshell.manager", level="INFO") as cm:
                    result = runner.invoke(
                        cli,
                        [
                            "--db",
                            db_path,
                            "extract",
                            "--mode",
                            "llm:test-model",
                            "--dry-run",
                            f"@{list_path}",
                        ],
                    )

            self.assertEqual(result.exit_code, 0, result.output)
            self.assertIn("fake/release/1/echo.1.gz", "\n".join(cm.output))
        finally:
            os.unlink(list_path)

    @patch("explainshell.extraction.common.gz_sha256", side_effect=lambda p: p)
    @patch("os.path.islink", return_value=False)
    @patch(
        "explainshell.manager.config.source_from_path",
        side_effect=lambda p: f"fake/release/1/{os.path.basename(p)}",
    )
    def test_extract_at_file_skips_blanks_and_comments(
        self,
        _mock_source: MagicMock,
        _mock_link: MagicMock,
        _mock_sha: MagicMock,
    ) -> None:
        with tempfile.NamedTemporaryFile(mode="w", suffix=".txt", delete=False) as f:
            f.write("/fake/echo.1.gz\n\n# comment\n  \n")