test: refactor speed-tester.test.js

This commit is contained in:
Micooz 2018-07-05 18:28:47 +08:00
parent 6ac8ee463f
commit 8edddc9c99
3 changed files with 8 additions and 5 deletions

3
test/common/sleep.js Normal file

@ -0,0 +1,3 @@
export default function sleep(ms) {
return new Promise((resolve) => setTimeout(resolve, ms));
}

@ -1,10 +1,7 @@
import net from 'net';
import sleep from '../../common/sleep';
import { DNSCache } from '../../../src/utils/dns-cache';
async function sleep(ms) {
return new Promise((resolve) => setTimeout(resolve, ms));
}
it('should return an ip address', async function() {
DNSCache.clear();
expect(net.isIP(await DNSCache.get('localhost'))).toBe(4);

@ -1,7 +1,10 @@
import sleep from '../../common/sleep';
import { SpeedTester } from '../../../src/utils';
test('SpeedTester::getSpeed()', () => {
test('SpeedTester::getSpeed()', async () => {
const st = new SpeedTester();
expect(st.getSpeed()).toBe(0);
st.feed(10);
await sleep(20);
expect(st.getSpeed()).toBeGreaterThan(0);
});