From e4900fc27c71c8ce8d32f53c922c1ffeeb5c1077 Mon Sep 17 00:00:00 2001
From: picklesrus <picklesrus@users.noreply.github.com>
Date: Wed, 8 Jun 2016 10:03:14 -0700
Subject: [PATCH] Fix airstrike by grabbing the correct toolbox element. (#411)

Probably broken in 266e2ffa9a017d21d7ca2f151730d6ecfcecf173.
---
 tests/playground.html | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/tests/playground.html b/tests/playground.html
index d43d32ed4..e7e6e1166 100644
--- a/tests/playground.html
+++ b/tests/playground.html
@@ -68,9 +68,7 @@ function start() {
   var match = location.search.match(/dir=([^&]+)/);
   var rtl = match && match[1] == 'rtl';
   document.forms.options.elements.dir.selectedIndex = Number(rtl);
-  var match = location.search.match(/toolbox=([^&]+)/);
-  var toolbox =
-      document.getElementById('toolbox-' + (match ? match[1] : 'categories'));
+  var toolbox = getToolboxElement();
   document.forms.options.elements.toolbox.selectedIndex =
       Number(toolbox.getElementsByTagName('category').length == 0);
   match = location.search.match(/side=([^&]+)/);
@@ -118,6 +116,11 @@ function start() {
   taChange();
 }
 
+function getToolboxElement() {
+  var match = location.search.match(/toolbox=([^&]+)/);
+  return document.getElementById('toolbox-' + (match ? match[1] : 'categories'));
+}
+
 function toXml() {
   var output = document.getElementById('importExport');
   var xml = Blockly.Xml.workspaceToDom(workspace);
@@ -175,7 +178,7 @@ function logger(e) {
 
 function airstrike(n) {
   var prototypes = [];
-  var toolbox = document.getElementById('toolbox');
+  var toolbox = getToolboxElement();
   var blocks = toolbox.getElementsByTagName('block');
   for (var i = 0, block; block = blocks[i]; i++) {
     prototypes.push(block.getAttribute('type'));