Add Page.emulateMedia binding

zploskey · Feb 28, 2018 · 48baea9 · 48baea9
1 parent 4bf5e9f
commit 48baea9
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/__tests__/puppeteer_test.re b/__tests__/puppeteer_test.re
@@ -363,6 +363,11 @@ describe("Page", () => {
          )
     )
   );
+  testPromise("emulateMedia()", () =>
+    Js.Promise.(
+      page^ |> Page.emulateMedia(Some(Print)) |> then_(() => pass |> resolve)
+    )
+  );
   afterAllPromise(() =>
     Js.Promise.(Page.close(page^) |> then_(() => Browser.close(browser^)))
   );

diff --git a/src/Page.re b/src/Page.re
@@ -166,6 +166,24 @@ external setCookie : array(cookie) => Js.Promise.t(unit) = "";
 external emulate : emulateOption => Js.Promise.t(unit) = "";
 
 [@bs.send.pipe : t] external viewport : unit => viewport = "";
+
+[@bs.send]
+external _emulateMedia : (t, Js.Nullable.t(string)) => Js.Promise.t(unit) =
+  "emulateMedia";
+
+type mediaType =
+  | Screen
+  | Print;
+
+let emulateMedia = (mediaType: option(mediaType), t) =>
+  switch mediaType {
+  | Some(v) =>
+    switch v {
+    | Screen => _emulateMedia(t, Js.Nullable.return("screen"))
+    | Print => _emulateMedia(t, Js.Nullable.return("print"))
+    }
+  | None => _emulateMedia(t, Js.Nullable.null)
+  };
 /* TODO: the rest of Page */
 /* cookies(...urls: string[]): Promise<Cookie[]>; */
 /*  type 'cookie;