DerekSurge
125 lines
4.0 KiB
Python
125 lines
4.0 KiB
Python
"""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
|