1import * as GCP from "@/GCP";
2import * as Test from "@/Test/Alchemy";
3import * as bqdp from "@distilled.cloud/gcp/bigquerydatapolicy_v2";
4import { expect } from "alchemy-test";
5import * as Effect from "effect/Effect";
6import { MinimumLogLevel } from "effect/References";
7import * as Schedule from "effect/Schedule";
8import { GcpEnvironment } from "@/GCP/Environment";
9
10const { test } = Test.make({ providers: GCP.providers() });
11
12const logLevel = Effect.provideService(
13 MinimumLogLevel,
14 process.env.DEBUG ? "Debug" : "Info",
15);
16const location = "us-central1";
17
18const waitUntilGone = (name: string) =>
19 bqdp.getProjectsLocationsDataPolicies({ name }).pipe(
20 Effect.as("found" as const),
21 Effect.catchTag("NotFound", () => Effect.succeed("gone" as const)),
22 Effect.repeat({
23 schedule: Schedule.spaced("1 second"),
24 until: (status) => status === "gone",
25 times: 10,
26 }),
27 );
28
29test.provider(
30 "getProjectsLocationsDataPolicies on a missing policy fails with a typed tag",
31 (stack) =>
32 Effect.gen(function* () {
33 const { project } = yield* GcpEnvironment.current;
34
35 yield* stack.destroy();
36
37 const error = yield* Effect.flip(
38 bqdp.getProjectsLocationsDataPolicies({
39 name: `projects/${project}/locations/${location}/dataPolicies/alchemy-missing-data-policy`,
40 }),
41 );
42 expect(error._tag).toEqual("NotFound");
43
44 yield* stack.destroy();
45 }).pipe(logLevel),
46 {
47 tags: ["provider:gcp", "provider:gcp:bigquerydatapolicy", "live"],
48 timeout: 90_000,
49 },
50);
51
52test.provider.skipIf(!!process.env.FAST)(
53 "create, update, and delete a data policy",
54 (stack) =>
55 Effect.gen(function* () {
56 const { project } = yield* GcpEnvironment.current;
57
58 yield* stack.destroy();
59
60 const created = yield* stack.deploy(
61 Effect.gen(function* () {
62 return yield* GCP.BigQueryDataPolicy.DataPolicy("Mask", {
63 location,
64 dataPolicyType: "DATA_MASKING_POLICY",
65 dataMaskingPolicy: { predefinedExpression: "SHA256" },
66 });
67 }),
68 );
69
70 expect(created.name).toContain("/dataPolicies/");
71 expect(created.dataPolicyId).toEqual(expect.any(String));
72 expect(created.dataPolicyId.startsWith("alch_")).toEqual(true);
73 expect(created.project).toEqual(project);
74 expect(created.location).toEqual(location);
75 expect(created.dataPolicyType).toEqual("DATA_MASKING_POLICY");
76 expect(created.dataMaskingPolicy?.predefinedExpression).toEqual("SHA256");
77
78 const fetched = yield* bqdp.getProjectsLocationsDataPolicies({
79 name: created.name,
80 });
81 expect(fetched.name).toEqual(created.name);
82 expect(fetched.dataPolicyId).toEqual(created.dataPolicyId);
83 expect(fetched.dataPolicyType).toEqual("DATA_MASKING_POLICY");
84 expect(fetched.dataMaskingPolicy?.predefinedExpression).toEqual("SHA256");
85
86 const updated = yield* stack.deploy(
87 Effect.gen(function* () {
88 return yield* GCP.BigQueryDataPolicy.DataPolicy("Mask", {
89 dataPolicyId: created.dataPolicyId,
90 location,
91 dataPolicyType: "DATA_MASKING_POLICY",
92 dataMaskingPolicy: { predefinedExpression: "ALWAYS_NULL" },
93 });
94 }),
95 );
96
97 expect(updated.name).toEqual(created.name);
98 expect(updated.dataPolicyId).toEqual(created.dataPolicyId);
99 expect(updated.dataMaskingPolicy?.predefinedExpression).toEqual(
100 "ALWAYS_NULL",
101 );
102
103 const refetched = yield* bqdp.getProjectsLocationsDataPolicies({
104 name: created.name,
105 });
106 expect(refetched.dataMaskingPolicy?.predefinedExpression).toEqual(
107 "ALWAYS_NULL",
108 );
109
110 yield* stack.destroy();
111 const gone = yield* waitUntilGone(created.name);
112 expect(gone).toEqual("gone");
113 }).pipe(logLevel),
114 {
115 tags: ["provider:gcp", "provider:gcp:bigquerydatapolicy", "live"],
116 timeout: 90_000,
117 },
118);