1import * as AWS from "alchemy/AWS";
2import * as Test from "alchemy/Test/Bun";
3import { expect } from "bun:test";
4import * as Console from "effect/Console";
5import * as Data from "effect/Data";
6import * as Effect from "effect/Effect";
7import * as Schedule from "effect/Schedule";
8import Stack from "../alchemy.run.ts";
9
10// A fresh CloudFront distribution (and the Lambda behind it) can serve
11// transient 404/5xx responses while it propagates. `Test.getWhenReady`
12// fails on that cold-start window and retries until the site serves a
13// real response.
14const { getWhenReady } = Test;
15
16class AssetNotReady extends Data.TaggedError("AssetNotReady")<{
17 body: string;
18}> {}
19
20// While the asset manifest and CloudFront edge caches are still
21// propagating, a 200 body can be stale — the status alone can't
22// distinguish "not yet" from "served", so retry until the body matches.
23const getBodyWhenReady = (url: string, expected: string) =>
24 Effect.gen(function* () {
25 const res = yield* getWhenReady(url);
26 expect(res.status).toBe(200);
27 const body = yield* res.text;
28 if (!body.includes(expected)) {
29 return yield* Effect.fail(new AssetNotReady({ body }));
30 }
31 return body;
32 }).pipe(
33 Effect.retry({
34 while: (error) => error instanceof AssetNotReady,
35 schedule: Schedule.max([
36 Schedule.min([
37 Schedule.exponential("500 millis"),
38 Schedule.spaced("3 seconds"),
39 ]),
40 Schedule.recurs(20),
41 ]),
42 }),
43 );
44
45const { test, beforeAll, afterAll, deploy, destroy } = Test.make({
46 providers: AWS.providers(),
47 state: AWS.state(),
48});
49
50// The first deploy runs the full Astro build AND creates a CloudFront
51// distribution (~5-10 minutes), so give the hook far more headroom than
52// the default 120s.
53const stack = beforeAll(deploy(Stack).pipe(Effect.tap(Console.log)), {
54 timeout: 1_200_000,
55});
56// Deleting the CloudFront distribution takes several minutes too.
57afterAll.skipIf(!!process.env.NO_DESTROY)(destroy(Stack), {
58 timeout: 1_200_000,
59});
60
61const base = Effect.map(stack, ({ url }) => {
62 if (!url) throw new Error("expected the site to expose a CloudFront url");
63 return String(url).replace(/\/+$/, "");
64});
65
66test(
67 "deploys and exposes a url",
68 Effect.gen(function* () {
69 const { url } = yield* stack;
70 expect(url).toBeString();
71 }),
72 { timeout: 180_000 },
73);
74
75test(
76 "serves the server-rendered home page",
77 Effect.gen(function* () {
78 const url = yield* base;
79 const res = yield* getWhenReady(url);
80 expect(res.status).toBe(200);
81 const html = yield* res.text;
82 // The `GREETING` env value from alchemy.run.ts, read via
83 // `process.env` in the page frontmatter — proves the Lambda
84 // rendered it at request time.
85 expect(html).toContain("Hello from Astro on AWS!");
86 expect(html).toContain("server-rendered in an AWS Lambda");
87 }),
88 { timeout: 180_000 },
89);
90
91test(
92 "serves the prerendered about page",
93 Effect.gen(function* () {
94 const url = yield* base;
95 const res = yield* getWhenReady(`${url}/about/`);
96 expect(res.status).toBe(200);
97 }),
98 { timeout: 180_000 },
99);
100
101test(
102 "compiles tailwind from astro.config.ts",
103 Effect.gen(function* () {
104 const url = yield* base;
105 // The utility class in the markup proves the page shipped with Tailwind
106 // classes; wait until the deployed HTML includes it.
107 const html = yield* getBodyWhenReady(url, "text-3xl");
108
109 // Astro either links an external compiled stylesheet or inlines small
110 // ones as a <style> block — accept both, but the compiled rule for the
111 // utility must be served either way. That rule only exists if the
112 // @tailwindcss/vite plugin from the project's own astro.config.ts ran.
113 const link = html.match(
114 /<link[^>]*rel="stylesheet"[^>]*href="([^"]+)"[^>]*>/,
115 );
116 if (link) {
117 const href = link[1]!;
118 const cssUrl = href.startsWith("http")
119 ? href
120 : `${url}${href.startsWith("/") ? "" : "/"}${href}`;
121 const css = yield* getBodyWhenReady(cssUrl, ".text-3xl");
122 expect(css).toContain(".text-3xl");
123 } else {
124 const style = html.match(/<style[^>]*>([\s\S]*?)<\/style>/);
125 expect(style).not.toBeNull();
126 expect(style![1]).toContain(".text-3xl");
127 }
128 }),
129 { timeout: 180_000 },
130);
131
132test(
133 "serves a static asset from public/",
134 Effect.gen(function* () {
135 const url = yield* base;
136 const body = yield* getBodyWhenReady(`${url}/robots.txt`, "User-agent: *");
137 expect(body).toContain("User-agent: *");
138 }),
139 { timeout: 180_000 },
140);