group iframe setup into function

Shopify · Sep 29, 2016 · 9c72ece · 9c72ece
1 parent 53c6667
commit 9c72ece
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/test/javascripts/turbolinks_test.coffee b/test/javascripts/turbolinks_test.coffee
@@ -62,6 +62,12 @@ describe 'Turbolinks', ->
       <div id="turbo-area" refresh="turbo-area"></div>
     """
 
+  setupIframe = ->
+    iframe = document.createElement('iframe')
+    document.body.appendChild(iframe)
+    iframe.contentDocument.write(baseHTML)
+    iframe.contentDocument
+
   startFromFixture = (route) ->
     fixtureHTML = ROUTES[route][2]
     testDocument.documentElement.innerHTML = fixtureHTML
@@ -76,11 +82,7 @@ describe 'Turbolinks', ->
     Turbolinks.visit('/' + url, options)
 
   beforeEach ->
-    unless iframe
-      iframe = document.createElement('iframe')
-      document.body.appendChild(iframe)
-      iframe.contentDocument.write(baseHTML)
-      testDocument = iframe.contentDocument
+    testDocument = setupIframe() unless iframe
     sandbox = sinon.sandbox.create()
     pushStateStub = sandbox.stub(Turbolinks, 'pushState')
     replaceStateStub = sandbox.stub(Turbolinks, 'replaceState')