1 files changed, 61 insertions, 0 deletions
diff --git a/remote/test/puppeteer/test/src/bidi/Connection.spec.ts b/remote/test/puppeteer/test/src/bidi/Connection.spec.ts
new file mode 100644
index 0000000000..e5cd9de0a1
--- /dev/null
+++ b/remote/test/puppeteer/test/src/bidi/Connection.spec.ts
@@ -0,0 +1,61 @@
+/**
+ * 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(transport);
+      const responsePromise = connection.send('session.new', {
+        capabilities: {
+          proxy: {},
+        },
+      });
+      expect(transport.sent).toEqual([
+        `{"id":1,"method":"session.new","params":{"capabilities":{"proxy":{}}}}`,
+      ]);
+      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();
+    });
+  });
+});