From a0cf33bbb02d43596f0a5e3479eadbcbe78aaf14 Mon Sep 17 00:00:00 2001
From: Andrey Lushnikov <lushnikov@chromium.org>
Date: Tue, 1 Mar 2022 00:30:00 -0700
Subject: [PATCH] fix: beautiful toHaveScreenshot in tracing (#12422)

---
 packages/playwright-core/src/protocol/channels.ts  | 1 +
 packages/playwright-core/src/protocol/protocol.yml | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/packages/playwright-core/src/protocol/channels.ts b/packages/playwright-core/src/protocol/channels.ts
index 8eef17b58e..09def4da7e 100644
--- a/packages/playwright-core/src/protocol/channels.ts
+++ b/packages/playwright-core/src/protocol/channels.ts
@@ -4089,6 +4089,7 @@ export const commandsWithTracingSnapshots = new Set([
   'Page.goBack',
   'Page.goForward',
   'Page.reload',
+  'Page.expectScreenshot',
   'Page.screenshot',
   'Page.setViewportSize',
   'Page.keyboardDown',
diff --git a/packages/playwright-core/src/protocol/protocol.yml b/packages/playwright-core/src/protocol/protocol.yml
index 8e350bc7a5..3ccf0c99f6 100644
--- a/packages/playwright-core/src/protocol/protocol.yml
+++ b/packages/playwright-core/src/protocol/protocol.yml
@@ -1028,6 +1028,8 @@ Page:
         log:
           type: array?
           items: string
+      tracing:
+        snapshot: true
 
     screenshot:
       parameters: