Add Handbook.md benchmark tasks

2026-06-24 12:44:34 -07:00
commit 72f9a65917
1800 changed files with 323589 additions and 0 deletions
@@ -0,0 +1,124 @@
+"""Tests for store policy tools."""
+
+import json
+
+import pytest
+
+from shopify import state as shopify_state
+from shopify.models import (
+    CreatePolicyArgs,
+    DeletePolicyArgs,
+    ListPoliciesArgs,
+    UpdatePolicyArgs,
+)
+from shopify.tools.policies import (
+    handle_create_policy,
+    handle_delete_policy,
+    handle_list_policies,
+    handle_update_policy,
+)
+
+
+@pytest.fixture
+def shopify_data(tmp_path):
+    data_file = tmp_path / "shopify_data.json"
+    data_file.write_text(
+        json.dumps(
+            {
+                "products": {},
+                "carts": {},
+                "orders": {},
+                "customers": {},
+                "collections": {},
+                "reviews": {},
+                "returns": {},
+                "discount_codes": {},
+                "policies": [
+                    {
+                        "id": "gid://shopify/ShopPolicy/1",
+                        "title": "Return Policy",
+                        "body": "<p>30-day returns on all items.</p>",
+                        "url": "https://shop.example.com/policies/1",
+                    }
+                ],
+                "counters": {
+                    "cart_id": 1000,
+                    "line_id": 1000,
+                    "order_id": 2000,
+                    "line_item_id": 3000,
+                    "customer_id": 4000,
+                    "collection_id": 5000,
+                    "review_id": 6000,
+                    "return_id": 7000,
+                    "product_id": 8000,
+                    "variant_id": 9000,
+                    "discount_id": 10000,
+                    "policy_id": 11000,
+                },
+            }
+        )
+    )
+    return data_file
+
+
+@pytest.fixture(autouse=True)
+def _patch_state(shopify_data, monkeypatch):
+    monkeypatch.setattr(shopify_state, "_STATE_FILE", shopify_data)
+    shopify_state._current_state = None
+    shopify_state._stores.clear()
+    shopify_state._active_store_id = "default"
+    shopify_state.load_state()
+
+
+class TestCreatePolicy:
+    def test_create(self):
+        result = handle_create_policy(CreatePolicyArgs(title="Shipping Policy", body="<p>Free shipping over $50.</p>"))
+        assert result["userErrors"] == []
+        p = result["policy"]
+        assert p["title"] == "Shipping Policy"
+        assert "Free shipping" in p["body"]
+        assert p["id"].startswith("gid://shopify/ShopPolicy/")
+
+    def test_create_adds_to_state(self):
+        handle_create_policy(CreatePolicyArgs(title="Privacy Policy", body="<p>We respect privacy.</p>"))
+        state = shopify_state.get_state()
+        assert len(state.policies) == 2
+
+
+class TestListPolicies:
+    def test_list(self):
+        result = handle_list_policies(ListPoliciesArgs())
+        assert result["totalCount"] == 1
+        assert result["policies"][0]["title"] == "Return Policy"
+
+
+class TestUpdatePolicy:
+    def test_update_title(self):
+        result = handle_update_policy(
+            UpdatePolicyArgs(policy_id="gid://shopify/ShopPolicy/1", title="Updated Return Policy")
+        )
+        assert result["userErrors"] == []
+        assert result["policy"]["title"] == "Updated Return Policy"
+
+    def test_update_body(self):
+        result = handle_update_policy(
+            UpdatePolicyArgs(policy_id="gid://shopify/ShopPolicy/1", body="<p>60-day returns.</p>")
+        )
+        assert "60-day" in result["policy"]["body"]
+
+    def test_update_nonexistent(self):
+        result = handle_update_policy(UpdatePolicyArgs(policy_id="nonexistent", title="X"))
+        assert result["policy"] is None
+        assert len(result["userErrors"]) == 1
+
+
+class TestDeletePolicy:
+    def test_delete(self):
+        result = handle_delete_policy(DeletePolicyArgs(policy_id="gid://shopify/ShopPolicy/1"))
+        assert result["deletedPolicyId"] == "gid://shopify/ShopPolicy/1"
+        state = shopify_state.get_state()
+        assert len(state.policies) == 0
+
+    def test_delete_nonexistent(self):
+        result = handle_delete_policy(DeletePolicyArgs(policy_id="nonexistent"))
+        assert result["deletedPolicyId"] is None