Small improvements to the browser test samples

espirl · Jan 5, 2017 · 977571e · 977571e
1 parent 52a42f8
commit 977571e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/ch10-testing/selenium/index.js b/ch10-testing/selenium/index.js
@@ -1,5 +1,6 @@
 const express = require('express');
 const app = express();
+const port = process.env.PORT || 4000;
 
 app.get('/', (req, res) => {
   res.send(`
@@ -14,6 +15,6 @@ app.get('/', (req, res) => {
   `);
 });
 
-app.listen(process.env.PORT, () => {
-  console.log('Running on port', process.env.PORT);
+app.listen(port, () => {
+  console.log('Running on port', port);
 });
diff --git a/ch10-testing/selenium/test/specs/todo.spec.js b/ch10-testing/selenium/test/specs/todo.spec.js
@@ -5,7 +5,7 @@ const webdriverio = require('webdriverio');
 describe('todo tests', () => {
   let client;
 
-  before(function(){
+  before(() => {
     client = webdriverio.remote();
     return client.init();
   });
@@ -14,6 +14,6 @@ describe('todo tests', () => {
     return client
       .url('/')
       .getTitle()
-        .then(title => assert(title === 'My to-do list'));
+        .then(title => assert.equal(title, 'My to-do list'));
   });
 });