From 09fa1d49a1de70e07d9069e264e9e03211ea1672 Mon Sep 17 00:00:00 2001
From: mluto <m@luto.at>
Date: Tue, 15 Jan 2013 22:17:40 +0100
Subject: [PATCH] Added ability to load the same pad twice with helper.newPad,
 use this in load-message-tests

---
 tests/frontend/helper.js                   | 11 ++++++-----
 tests/frontend/specs/chat_load_messages.js | 20 ++++++++++----------
 2 files changed, 16 insertions(+), 15 deletions(-)

diff --git a/tests/frontend/helper.js b/tests/frontend/helper.js
index ee57c8690..bf5df5082 100644
--- a/tests/frontend/helper.js
+++ b/tests/frontend/helper.js
@@ -56,13 +56,13 @@ var helper = {};
     window.document.cookie = "";
   }
 
-  helper.newPad = function(){
+  helper.newPad = function(cb, padName){
     //build opts object
     var opts = {clearCookies: true}
-    if(typeof arguments[0] === 'function'){
-      opts.cb = arguments[0]
+    if(typeof cb === 'function'){
+      opts.cb = cb
     } else {
-      opts = _.defaults(arguments[0], opts);
+      opts = _.defaults(cb, opts);
     }
 
     //clear cookies
@@ -70,7 +70,8 @@ var helper = {};
       helper.clearCookies();
     }
 
-    var padName = "FRONTEND_TEST_" + helper.randomString(20);
+    if(!padName)
+      padName = "FRONTEND_TEST_" + helper.randomString(20);
     $iframe = $("<iframe src='/p/" + padName + "'></iframe>");
     
     //clean up inner iframe references
diff --git a/tests/frontend/specs/chat_load_messages.js b/tests/frontend/specs/chat_load_messages.js
index 8dc98691e..cf625dc46 100644
--- a/tests/frontend/specs/chat_load_messages.js
+++ b/tests/frontend/specs/chat_load_messages.js
@@ -1,6 +1,9 @@
 describe("chat-load-messages", function(){
+  var padName;
+ 
   it("creates a pad", function(done) {
-    helper.newPad(done);
+    padName = helper.newPad(done);
+    this.timeout(60000);
   });
 
   it("adds a lot of messages", function(done) {
@@ -11,6 +14,8 @@ describe("chat-load-messages", function(){
     var chatInput = chrome$("#chatinput");
     var chatText = chrome$("#chattext");
     
+    this.timeout(10000);
+    
     var messages = 140;
     for(var i=1; i <= messages; i++) {
       var num = ''+i;
@@ -25,20 +30,15 @@ describe("chat-load-messages", function(){
       return chatText.children("p").length == messages;
     }).always(function(){
       expect(chatText.children("p").length).to.be(messages);
-      $('#iframe-container iframe')[0].contentWindow.location.reload();
-      done();
+      helper.newPad(done, padName);
      });
   });
   
   it("checks initial message count", function(done) {
     var chatText;
     var expectedCount = 101;
+    var chrome$ = helper.padChrome$;
     helper.waitFor(function(){
-      // wait for the frame to load
-      var chrome$ = $('#iframe-container iframe')[0].contentWindow.$;
-      if(!chrome$) // page not fully loaded
-        return false;
-    
       var chatButton = chrome$("#chaticon");
       chatButton.click();
       chatText = chrome$("#chattext");
@@ -51,7 +51,7 @@ describe("chat-load-messages", function(){
   
   it("loads more messages", function(done) {
     var expectedCount = 122;
-    var chrome$ = $('#iframe-container iframe')[0].contentWindow.$;
+    var chrome$ = helper.padChrome$;
     var chatButton = chrome$("#chaticon");
     chatButton.click();
     var chatText = chrome$("#chattext");
@@ -68,7 +68,7 @@ describe("chat-load-messages", function(){
   
   it("checks for button vanishing", function(done) {
     var expectedDisplay = 'none';
-    var chrome$ = $('#iframe-container iframe')[0].contentWindow.$;
+    var chrome$ = helper.padChrome$;
     var chatButton = chrome$("#chaticon");
     chatButton.click();
     var chatText = chrome$("#chattext");