coze-studio/frontend/packages/common/chat-area/utils/__tests__/async.test.ts

54 lines
1.4 KiB
TypeScript

/*
* Copyright 2025 coze-dev Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import { describe, expect, vi } from 'vitest';
import { Deferred, sleep } from '../src/async';
it('sleep', async () => {
vi.useFakeTimers();
let count = 0;
sleep(1000).then(() => (count = 1));
vi.runAllTimers();
expect(count).toBe(0);
await Promise.resolve();
expect(count).toBe(1);
});
describe('test deferred', () => {
it('works', async () => {
const deferred = new Deferred<number>();
deferred.resolve(1);
expect(await deferred.promise).toBe(1);
});
it('reject', async () => {
const deferred = new Deferred();
deferred.reject(123);
try {
await deferred.promise;
} catch (err) {
expect(err).toBe(123);
}
});
it('perform like promise', async () => {
const deferred = new Deferred<string>();
deferred.resolve('1');
expect(await deferred).toBe('1');
});
});