Initial commit

.codex-home/plugins/cache/openai-curated-remote/openai-templates/0.1.1/tests/test_openai_templates_plugin.py

@@ -0,0 +1,266 @@
+from __future__ import annotations
+
+import hashlib
+import io
+import json
+import re
+import struct
+import xml.etree.ElementTree as ET
+from pathlib import Path
+from zipfile import ZIP_DEFLATED, ZipFile
+
+REPO_ROOT = Path(__file__).resolve().parents[3]
+PLUGIN_ROOT = REPO_ROOT / "plugins" / "openai-templates"
+PNG_SIGNATURE = b"\x89PNG\r\n\x1a\n"
+ZIP_SIGNATURE = b"PK\x03\x04"
+MAX_PREVIEW_EDGE_PX = 800
+MAX_PLUGIN_ARCHIVE_BYTES = 26 * 1024 * 1024
+MAX_PLUGIN_ARCHIVE_ENTRIES = 5_000
+EXPECTED_BINARY_ASSET_SET_SHA256 = (
+    "ee1a1469054c77c71dd54a5bc36122fa679a9b639daddf51f20a057fcafac136"
+)
+TEMPLATES = (
+    (
+        "artifact-template-analytics-dashboard",
+        "Analytics Dashboard",
+        "spreadsheet",
+        "xlsx",
+    ),
+    ("artifact-template-business-review", "Business Review", "presentation", "pptx"),
+    (
+        "artifact-template-design-report",
+        "Design Report",
+        "document",
+        "docx",
+    ),
+    (
+        "artifact-template-experiment-analysis",
+        "Experiment Analysis",
+        "document",
+        "docx",
+    ),
+    (
+        "artifact-template-financial-budget",
+        "Financial Budget",
+        "spreadsheet",
+        "xlsx",
+    ),
+    (
+        "artifact-template-investment-committee-memo",
+        "Investment Committee Memo",
+        "document",
+        "docx",
+    ),
+    ("artifact-template-legal-memorandum", "Legal Memorandum", "document", "docx"),
+    (
+        "artifact-template-market-trends-report",
+        "Market Trends Report",
+        "presentation",
+        "pptx",
+    ),
+    (
+        "artifact-template-minimal-letterhead",
+        "Minimal Letterhead",
+        "document",
+        "docx",
+    ),
+    (
+        "artifact-template-operating-calendar",
+        "Operating Calendar",
+        "spreadsheet",
+        "xlsx",
+    ),
+    (
+        "artifact-template-operating-review",
+        "Operating Review",
+        "presentation",
+        "pptx",
+    ),
+    (
+        "artifact-template-project-kickoff",
+        "Project Kickoff",
+        "presentation",
+        "pptx",
+    ),
+    ("artifact-template-project-tracker", "Project Tracker", "spreadsheet", "xlsx"),
+    ("artifact-template-sales-pipeline", "Sales Pipeline", "spreadsheet", "xlsx"),
+    (
+        "artifact-template-simple-dark-mode",
+        "Simple Dark Mode",
+        "presentation",
+        "pptx",
+    ),
+    (
+        "artifact-template-simple-light-mode",
+        "Simple Light Mode",
+        "presentation",
+        "pptx",
+    ),
+    (
+        "artifact-template-strategy-memorandum",
+        "Strategy Memorandum",
+        "document",
+        "docx",
+    ),
+    ("artifact-template-system-design", "System Design", "document", "docx"),
+    (
+        "artifact-template-team-alignment",
+        "Team Alignment",
+        "presentation",
+        "pptx",
+    ),
+    (
+        "artifact-template-three-statement-forecast",
+        "Three-Statement Forecast",
+        "spreadsheet",
+        "xlsx",
+    ),
+)
+
+
+def test_openai_templates_plugin_contract() -> None:
+    manifest = json.loads((PLUGIN_ROOT / ".codex-plugin" / "plugin.json").read_text())
+    app_manifest = json.loads((PLUGIN_ROOT / ".app.json").read_text())
+    marketplace = json.loads((REPO_ROOT / "marketplace.json").read_text())
+
+    assert manifest["name"] == "openai-templates"
+    assert manifest["skills"] == "./skills/"
+    assert manifest["apps"] == "./.app.json"
+    assert manifest["interface"]["displayName"] == "Default templates"
+    assert manifest["interface"]["shortDescription"] == (
+        "Default templates for documents, spreadsheets, and presentations"
+    )
+    assert manifest["interface"]["defaultPrompt"] == [
+        "Create a document using the Minimal Letterhead template",
+        "Create an Analytics Dashboard spreadsheet using the Analytics Dashboard template",
+        "Create a planning presentation with the Team Alignment template",
+    ]
+    assert app_manifest == {
+        "apps": {
+            "default_templates": {
+                "id": "connector_openai_default_templates",
+                "required": True,
+            },
+        }
+    }
+    assert {
+        "name": "openai-templates",
+        "source": {
+            "source": "local",
+            "path": "./plugins/openai-templates",
+        },
+        "policy": {
+            "installation": "AVAILABLE",
+            "authentication": "ON_USE",
+        },
+        "category": "Productivity",
+    } in marketplace["plugins"]
+
+
+def test_openai_templates_are_packaged_for_chatgpt() -> None:
+    package_files = sorted(
+        path
+        for path in PLUGIN_ROOT.rglob("*")
+        if path.is_file() and "__pycache__" not in path.parts and path.suffix != ".pyc"
+    )
+    assert len(package_files) < MAX_PLUGIN_ARCHIVE_ENTRIES
+
+    archive_buffer = io.BytesIO()
+    with ZipFile(archive_buffer, "w", compression=ZIP_DEFLATED) as archive:
+        for path in package_files:
+            archive.write(path, path.relative_to(PLUGIN_ROOT))
+    assert archive_buffer.tell() < MAX_PLUGIN_ARCHIVE_BYTES
+
+    assert sorted(path.name for path in (PLUGIN_ROOT / "skills").iterdir()) == sorted(
+        name for name, _display_name, _kind, _extension in TEMPLATES
+    )
+    assert {
+        kind: sum(template_kind == kind for _, _, template_kind, _ in TEMPLATES)
+        for kind in ("document", "presentation", "spreadsheet")
+    } == {"document": 7, "presentation": 7, "spreadsheet": 6}
+
+    binary_asset_set = hashlib.sha256()
+    for name, display_name, kind, extension in TEMPLATES:
+        skill_root = PLUGIN_ROOT / "skills" / name
+        manifest = json.loads((skill_root / "artifact-template.json").read_text())
+        metadata = (skill_root / "agents" / "openai.yaml").read_text()
+        skill = (skill_root / "SKILL.md").read_text()
+        assets_root = skill_root / "assets"
+        preview_path = assets_root / "preview.png"
+        reference_path = assets_root / f"reference.{extension}"
+
+        assert manifest == {
+            "schemaVersion": 1,
+            "kind": kind,
+            "reference": f"assets/reference.{extension}",
+            "preview": "assets/preview.png",
+        }
+        assert sorted(path.name for path in assets_root.iterdir()) == [
+            "preview.png",
+            f"reference.{extension}",
+        ]
+        assert f"\nname: {name}\n" in skill
+        assert f'description: "Create a {kind} using the {display_name} template' in skill
+        assert f'display_name: "{display_name}"' in metadata
+        assert 'icon_large: "./assets/preview.png"' in metadata
+        assert f'default_prompt: "Create a new {kind} with this template."' in metadata
+        assert metadata.endswith("policy:\n  allow_implicit_invocation: false\n")
+        assert "plugin://" not in skill
+        assert "prompt-advertised preinstalled" in skill
+        assert "artifact_handoff.prepare_artifact_generation" not in skill
+        preview = preview_path.read_bytes()
+        assert preview.startswith(PNG_SIGNATURE)
+        width, height = struct.unpack(">II", preview[16:24])
+        assert width > 0 and height > 0
+        assert max(width, height) <= MAX_PREVIEW_EDGE_PX
+
+        reference = reference_path.read_bytes()
+        assert reference.startswith(ZIP_SIGNATURE)
+        _assert_sanitized_office_archive(reference, name)
+
+        for binary_path, contents in (
+            (preview_path, preview),
+            (reference_path, reference),
+        ):
+            binary_asset_set.update(binary_path.relative_to(PLUGIN_ROOT).as_posix().encode())
+            binary_asset_set.update(b"\0")
+            binary_asset_set.update(hashlib.sha256(contents).digest())
+
+    assert binary_asset_set.hexdigest() == EXPECTED_BINARY_ASSET_SET_SHA256
+
+
+def _assert_sanitized_office_archive(archive: bytes, label: str) -> None:
+    with ZipFile(io.BytesIO(archive)) as package:
+        names = package.namelist()
+        assert all(not name.lower().startswith("customxml/") for name in names)
+
+        for name in names:
+            contents = package.read(name)
+            if name == "docProps/core.xml":
+                root = ET.fromstring(contents)
+                private_fields = {
+                    "creator",
+                    "lastModifiedBy",
+                    "created",
+                    "modified",
+                    "lastPrinted",
+                    "revision",
+                }
+                for element in root.iter():
+                    local_name = element.tag.rsplit("}", 1)[-1]
+                    if local_name in private_fields:
+                        assert not (element.text or "").strip(), f"{label} retains {local_name}"
+
+            if name.endswith((".xml", ".rels")):
+                assert not re.search(rb"\bw:rsid(?:[A-Za-z]+)?=", contents), (
+                    f"{label}:{name} retains Word editing metadata"
+                )
+                assert b"x15ac:absPath" not in contents
+                assert not re.search(
+                    rb"(?:file:///)?(?:/Users/|/home/|[A-Za-z]:[\\/](?:Users|Documents and Settings)[\\/])",
+                    contents,
+                    re.IGNORECASE,
+                ), f"{label}:{name} retains a local path"
+
+            if contents.startswith(ZIP_SIGNATURE):
+                _assert_sanitized_office_archive(contents, f"{label}:{name}")