From 7baa192d59f7565cfc4107d2a207b8d37de8b30e Mon Sep 17 00:00:00 2001
From: Sam El-Husseini <sam.elhusseini@gmail.com>
Date: Mon, 9 Dec 2019 08:33:00 -0800
Subject: [PATCH] Update zelos rendering playground so that screenshots reflect
 state of the workspace (#3491)

---
 tests/rendering/zelos/pxtblockly.html | 33 ++++++++++++++++--------
 tests/rendering/zelos/zelos.html      | 37 +++++++++++++++++----------
 2 files changed, 45 insertions(+), 25 deletions(-)

diff --git a/tests/rendering/zelos/pxtblockly.html b/tests/rendering/zelos/pxtblockly.html
index 77a282ac4..d3aa9dffe 100644
--- a/tests/rendering/zelos/pxtblockly.html
+++ b/tests/rendering/zelos/pxtblockly.html
@@ -37,17 +37,28 @@
         }
       });
 
-      try {
-        Blockly.Xml.domToWorkspace(Blockly.Xml.textToDom(xml), workspace);
-        var topBlock = workspace.getTopBlocks()[0];
-        workspaceToSvg_(workspace, function (datauri) {
-          window.parent.postMessage({
-            type: 'svg',
-            from: 'pxtblockly',
-            text: datauri
-          }, '*');
-        });
-      } catch { }
+      Blockly.Xml.domToWorkspace(Blockly.Xml.textToDom(xml), workspace);
+
+      function postChange() {
+        try {
+          var topBlock = workspace.getTopBlocks()[0];
+          workspaceToSvg_(workspace, function (datauri) {
+            window.parent.postMessage({
+              type: 'svg',
+              from: 'pxtblockly',
+              text: datauri
+            }, '*');
+          });
+        } catch { }
+      }
+
+      workspace.addChangeListener(function(e) {
+        if (e.type != 'ui') {
+          postChange();
+        }
+      });
+
+      postChange();
     });
   </script>
 </body>
diff --git a/tests/rendering/zelos/zelos.html b/tests/rendering/zelos/zelos.html
index 20f4ecd6d..af239c89d 100644
--- a/tests/rendering/zelos/zelos.html
+++ b/tests/rendering/zelos/zelos.html
@@ -32,7 +32,6 @@
   <script>
     goog.require('Blockly.blockRendering.Debug');
     goog.require('Blockly.zelos.Renderer');
-    goog.require('Blockly.zelos.Themes.Classic');
     // Blockly.blockRendering.startDebugger();
 
     // This stub is a workaround in order to load pxt-blockly blocks, as they
@@ -62,7 +61,6 @@
           drag: true,
           wheel: false,
         },
-        theme: Blockly.zelos.Themes.Classic,
         zoom: {
           wheel: true,
           startScale: 2,
@@ -73,19 +71,30 @@
       constants.FIELD_TEXT_FONTWEIGHT = 'bold';
       constants.FIELD_TEXT_FONTFAMILY = 'Helvetica Neue';
 
-      try {
-        Blockly.Xml.domToWorkspace(Blockly.Xml.textToDom(xml), workspace);
-        var topBlock = workspace.getTopBlocks()[0];
-        workspaceToSvg_(workspace, function (datauri) {
-          window.parent.postMessage({
-            type: 'svg',
-            from: 'zelos',
-            text: datauri
-          }, '*');
-        }, document.getElementById('blocklycss').innerText);
-      } catch (err) {
-        console.error(err);
+      Blockly.Xml.domToWorkspace(Blockly.Xml.textToDom(xml), workspace);
+
+      function postChange() {
+        try {
+          var topBlock = workspace.getTopBlocks()[0];
+          workspaceToSvg_(workspace, function (datauri) {
+            window.parent.postMessage({
+              type: 'svg',
+              from: 'zelos',
+              text: datauri
+            }, '*');
+          }, document.getElementById('blocklycss').innerText);
+        } catch (err) {
+          console.error(err);
+        }
       }
+
+      workspace.addChangeListener(function(e) {
+        if (e.type != 'ui') {
+          postChange();
+        }
+      });
+
+      postChange();
     });
   </script>
 </body>