testPreviewURL/test_all_templates.py at main · edrouhardmicrosoft/testPreviewURL · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
"""Test code generation for all template types."""
from schemas.page import (
    PageSchema, PageMeta, StoryRef, ContainerType,
    TitleConfig, SideNavConfig, NavItem, NavGroup,
    EssentialsConfig, EssentialsField,
    CommandBar, CommandBarItem,
    ListTableTemplate, TableColumn, FilterConfig,
    CardsGridTemplate, CardItem,
    FormTemplate, FormField,
    DetailTemplate, DetailSection,
    CustomTemplate,
)
from schemas.codegen import generate_page


def test_cards_grid():
    page = PageSchema(
        meta=PageMeta(title="Dashboard Overview"),
        container="azure",
        title=TitleConfig(resourceName="My App", pageName="Overview", icon="App", breadcrumbs=["Home"]),
        template=CardsGridTemplate(
            kind="cards-grid",
            columns=3,
            cards=[
                CardItem(cardId="cpu", title="CPU Usage", description="85%"),
                CardItem(cardId="mem", title="Memory", description="60%"),
                CardItem(cardId="disk", title="Disk", description="45%"),
            ],
        ),
    )
    code = generate_page(page, "DashboardOverview")
    print("=== CARDS-GRID ===")
    start = code.index("const DashboardOverview")
    print(code[start:])
    return code


def test_form():
    page = PageSchema(
        meta=PageMeta(title="Create Resource"),
        container="azure",
        title=TitleConfig(resourceName="Create a resource", icon="Add"),
        template=FormTemplate(
            kind="form",
            fields=[
                FormField(name="name", label="Name", required=True, placeholder="Enter name"),
                FormField(name="region", label="Region", inputType="select", options=["East US", "West US"]),
            ],
            submitLabel="Review + Create",
        ),
    )
    code = generate_page(page, "CreateResource")
    print("=== FORM ===")
    start = code.index("const CreateResource")
    print(code[start:])
    return code


def test_detail_with_essentials():
    page = PageSchema(
        meta=PageMeta(title="Resource Detail"),
        container="azure",
        sideNav=SideNavConfig(
            defaultSelected="overview",
            entries=[
                NavItem(key="overview", label="Overview"),
                NavItem(key="properties", label="Properties"),
            ],
        ),
        title=TitleConfig(resourceName="my-db", pageName="Overview", icon="Database"),
        essentials=EssentialsConfig(
            fields=[
                EssentialsField(label="Status", value="Running"),
                EssentialsField(label="Location", value="East US"),
                EssentialsField(label="Resource group", value="prod-rg", copyable=True),
            ]
        ),
        template=DetailTemplate(
            kind="detail",
            sections=[
                DetailSection(sectionId="metrics", label="Metrics"),
                DetailSection(sectionId="activity", label="Activity Log"),
            ],
        ),
    )
    code = generate_page(page, "ResourceDetail")
    print("=== DETAIL + ESSENTIALS + SIDE NAV ===")
    start = code.index("const ResourceDetail")
    print(code[start:])
    return code


def test_custom():
    page = PageSchema(
        meta=PageMeta(title="Custom Page"),
        container="sre",
        template=CustomTemplate(
            kind="custom",
            stories=[
                StoryRef(storyId="components-button--primary", instanceId="hero-btn"),
            ],
        ),
    )
    code = generate_page(page, "CustomPage")
    print("=== CUSTOM (SRE container, no nav/title) ===")
    start = code.index("const CustomPage")
    print(code[start:])
    return code


if __name__ == "__main__":
    test_cards_grid()
    print()
    test_form()
    print()
    test_detail_with_essentials()
    print()
    test_custom()
    print("\n✓ All template types generated successfully")