chore: more BiDi Connection test to unit (#10839)

This commit is contained in:
Nikolay Vitkov 2023-09-04 14:06:23 +02:00 committed by GitHub
parent eabaf6c605
commit 12bdd70651
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 59 additions and 59 deletions

View File

@ -0,0 +1,59 @@
/**
* Copyright 2022 Google Inc. All rights reserved.
*
* 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, it} from 'node:test';
import expect from 'expect';
import {ConnectionTransport} from '../ConnectionTransport.js';
import {Connection} from './Connection.js';
describe('WebDriver BiDi Connection', () => {
class TestConnectionTransport implements ConnectionTransport {
sent: string[] = [];
closed = false;
send(message: string) {
this.sent.push(message);
}
close(): void {
this.closed = true;
}
}
it('should work', async () => {
const transport = new TestConnectionTransport();
const connection = new Connection('ws://127.0.0.1', transport);
const responsePromise = connection.send('session.new', {
capabilities: {},
});
expect(transport.sent).toEqual([
`{"id":1,"method":"session.new","params":{"capabilities":{}}}`,
]);
const id = JSON.parse(transport.sent[0]!).id;
const rawResponse = {
id,
result: {ready: false, message: 'already connected'},
};
(transport as ConnectionTransport).onmessage?.(JSON.stringify(rawResponse));
const response = await responsePromise;
expect(response).toEqual(rawResponse);
connection.dispose();
expect(transport.closed).toBeTruthy();
});
});

View File

@ -1,59 +0,0 @@
/**
* Copyright 2022 Google Inc. All rights reserved.
*
* 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 expect from 'expect';
import {Connection} from 'puppeteer-core/internal/common/bidi/Connection.js';
import {ConnectionTransport} from 'puppeteer-core/internal/common/ConnectionTransport.js';
describe('WebDriver BiDi', () => {
describe('Connection', () => {
class TestConnectionTransport implements ConnectionTransport {
sent: string[] = [];
closed = false;
send(message: string) {
this.sent.push(message);
}
close(): void {
this.closed = true;
}
}
it('should work', async () => {
const transport = new TestConnectionTransport();
const connection = new Connection('ws://127.0.0.1', transport);
const responsePromise = connection.send('session.new', {
capabilities: {},
});
expect(transport.sent).toEqual([
`{"id":1,"method":"session.new","params":{"capabilities":{}}}`,
]);
const id = JSON.parse(transport.sent[0]!).id;
const rawResponse = {
id,
result: {ready: false, message: 'already connected'},
};
(transport as ConnectionTransport).onmessage?.(
JSON.stringify(rawResponse)
);
const response = await responsePromise;
expect(response).toEqual(rawResponse);
connection.dispose();
expect(transport.closed).toBeTruthy();
});
});
});