wrangler 2.0.7 → 2.0.11
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +1 -1
- package/bin/wrangler.js +16 -4
- package/package.json +2 -2
- package/src/__tests__/configuration.test.ts +165 -70
- package/src/__tests__/dev.test.tsx +158 -66
- package/src/__tests__/helpers/mock-dialogs.ts +41 -1
- package/src/__tests__/init.test.ts +191 -111
- package/src/__tests__/kv.test.ts +8 -8
- package/src/__tests__/package-manager.test.ts +154 -7
- package/src/__tests__/pages.test.ts +115 -18
- package/src/__tests__/publish.test.ts +431 -140
- package/src/__tests__/secret.test.ts +4 -4
- package/src/__tests__/whoami.test.tsx +34 -0
- package/src/cfetch/index.ts +17 -2
- package/src/cfetch/internal.ts +12 -9
- package/src/config/config.ts +1 -1
- package/src/config/validation-helpers.ts +10 -1
- package/src/config/validation.ts +59 -33
- package/src/create-worker-preview.ts +15 -15
- package/src/dev/dev.tsx +4 -15
- package/src/dev/remote.tsx +26 -16
- package/src/dialogs.tsx +48 -0
- package/src/index.tsx +181 -167
- package/src/package-manager.ts +50 -3
- package/src/pages.tsx +298 -228
- package/src/publish.ts +148 -15
- package/src/sites.tsx +52 -14
- package/src/user.tsx +12 -1
- package/src/whoami.tsx +3 -2
- package/src/worker.ts +2 -1
- package/src/zones.ts +73 -0
- package/templates/new-worker-scheduled.js +17 -0
- package/templates/new-worker-scheduled.ts +32 -0
- package/wrangler-dist/cli.js +707 -407
|
@@ -3,7 +3,6 @@ import getPort from "get-port";
|
|
|
3
3
|
import patchConsole from "patch-console";
|
|
4
4
|
import dedent from "ts-dedent";
|
|
5
5
|
import Dev from "../dev/dev";
|
|
6
|
-
import { mockAccountId, mockApiToken } from "./helpers/mock-account-id";
|
|
7
6
|
import { setMockResponse, unsetAllMocks } from "./helpers/mock-cfetch";
|
|
8
7
|
import { mockConsoleMethods } from "./helpers/mock-console";
|
|
9
8
|
import { runInTempDir } from "./helpers/run-in-tmp";
|
|
@@ -11,8 +10,6 @@ import { runWrangler } from "./helpers/run-wrangler";
|
|
|
11
10
|
import writeWranglerToml from "./helpers/write-wrangler-toml";
|
|
12
11
|
|
|
13
12
|
describe("wrangler dev", () => {
|
|
14
|
-
mockAccountId();
|
|
15
|
-
mockApiToken();
|
|
16
13
|
runInTempDir();
|
|
17
14
|
const std = mockConsoleMethods();
|
|
18
15
|
afterEach(() => {
|
|
@@ -83,7 +80,7 @@ describe("wrangler dev", () => {
|
|
|
83
80
|
|
|
84
81
|
expect(std.out).toMatchInlineSnapshot(`
|
|
85
82
|
"
|
|
86
|
-
[32mIf you think this is a bug then please create an issue at https://github.com/cloudflare/wrangler2/issues/new
|
|
83
|
+
[32mIf you think this is a bug then please create an issue at https://github.com/cloudflare/wrangler2/issues/new/choose[0m"
|
|
87
84
|
`);
|
|
88
85
|
expect(std.err).toMatchInlineSnapshot(`
|
|
89
86
|
"[31mX [41;31m[[41;97mERROR[41;31m][0m [1mMissing entry-point: The entry-point should be specified via the command line (e.g. \`wrangler dev path/to/script\`) or the \`main\` config field.[0m
|
|
@@ -152,10 +149,12 @@ describe("wrangler dev", () => {
|
|
|
152
149
|
fs.writeFileSync("index.js", `export default {};`);
|
|
153
150
|
mockGetZones("some-host.com", [{ id: "some-zone-id" }]);
|
|
154
151
|
await runWrangler("dev --host some-host.com");
|
|
155
|
-
expect((Dev as jest.Mock).mock.calls[0][0]
|
|
156
|
-
|
|
157
|
-
|
|
158
|
-
|
|
152
|
+
expect((Dev as jest.Mock).mock.calls[0][0]).toEqual(
|
|
153
|
+
expect.objectContaining({
|
|
154
|
+
host: "some-host.com",
|
|
155
|
+
zone: "some-zone-id",
|
|
156
|
+
})
|
|
157
|
+
);
|
|
159
158
|
});
|
|
160
159
|
|
|
161
160
|
it("should read wrangler.toml's dev.host", async () => {
|
|
@@ -168,9 +167,7 @@ describe("wrangler dev", () => {
|
|
|
168
167
|
fs.writeFileSync("index.js", `export default {};`);
|
|
169
168
|
mockGetZones("some-host.com", [{ id: "some-zone-id" }]);
|
|
170
169
|
await runWrangler("dev");
|
|
171
|
-
expect((Dev as jest.Mock).mock.calls[0][0].
|
|
172
|
-
"some-host.com"
|
|
173
|
-
);
|
|
170
|
+
expect((Dev as jest.Mock).mock.calls[0][0].host).toEqual("some-host.com");
|
|
174
171
|
});
|
|
175
172
|
|
|
176
173
|
it("should read --route", async () => {
|
|
@@ -180,9 +177,7 @@ describe("wrangler dev", () => {
|
|
|
180
177
|
fs.writeFileSync("index.js", `export default {};`);
|
|
181
178
|
mockGetZones("some-host.com", [{ id: "some-zone-id" }]);
|
|
182
179
|
await runWrangler("dev --route http://some-host.com/some/path/*");
|
|
183
|
-
expect((Dev as jest.Mock).mock.calls[0][0].
|
|
184
|
-
"some-host.com"
|
|
185
|
-
);
|
|
180
|
+
expect((Dev as jest.Mock).mock.calls[0][0].host).toEqual("some-host.com");
|
|
186
181
|
});
|
|
187
182
|
|
|
188
183
|
it("should read wrangler.toml's routes", async () => {
|
|
@@ -196,9 +191,7 @@ describe("wrangler dev", () => {
|
|
|
196
191
|
fs.writeFileSync("index.js", `export default {};`);
|
|
197
192
|
mockGetZones("some-host.com", [{ id: "some-zone-id" }]);
|
|
198
193
|
await runWrangler("dev");
|
|
199
|
-
expect((Dev as jest.Mock).mock.calls[0][0].
|
|
200
|
-
"some-host.com"
|
|
201
|
-
);
|
|
194
|
+
expect((Dev as jest.Mock).mock.calls[0][0].host).toEqual("some-host.com");
|
|
202
195
|
});
|
|
203
196
|
|
|
204
197
|
it("should read wrangler.toml's environment specific routes", async () => {
|
|
@@ -220,9 +213,7 @@ describe("wrangler dev", () => {
|
|
|
220
213
|
fs.writeFileSync("index.js", `export default {};`);
|
|
221
214
|
mockGetZones("some-host.com", [{ id: "some-zone-id" }]);
|
|
222
215
|
await runWrangler("dev --env staging");
|
|
223
|
-
expect((Dev as jest.Mock).mock.calls[0][0].
|
|
224
|
-
"some-host.com"
|
|
225
|
-
);
|
|
216
|
+
expect((Dev as jest.Mock).mock.calls[0][0].host).toEqual("some-host.com");
|
|
226
217
|
});
|
|
227
218
|
|
|
228
219
|
it("should strip leading `*` from given host when deducing a zone id", async () => {
|
|
@@ -233,9 +224,7 @@ describe("wrangler dev", () => {
|
|
|
233
224
|
fs.writeFileSync("index.js", `export default {};`);
|
|
234
225
|
mockGetZones("some-host.com", [{ id: "some-zone-id" }]);
|
|
235
226
|
await runWrangler("dev");
|
|
236
|
-
expect((Dev as jest.Mock).mock.calls[0][0].
|
|
237
|
-
"some-host.com"
|
|
238
|
-
);
|
|
227
|
+
expect((Dev as jest.Mock).mock.calls[0][0].host).toEqual("some-host.com");
|
|
239
228
|
});
|
|
240
229
|
|
|
241
230
|
it("should strip leading `*.` from given host when deducing a zone id", async () => {
|
|
@@ -246,9 +235,7 @@ describe("wrangler dev", () => {
|
|
|
246
235
|
fs.writeFileSync("index.js", `export default {};`);
|
|
247
236
|
mockGetZones("some-host.com", [{ id: "some-zone-id" }]);
|
|
248
237
|
await runWrangler("dev");
|
|
249
|
-
expect((Dev as jest.Mock).mock.calls[0][0].
|
|
250
|
-
"some-host.com"
|
|
251
|
-
);
|
|
238
|
+
expect((Dev as jest.Mock).mock.calls[0][0].host).toEqual("some-host.com");
|
|
252
239
|
});
|
|
253
240
|
|
|
254
241
|
it("should, when provided, use a configured zone_id", async () => {
|
|
@@ -260,10 +247,12 @@ describe("wrangler dev", () => {
|
|
|
260
247
|
});
|
|
261
248
|
fs.writeFileSync("index.js", `export default {};`);
|
|
262
249
|
await runWrangler("dev");
|
|
263
|
-
expect((Dev as jest.Mock).mock.calls[0][0]
|
|
264
|
-
|
|
265
|
-
|
|
266
|
-
|
|
250
|
+
expect((Dev as jest.Mock).mock.calls[0][0]).toEqual(
|
|
251
|
+
expect.objectContaining({
|
|
252
|
+
host: "some-domain.com",
|
|
253
|
+
zone: "some-zone-id",
|
|
254
|
+
})
|
|
255
|
+
);
|
|
267
256
|
});
|
|
268
257
|
|
|
269
258
|
it("should, when provided, use a zone_name to get a zone_id", async () => {
|
|
@@ -276,11 +265,13 @@ describe("wrangler dev", () => {
|
|
|
276
265
|
fs.writeFileSync("index.js", `export default {};`);
|
|
277
266
|
mockGetZones("some-zone.com", [{ id: "a-zone-id" }]);
|
|
278
267
|
await runWrangler("dev");
|
|
279
|
-
expect((Dev as jest.Mock).mock.calls[0][0]
|
|
280
|
-
|
|
281
|
-
|
|
282
|
-
|
|
283
|
-
|
|
268
|
+
expect((Dev as jest.Mock).mock.calls[0][0]).toEqual(
|
|
269
|
+
expect.objectContaining({
|
|
270
|
+
// note that it uses the provided zone_name as a host too
|
|
271
|
+
host: "some-zone.com",
|
|
272
|
+
zone: "a-zone-id",
|
|
273
|
+
})
|
|
274
|
+
);
|
|
284
275
|
});
|
|
285
276
|
|
|
286
277
|
it("given a long host, it should use the longest subdomain that resolves to a zone", async () => {
|
|
@@ -292,10 +283,12 @@ describe("wrangler dev", () => {
|
|
|
292
283
|
mockGetZones("222.333.some-host.com", []);
|
|
293
284
|
mockGetZones("333.some-host.com", [{ id: "some-zone-id" }]);
|
|
294
285
|
await runWrangler("dev --host 111.222.333.some-host.com");
|
|
295
|
-
expect((Dev as jest.Mock).mock.calls[0][0]
|
|
296
|
-
|
|
297
|
-
|
|
298
|
-
|
|
286
|
+
expect((Dev as jest.Mock).mock.calls[0][0]).toEqual(
|
|
287
|
+
expect.objectContaining({
|
|
288
|
+
host: "111.222.333.some-host.com",
|
|
289
|
+
zone: "some-zone-id",
|
|
290
|
+
})
|
|
291
|
+
);
|
|
299
292
|
});
|
|
300
293
|
|
|
301
294
|
it("should, in order, use args.host/config.dev.host/args.routes/(config.route|config.routes)", async () => {
|
|
@@ -310,10 +303,12 @@ describe("wrangler dev", () => {
|
|
|
310
303
|
routes: ["http://5.some-host.com/some/path/*"],
|
|
311
304
|
});
|
|
312
305
|
await runWrangler("dev");
|
|
313
|
-
expect((Dev as jest.Mock).mock.calls[0][0]
|
|
314
|
-
|
|
315
|
-
|
|
316
|
-
|
|
306
|
+
expect((Dev as jest.Mock).mock.calls[0][0]).toEqual(
|
|
307
|
+
expect.objectContaining({
|
|
308
|
+
host: "5.some-host.com",
|
|
309
|
+
zone: "some-zone-id-5",
|
|
310
|
+
})
|
|
311
|
+
);
|
|
317
312
|
(Dev as jest.Mock).mockClear();
|
|
318
313
|
|
|
319
314
|
// config.route
|
|
@@ -323,10 +318,12 @@ describe("wrangler dev", () => {
|
|
|
323
318
|
route: "https://4.some-host.com/some/path/*",
|
|
324
319
|
});
|
|
325
320
|
await runWrangler("dev");
|
|
326
|
-
expect((Dev as jest.Mock).mock.calls[0][0]
|
|
327
|
-
|
|
328
|
-
|
|
329
|
-
|
|
321
|
+
expect((Dev as jest.Mock).mock.calls[0][0]).toEqual(
|
|
322
|
+
expect.objectContaining({
|
|
323
|
+
host: "4.some-host.com",
|
|
324
|
+
zone: "some-zone-id-4",
|
|
325
|
+
})
|
|
326
|
+
);
|
|
330
327
|
(Dev as jest.Mock).mockClear();
|
|
331
328
|
|
|
332
329
|
// --routes
|
|
@@ -336,10 +333,12 @@ describe("wrangler dev", () => {
|
|
|
336
333
|
route: "https://4.some-host.com/some/path/*",
|
|
337
334
|
});
|
|
338
335
|
await runWrangler("dev --routes http://3.some-host.com/some/path/*");
|
|
339
|
-
expect((Dev as jest.Mock).mock.calls[0][0]
|
|
340
|
-
|
|
341
|
-
|
|
342
|
-
|
|
336
|
+
expect((Dev as jest.Mock).mock.calls[0][0]).toEqual(
|
|
337
|
+
expect.objectContaining({
|
|
338
|
+
host: "3.some-host.com",
|
|
339
|
+
zone: "some-zone-id-3",
|
|
340
|
+
})
|
|
341
|
+
);
|
|
343
342
|
(Dev as jest.Mock).mockClear();
|
|
344
343
|
|
|
345
344
|
// config.dev.host
|
|
@@ -352,10 +351,12 @@ describe("wrangler dev", () => {
|
|
|
352
351
|
route: "4.some-host.com/some/path/*",
|
|
353
352
|
});
|
|
354
353
|
await runWrangler("dev --routes http://3.some-host.com/some/path/*");
|
|
355
|
-
expect((Dev as jest.Mock).mock.calls[0][0]
|
|
356
|
-
|
|
357
|
-
|
|
358
|
-
|
|
354
|
+
expect((Dev as jest.Mock).mock.calls[0][0]).toEqual(
|
|
355
|
+
expect.objectContaining({
|
|
356
|
+
host: "2.some-host.com",
|
|
357
|
+
zone: "some-zone-id-2",
|
|
358
|
+
})
|
|
359
|
+
);
|
|
359
360
|
(Dev as jest.Mock).mockClear();
|
|
360
361
|
|
|
361
362
|
// --host
|
|
@@ -370,10 +371,12 @@ describe("wrangler dev", () => {
|
|
|
370
371
|
await runWrangler(
|
|
371
372
|
"dev --routes http://3.some-host.com/some/path/* --host 1.some-host.com"
|
|
372
373
|
);
|
|
373
|
-
expect((Dev as jest.Mock).mock.calls[0][0]
|
|
374
|
-
|
|
375
|
-
|
|
376
|
-
|
|
374
|
+
expect((Dev as jest.Mock).mock.calls[0][0]).toEqual(
|
|
375
|
+
expect.objectContaining({
|
|
376
|
+
host: "1.some-host.com",
|
|
377
|
+
zone: "some-zone-id-1",
|
|
378
|
+
})
|
|
379
|
+
);
|
|
377
380
|
(Dev as jest.Mock).mockClear();
|
|
378
381
|
});
|
|
379
382
|
|
|
@@ -467,7 +470,7 @@ describe("wrangler dev", () => {
|
|
|
467
470
|
expect(std.out).toMatchInlineSnapshot(`
|
|
468
471
|
"Running custom build: node -e \\"console.log('custom build');\\"
|
|
469
472
|
|
|
470
|
-
[32mIf you think this is a bug then please create an issue at https://github.com/cloudflare/wrangler2/issues/new
|
|
473
|
+
[32mIf you think this is a bug then please create an issue at https://github.com/cloudflare/wrangler2/issues/new/choose[0m"
|
|
471
474
|
`);
|
|
472
475
|
expect(std.err).toMatchInlineSnapshot(`
|
|
473
476
|
"[31mX [41;31m[[41;97mERROR[41;31m][0m [1mThe expected output file at \\"index.js\\" was not found after running custom build: node -e \\"console.log('custom build');\\".[0m
|
|
@@ -698,13 +701,28 @@ describe("wrangler dev", () => {
|
|
|
698
701
|
fs.writeFileSync("index.js", `export default {};`);
|
|
699
702
|
await runWrangler("dev");
|
|
700
703
|
expect((Dev as jest.Mock).mock.calls[0][0].ip).toEqual("localhost");
|
|
701
|
-
expect(std.out).toMatchInlineSnapshot(`
|
|
704
|
+
expect(std.out).toMatchInlineSnapshot(`
|
|
705
|
+
"Your worker has access to the following bindings:
|
|
706
|
+
- Durable Objects:
|
|
707
|
+
- NAME_1: CLASS_1
|
|
708
|
+
- NAME_2: CLASS_2 (defined in SCRIPT_A)
|
|
709
|
+
- NAME_3: CLASS_3
|
|
710
|
+
- NAME_4: CLASS_4 (defined in SCRIPT_B)"
|
|
711
|
+
`);
|
|
702
712
|
expect(std.warn).toMatchInlineSnapshot(`
|
|
703
713
|
"[33m▲ [43;33m[[43;30mWARNING[43;33m][0m [1mProcessing wrangler.toml configuration:[0m
|
|
704
714
|
|
|
705
715
|
- In wrangler.toml, you have configured [durable_objects] exported by this Worker (CLASS_1,
|
|
706
716
|
CLASS_3), but no [migrations] for them. This may not work as expected until you add a [migrations]
|
|
707
|
-
section to your wrangler.toml.
|
|
717
|
+
section to your wrangler.toml. Add this configuration to your wrangler.toml:
|
|
718
|
+
|
|
719
|
+
\`\`\`
|
|
720
|
+
[[migrations]]
|
|
721
|
+
tag = \\"v1\\" # Should be unique for each entry
|
|
722
|
+
new_classes = [\\"CLASS_1\\", \\"CLASS_3\\"]
|
|
723
|
+
\`\`\`
|
|
724
|
+
|
|
725
|
+
Refer to
|
|
708
726
|
[4mhttps://developers.cloudflare.com/workers/learning/using-durable-objects/#durable-object-migrations-in-wranglertoml[0m
|
|
709
727
|
for more details.
|
|
710
728
|
|
|
@@ -764,9 +782,18 @@ describe("wrangler dev", () => {
|
|
|
764
782
|
EMPTY: "",
|
|
765
783
|
UNQUOTED: "unquoted value", // Note that whitespace is trimmed
|
|
766
784
|
});
|
|
767
|
-
expect(std.out).toMatchInlineSnapshot(
|
|
768
|
-
|
|
769
|
-
|
|
785
|
+
expect(std.out).toMatchInlineSnapshot(`
|
|
786
|
+
"Using vars defined in .dev.vars
|
|
787
|
+
Your worker has access to the following bindings:
|
|
788
|
+
- Vars:
|
|
789
|
+
- VAR_1: \\"(hidden)\\"
|
|
790
|
+
- VAR_2: \\"original value 2\\"
|
|
791
|
+
- VAR_3: \\"(hidden)\\"
|
|
792
|
+
- VAR_MULTI_LINE_1: \\"(hidden)\\"
|
|
793
|
+
- VAR_MULTI_LINE_2: \\"(hidden)\\"
|
|
794
|
+
- EMPTY: \\"(hidden)\\"
|
|
795
|
+
- UNQUOTED: \\"(hidden)\\""
|
|
796
|
+
`);
|
|
770
797
|
expect(std.warn).toMatchInlineSnapshot(`""`);
|
|
771
798
|
expect(std.err).toMatchInlineSnapshot(`""`);
|
|
772
799
|
});
|
|
@@ -862,7 +889,41 @@ describe("wrangler dev", () => {
|
|
|
862
889
|
Object {
|
|
863
890
|
"debug": "",
|
|
864
891
|
"err": "",
|
|
865
|
-
"out": "
|
|
892
|
+
"out": "Your worker has access to the following bindings:
|
|
893
|
+
- Services:
|
|
894
|
+
- WorkerA: A
|
|
895
|
+
- WorkerB: B - staging",
|
|
896
|
+
"warn": "[33m▲ [43;33m[[43;30mWARNING[43;33m][0m [1mProcessing wrangler.toml configuration:[0m
|
|
897
|
+
|
|
898
|
+
- \\"services\\" fields are experimental and may change or break at any time.
|
|
899
|
+
|
|
900
|
+
|
|
901
|
+
[33m▲ [43;33m[[43;30mWARNING[43;33m][0m [1mThis worker is bound to live services: WorkerA (A), WorkerB (B@staging)[0m
|
|
902
|
+
|
|
903
|
+
",
|
|
904
|
+
}
|
|
905
|
+
`);
|
|
906
|
+
});
|
|
907
|
+
});
|
|
908
|
+
|
|
909
|
+
describe("print bindings", () => {
|
|
910
|
+
it("should print bindings", async () => {
|
|
911
|
+
writeWranglerToml({
|
|
912
|
+
services: [
|
|
913
|
+
{ binding: "WorkerA", service: "A" },
|
|
914
|
+
{ binding: "WorkerB", service: "B", environment: "staging" },
|
|
915
|
+
],
|
|
916
|
+
});
|
|
917
|
+
fs.writeFileSync("index.js", `export default {};`);
|
|
918
|
+
await runWrangler("dev index.js");
|
|
919
|
+
expect(std).toMatchInlineSnapshot(`
|
|
920
|
+
Object {
|
|
921
|
+
"debug": "",
|
|
922
|
+
"err": "",
|
|
923
|
+
"out": "Your worker has access to the following bindings:
|
|
924
|
+
- Services:
|
|
925
|
+
- WorkerA: A
|
|
926
|
+
- WorkerB: B - staging",
|
|
866
927
|
"warn": "[33m▲ [43;33m[[43;30mWARNING[43;33m][0m [1mProcessing wrangler.toml configuration:[0m
|
|
867
928
|
|
|
868
929
|
- \\"services\\" fields are experimental and may change or break at any time.
|
|
@@ -874,6 +935,37 @@ describe("wrangler dev", () => {
|
|
|
874
935
|
}
|
|
875
936
|
`);
|
|
876
937
|
});
|
|
938
|
+
|
|
939
|
+
it("should mask vars that were overriden in .dev.vars", async () => {
|
|
940
|
+
writeWranglerToml({
|
|
941
|
+
vars: {
|
|
942
|
+
variable: 123,
|
|
943
|
+
overriden: "original values",
|
|
944
|
+
},
|
|
945
|
+
});
|
|
946
|
+
fs.writeFileSync(
|
|
947
|
+
".dev.vars",
|
|
948
|
+
`
|
|
949
|
+
SECRET = "A secret"
|
|
950
|
+
overriden = "overriden value"
|
|
951
|
+
`
|
|
952
|
+
);
|
|
953
|
+
fs.writeFileSync("index.js", `export default {};`);
|
|
954
|
+
await runWrangler("dev index.js");
|
|
955
|
+
expect(std).toMatchInlineSnapshot(`
|
|
956
|
+
Object {
|
|
957
|
+
"debug": "",
|
|
958
|
+
"err": "",
|
|
959
|
+
"out": "Using vars defined in .dev.vars
|
|
960
|
+
Your worker has access to the following bindings:
|
|
961
|
+
- Vars:
|
|
962
|
+
- variable: \\"123\\"
|
|
963
|
+
- overriden: \\"(hidden)\\"
|
|
964
|
+
- SECRET: \\"(hidden)\\"",
|
|
965
|
+
"warn": "",
|
|
966
|
+
}
|
|
967
|
+
`);
|
|
968
|
+
});
|
|
877
969
|
});
|
|
878
970
|
});
|
|
879
971
|
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { confirm, prompt } from "../../dialogs";
|
|
1
|
+
import { confirm, prompt, select } from "../../dialogs";
|
|
2
2
|
import { normalizeSlashes } from "./mock-console";
|
|
3
3
|
|
|
4
4
|
/**
|
|
@@ -88,3 +88,43 @@ export function clearPromptMocks() {
|
|
|
88
88
|
);
|
|
89
89
|
});
|
|
90
90
|
}
|
|
91
|
+
|
|
92
|
+
/**
|
|
93
|
+
* The expected values for a select request.
|
|
94
|
+
*/
|
|
95
|
+
export interface SelectExpectation {
|
|
96
|
+
/** The text expected to be seen in the select dialog. */
|
|
97
|
+
text: string;
|
|
98
|
+
/** The mock response send back from the select dialog. */
|
|
99
|
+
result: string;
|
|
100
|
+
}
|
|
101
|
+
|
|
102
|
+
/**
|
|
103
|
+
* Mock the implementation of `select()` that will respond with configured results
|
|
104
|
+
* for configured select text messages.
|
|
105
|
+
*
|
|
106
|
+
* If there is a call to `select()` that does not match any of the expectations
|
|
107
|
+
* then an error is thrown.
|
|
108
|
+
*/
|
|
109
|
+
export function mockSelect(...expectations: SelectExpectation[]) {
|
|
110
|
+
(select as jest.Mock).mockImplementation((text: string) => {
|
|
111
|
+
for (const { text: expectedText, result } of expectations) {
|
|
112
|
+
if (normalizeSlashes(text) === normalizeSlashes(expectedText)) {
|
|
113
|
+
return Promise.resolve(result);
|
|
114
|
+
}
|
|
115
|
+
}
|
|
116
|
+
throw new Error(`Unexpected select message: ${text}`);
|
|
117
|
+
});
|
|
118
|
+
}
|
|
119
|
+
|
|
120
|
+
export function clearSelectMocks() {
|
|
121
|
+
(select as jest.Mock).mockReset();
|
|
122
|
+
// Because select was originally a spy, calling mockReset will simply reset
|
|
123
|
+
// it as a function with no return value (!), so we need to additionally reset
|
|
124
|
+
// the mock implementation to the one that throws (from jest.setup.js).
|
|
125
|
+
(select as jest.Mock).mockImplementation((text: string) => {
|
|
126
|
+
throw new Error(
|
|
127
|
+
`Unexpected call to \`select("${text}")\`.\nYou should use \`mockSelect()\` to mock calls to \`select()\` with expectations. Search the codebase for \`mockSelect\` to learn more.`
|
|
128
|
+
);
|
|
129
|
+
});
|
|
130
|
+
}
|