mirror of
https://github.com/denoland/deno.git
synced 2024-12-12 18:42:18 -05:00
490d2a5ca1
Fixes #5260
53 lines
1.3 KiB
TypeScript
53 lines
1.3 KiB
TypeScript
// Copyright 2018-2020 the Deno authors. All rights reserved. MIT license.
|
|
import { assertEquals, assertThrowsAsync } from "../testing/asserts.ts";
|
|
import { MuxAsyncIterator } from "./mux_async_iterator.ts";
|
|
|
|
// eslint-disable-next-line require-await
|
|
async function* gen123(): AsyncIterableIterator<number> {
|
|
yield 1;
|
|
yield 2;
|
|
yield 3;
|
|
}
|
|
|
|
// eslint-disable-next-line require-await
|
|
async function* gen456(): AsyncIterableIterator<number> {
|
|
yield 4;
|
|
yield 5;
|
|
yield 6;
|
|
}
|
|
|
|
// eslint-disable-next-line require-await
|
|
async function* genThrows(): AsyncIterableIterator<number> {
|
|
yield 7;
|
|
throw new Error("something went wrong");
|
|
}
|
|
|
|
Deno.test("[async] MuxAsyncIterator", async function (): Promise<void> {
|
|
const mux = new MuxAsyncIterator<number>();
|
|
mux.add(gen123());
|
|
mux.add(gen456());
|
|
const results = new Set();
|
|
for await (const value of mux) {
|
|
results.add(value);
|
|
}
|
|
assertEquals(results.size, 6);
|
|
});
|
|
|
|
Deno.test({
|
|
name: "[async] MuxAsyncIterator throws",
|
|
async fn() {
|
|
const mux = new MuxAsyncIterator<number>();
|
|
mux.add(gen123());
|
|
mux.add(genThrows());
|
|
const results = new Set();
|
|
await assertThrowsAsync(
|
|
async () => {
|
|
for await (const value of mux) {
|
|
results.add(value);
|
|
}
|
|
},
|
|
Error,
|
|
"something went wrong"
|
|
);
|
|
},
|
|
});
|