CopilotKit/showcase/integrations/langgraph-python/tests/e2e/auth.spec.ts at main · bungphe/CopilotKit · 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
import { test, expect } from "@playwright/test";

/**
 * Auth demo lifecycle. The demo defaults to UNAUTHENTICATED on first
 * paint and renders SignInCard. After sign-in, <CopilotKit> mounts
 * with the bearer header attached and the chat boots. After sign-out,
 * the entire <CopilotKit> tree unmounts and SignInCard reappears.
 */
test.describe("Authentication", () => {
  test.beforeEach(async ({ page }) => {
    await page.goto("/demos/auth");
  });

  test("page loads unauthenticated with SignInCard visible", async ({
    page,
  }) => {
    await expect(
      page.locator('[data-testid="auth-sign-in-card"]'),
    ).toBeVisible();
    await expect(
      page.locator('[data-testid="auth-sign-in-button"]'),
    ).toBeEnabled();
    await expect(page.locator('[data-testid="auth-demo-token"]')).toBeVisible();
    // Chat surface and AuthBanner only render after sign-in.
    await expect(page.locator('[data-testid="auth-banner"]')).toHaveCount(0);
    await expect(page.getByPlaceholder("Type a message")).toHaveCount(0);
  });

  test("signing in mounts the chat surface with AuthBanner", async ({
    page,
  }) => {
    await page.locator('[data-testid="auth-sign-in-button"]').click();

    const banner = page.locator('[data-testid="auth-banner"]');
    await expect(banner).toBeVisible();
    await expect(banner).toHaveAttribute("data-authenticated", "true");
    await expect(page.locator('[data-testid="auth-status"]')).toContainText(
      "Signed in",
    );
    await expect(
      page.locator('[data-testid="auth-sign-out-button"]'),
    ).toBeEnabled();
    await expect(page.getByPlaceholder("Type a message")).toBeVisible();
    // SignInCard is gone once we're authenticated.
    await expect(page.locator('[data-testid="auth-sign-in-card"]')).toHaveCount(
      0,
    );
  });

  test("authenticated send produces an assistant response", async ({
    page,
  }) => {
    await page.locator('[data-testid="auth-sign-in-button"]').click();
    await expect(page.getByPlaceholder("Type a message")).toBeVisible();

    const input = page.getByPlaceholder("Type a message");
    await input.fill("Hello");
    await input.press("Enter");

    await expect(
      page.locator('[data-testid="copilot-assistant-message"]').first(),
    ).toBeVisible({ timeout: 30000 });
  });

  test("signing out unmounts the chat tree and re-renders SignInCard", async ({
    page,
  }) => {
    await page.locator('[data-testid="auth-sign-in-button"]').click();
    await expect(
      page.locator('[data-testid="auth-sign-out-button"]'),
    ).toBeVisible();

    await page.locator('[data-testid="auth-sign-out-button"]').click();

    // SignInCard re-mounts; banner and chat are gone.
    await expect(page.locator('[data-testid="auth-sign-in-card"]')).toBeVisible(
      { timeout: 5000 },
    );
    await expect(page.locator('[data-testid="auth-banner"]')).toHaveCount(0);
    await expect(page.getByPlaceholder("Type a message")).toHaveCount(0);
  });

  test("signing back in re-mounts a fresh chat surface", async ({ page }) => {
    // Sign in, sign out, sign in again.
    await page.locator('[data-testid="auth-sign-in-button"]').click();
    await expect(
      page.locator('[data-testid="auth-sign-out-button"]'),
    ).toBeVisible();
    await page.locator('[data-testid="auth-sign-out-button"]').click();
    await expect(page.locator('[data-testid="auth-sign-in-card"]')).toBeVisible(
      { timeout: 5000 },
    );

    await page.locator('[data-testid="auth-sign-in-button"]').click();
    const banner = page.locator('[data-testid="auth-banner"]');
    await expect(banner).toBeVisible({ timeout: 5000 });
    await expect(banner).toHaveAttribute("data-authenticated", "true");

    // Fresh chat surface accepts a send post-remount.
    const input = page.getByPlaceholder("Type a message");
    await input.fill("Hello again");
    await input.press("Enter");
    await expect(
      page.locator('[data-testid="copilot-assistant-message"]').first(),
    ).toBeVisible({ timeout: 30000 });
  });
});