1 files changed, 373 insertions, 0 deletions
diff --git a/remote/test/puppeteer/test/src/jshandle.spec.ts b/remote/test/puppeteer/test/src/jshandle.spec.ts
new file mode 100644
index 0000000000..28097811e4
--- /dev/null
+++ b/remote/test/puppeteer/test/src/jshandle.spec.ts
@@ -0,0 +1,373 @@
+/**
+ * @license
+ * Copyright 2018 Google Inc.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import expect from 'expect';
+import {JSHandle} from 'puppeteer-core/internal/api/JSHandle.js';
+import {
+  asyncDisposeSymbol,
+  disposeSymbol,
+} from 'puppeteer-core/internal/util/disposable.js';
+import sinon from 'sinon';
+
+import {getTestState, setupTestBrowserHooks} from './mocha-utils.js';
+
+describe('JSHandle', function () {
+  setupTestBrowserHooks();
+
+  describe('Page.evaluateHandle', function () {
+    it('should work', async () => {
+      const {page} = await getTestState();
+
+      using windowHandle = await page.evaluateHandle(() => {
+        return window;
+      });
+      expect(windowHandle).toBeTruthy();
+    });
+    it('should return the RemoteObject', async () => {
+      const {page} = await getTestState();
+
+      using windowHandle = await page.evaluateHandle(() => {
+        return window;
+      });
+      expect(windowHandle.remoteObject()).toBeTruthy();
+    });
+    it('should accept object handle as an argument', async () => {
+      const {page} = await getTestState();
+
+      using navigatorHandle = await page.evaluateHandle(() => {
+        return navigator;
+      });
+      const text = await page.evaluate(e => {
+        return e.userAgent;
+      }, navigatorHandle);
+      expect(text).toContain('Mozilla');
+    });
+    it('should accept object handle to primitive types', async () => {
+      const {page} = await getTestState();
+
+      using aHandle = await page.evaluateHandle(() => {
+        return 5;
+      });
+      const isFive = await page.evaluate(e => {
+        return Object.is(e, 5);
+      }, aHandle);
+      expect(isFive).toBeTruthy();
+    });
+    it('should warn about recursive objects', async () => {
+      const {page} = await getTestState();
+
+      const test: {obj?: unknown} = {};
+      test.obj = test;
+      let error!: Error;
+      await page
+        .evaluateHandle(opts => {
+          return opts;
+        }, test)
+        .catch(error_ => {
+          return (error = error_);
+        });
+      expect(error.message).toContain('Recursive objects are not allowed.');
+    });
+    it('should accept object handle to unserializable value', async () => {
+      const {page} = await getTestState();
+
+      using aHandle = await page.evaluateHandle(() => {
+        return Infinity;
+      });
+      expect(
+        await page.evaluate(e => {
+          return Object.is(e, Infinity);
+        }, aHandle)
+      ).toBe(true);
+    });
+    it('should use the same JS wrappers', async () => {
+      const {page} = await getTestState();
+
+      using aHandle = await page.evaluateHandle(() => {
+        (globalThis as any).FOO = 123;
+        return window;
+      });
+      expect(
+        await page.evaluate(e => {
+          return (e as any).FOO;
+        }, aHandle)
+      ).toBe(123);
+    });
+  });
+
+  describe('JSHandle.getProperty', function () {
+    it('should work', async () => {
+      const {page} = await getTestState();
+
+      using aHandle = await page.evaluateHandle(() => {
+        return {
+          one: 1,
+          two: 2,
+          three: 3,
+        };
+      });
+      using twoHandle = await aHandle.getProperty('two');
+      expect(await twoHandle.jsonValue()).toEqual(2);
+    });
+  });
+
+  describe('JSHandle.jsonValue', function () {
+    it('should work', async () => {
+      const {page} = await getTestState();
+
+      using aHandle = await page.evaluateHandle(() => {
+        return {foo: 'bar'};
+      });
+      const json = await aHandle.jsonValue();
+      expect(json).toEqual({foo: 'bar'});
+    });
+
+    it('works with jsonValues that are not objects', async () => {
+      const {page} = await getTestState();
+
+      using aHandle = await page.evaluateHandle(() => {
+        return ['a', 'b'];
+      });
+      const json = await aHandle.jsonValue();
+      expect(json).toEqual(['a', 'b']);
+    });
+
+    it('works with jsonValues that are primitives', async () => {
+      const {page} = await getTestState();
+
+      using aHandle = await page.evaluateHandle(() => {
+        return 'foo';
+      });
+      expect(await aHandle.jsonValue()).toEqual('foo');
+
+      using bHandle = await page.evaluateHandle(() => {
+        return undefined;
+      });
+      expect(await bHandle.jsonValue()).toEqual(undefined);
+    });
+
+    it('should work with dates', async () => {
+      const {page} = await getTestState();
+
+      using dateHandle = await page.evaluateHandle(() => {
+        return new Date('2017-09-26T00:00:00.000Z');
+      });
+      const date = await dateHandle.jsonValue();
+      expect(date).toBeInstanceOf(Date);
+      expect(date.toISOString()).toEqual('2017-09-26T00:00:00.000Z');
+    });
+    it('should not throw for circular objects', async () => {
+      const {page} = await getTestState();
+
+      using handle = await page.evaluateHandle(() => {
+        const t: {t?: unknown; g: number} = {g: 1};
+        t.t = t;
+        return t;
+      });
+      await handle.jsonValue();
+    });
+  });
+
+  describe('JSHandle.getProperties', function () {
+    it('should work', async () => {
+      const {page} = await getTestState();
+
+      using aHandle = await page.evaluateHandle(() => {
+        return {
+          foo: 'bar',
+        };
+      });
+      const properties = await aHandle.getProperties();
+      using foo = properties.get('foo')!;
+      expect(foo).toBeTruthy();
+      expect(await foo.jsonValue()).toBe('bar');
+    });
+    it('should return even non-own properties', async () => {
+      const {page} = await getTestState();
+
+      using aHandle = await page.evaluateHandle(() => {
+        class A {
+          a: string;
+          constructor() {
+            this.a = '1';
+          }
+        }
+        class B extends A {
+          b: string;
+          constructor() {
+            super();
+            this.b = '2';
+          }
+        }
+        return new B();
+      });
+      const properties = await aHandle.getProperties();
+      expect(await properties.get('a')!.jsonValue()).toBe('1');
+      expect(await properties.get('b')!.jsonValue()).toBe('2');
+    });
+  });
+
+  describe('JSHandle.asElement', function () {
+    it('should work', async () => {
+      const {page} = await getTestState();
+
+      using aHandle = await page.evaluateHandle(() => {
+        return document.body;
+      });
+      using element = aHandle.asElement();
+      expect(element).toBeTruthy();
+    });
+    it('should return null for non-elements', async () => {
+      const {page} = await getTestState();
+
+      using aHandle = await page.evaluateHandle(() => {
+        return 2;
+      });
+      using element = aHandle.asElement();
+      expect(element).toBeFalsy();
+    });
+    it('should return ElementHandle for TextNodes', async () => {
+      const {page} = await getTestState();
+
+      await page.setContent('<div>ee!</div>');
+      using aHandle = await page.evaluateHandle(() => {
+        return document.querySelector('div')!.firstChild;
+      });
+      using element = aHandle.asElement();
+      expect(element).toBeTruthy();
+      expect(
+        await page.evaluate(e => {
+          return e?.nodeType === Node.TEXT_NODE;
+        }, element)
+      );
+    });
+  });
+
+  describe('JSHandle.toString', function () {
+    it('should work for primitives', async () => {
+      const {page} = await getTestState();
+
+      using numberHandle = await page.evaluateHandle(() => {
+        return 2;
+      });
+      expect(numberHandle.toString()).toBe('JSHandle:2');
+      using stringHandle = await page.evaluateHandle(() => {
+        return 'a';
+      });
+      expect(stringHandle.toString()).toBe('JSHandle:a');
+    });
+    it('should work for complicated objects', async () => {
+      const {page} = await getTestState();
+
+      using aHandle = await page.evaluateHandle(() => {
+        return window;
+      });
+      expect(aHandle.toString()).atLeastOneToContain([
+        'JSHandle@object',
+        'JSHandle@window',
+      ]);
+    });
+    it('should work with different subtypes', async () => {
+      const {page} = await getTestState();
+
+      expect((await page.evaluateHandle('(function(){})')).toString()).toBe(
+        'JSHandle@function'
+      );
+      expect((await page.evaluateHandle('12')).toString()).toBe('JSHandle:12');
+      expect((await page.evaluateHandle('true')).toString()).toBe(
+        'JSHandle:true'
+      );
+      expect((await page.evaluateHandle('undefined')).toString()).toBe(
+        'JSHandle:undefined'
+      );
+      expect((await page.evaluateHandle('"foo"')).toString()).toBe(
+        'JSHandle:foo'
+      );
+      expect((await page.evaluateHandle('Symbol()')).toString()).toBe(
+        'JSHandle@symbol'
+      );
+      expect((await page.evaluateHandle('new Map()')).toString()).toBe(
+        'JSHandle@map'
+      );
+      expect((await page.evaluateHandle('new Set()')).toString()).toBe(
+        'JSHandle@set'
+      );
+      expect((await page.evaluateHandle('[]')).toString()).toBe(
+        'JSHandle@array'
+      );
+      expect((await page.evaluateHandle('null')).toString()).toBe(
+        'JSHandle:null'
+      );
+      expect((await page.evaluateHandle('/foo/')).toString()).toBe(
+        'JSHandle@regexp'
+      );
+      expect((await page.evaluateHandle('document.body')).toString()).toBe(
+        'JSHandle@node'
+      );
+      expect((await page.evaluateHandle('new Date()')).toString()).toBe(
+        'JSHandle@date'
+      );
+      expect((await page.evaluateHandle('new WeakMap()')).toString()).toBe(
+        'JSHandle@weakmap'
+      );
+      expect((await page.evaluateHandle('new WeakSet()')).toString()).toBe(
+        'JSHandle@weakset'
+      );
+      expect((await page.evaluateHandle('new Error()')).toString()).toBe(
+        'JSHandle@error'
+      );
+      expect((await page.evaluateHandle('new Int32Array()')).toString()).toBe(
+        'JSHandle@typedarray'
+      );
+      expect((await page.evaluateHandle('new Proxy({}, {})')).toString()).toBe(
+        'JSHandle@proxy'
+      );
+    });
+  });
+
+  describe('JSHandle[Symbol.dispose]', () => {
+    it('should work', async () => {
+      const {page} = await getTestState();
+      using handle = await page.evaluateHandle('new Set()');
+      const spy = sinon.spy(handle, disposeSymbol);
+      {
+        using _ = handle;
+      }
+      expect(handle).toBeInstanceOf(JSHandle);
+      expect(spy.calledOnce).toBeTruthy();
+      expect(handle.disposed).toBeTruthy();
+    });
+  });
+
+  describe('JSHandle[Symbol.asyncDispose]', () => {
+    it('should work', async () => {
+      const {page} = await getTestState();
+      using handle = await page.evaluateHandle('new Set()');
+      const spy = sinon.spy(handle, asyncDisposeSymbol);
+      {
+        await using _ = handle;
+      }
+      expect(handle).toBeInstanceOf(JSHandle);
+      expect(spy.calledOnce).toBeTruthy();
+      expect(handle.disposed).toBeTruthy();
+    });
+  });
+
+  describe('JSHandle.move', () => {
+    it('should work', async () => {
+      const {page} = await getTestState();
+      using handle = await page.evaluateHandle('new Set()');
+      const spy = sinon.spy(handle, disposeSymbol);
+      {
+        using _ = handle;
+        handle.move();
+      }
+      expect(handle).toBeInstanceOf(JSHandle);
+      expect(spy.calledOnce).toBeTruthy();
+      expect(handle.disposed).toBeFalsy();
+    });
+  });
+});