Got tic-tac-toe page refresh test working

Author: jwngr

tests/protractor/tictactoe/tictactoe.js

@@ -1,7 +1,19 @@
 var app = angular.module('tictactoe', ['firebase']);
 app.controller('TicTacToeCtrl', function Chat($scope, $firebaseObject) {
   // Get a reference to the Firebase
-  var boardRef = new Firebase('https://angularfire.firebaseio-demo.com/tictactoe').push();
+  var boardRef = new Firebase('https://angularfire.firebaseio-demo.com/tictactoe');
+
+  // If the query string contains a push ID, use that as the child for data storage;
+  // otherwise, generate a new random push ID
+  var pushId;
+  if (window.location && window.location.search) {
+    pushId = window.location.search.substr(1).split('=')[1];
+  }
+  if (pushId) {
+    boardRef = boardRef.child(pushId);
+  } else {
+    boardRef = boardRef.push();
+  }
 
   // Put the random push ID into the DOM so that the test suite can grab it
   document.getElementById('pushId').innerHTML = boardRef.key();

tests/protractor/tictactoe/tictactoe.spec.js

@@ -94,14 +94,21 @@ describe('TicTacToe App', function () {
     expect(cells.get(6).getText()).toBe('X');
   });
 
-  xit('persists state across refresh', function() {
-    // Make sure the board has 9 cells
-    expect(cells.count()).toBe(9);
+  it('persists state across refresh', function(done) {
+    browser.get('tictactoe/tictactoe.html?pushId=' + firebaseRef.key()).then(function() {
+      // Wait for AngularFire to sync the initial state
+      sleep();
 
-    // Make sure the content of each clicked cell is correct
-    expect(cells.get(0).getText()).toBe('X');
-    expect(cells.get(2).getText()).toBe('O');
-    expect(cells.get(6).getText()).toBe('X');
+      // Make sure the board has 9 cells
+      expect(cells.count()).toBe(9);
+
+      // Make sure the content of each clicked cell is correct
+      expect(cells.get(0).getText()).toBe('X');
+      expect(cells.get(2).getText()).toBe('O');
+      expect(cells.get(6).getText()).toBe('X');
+
+      done();
+    });
   });
 
   it('stops updating Firebase once the AngularFire bindings are destroyed', function () {
@@ -117,7 +124,7 @@ describe('TicTacToe App', function () {
 
     sleep();
 
-    expect(cells.get(4).getText()).toBe('O');
+    expect(cells.get(4).getText()).toBe('X');
 
     sleep();