.travis.yml in date_book-0.1.1 vs .travis.yml in date_book-0.1.3

@@ -1,7 +1,25 @@
 language: ruby
+sudo: required
+dist: trusty
 rvm:
   - 2.4.1
+gemfile:
+  - Gemfile
+addons:
+  apt:
+    sources:
+      - google-chrome
+    packages:
+      - google-chrome-stable
+before_install:
+  - mkdir -p ~/drivers; export PATH=~/drivers:$PATH;
+  - CHROMEDRIVER_VERSION=$(wget -q -O - http://chromedriver.storage.googleapis.com/LATEST_RELEASE);
+    CHROMEDRIVER_URL="http://chromedriver.storage.googleapis.com/$CHROMEDRIVER_VERSION/chromedriver_linux64.zip";
+    FILE=`mktemp`; wget "$CHROMEDRIVER_URL" -qO $FILE && unzip $FILE chromedriver -d ~/drivers; rm $FILE; chmod 777 ~/drivers/chromedriver;
+    ~/drivers/chromedriver --version;
 before_script:
-  - "export DISPLAY=:99.0"
-  - "sh -e /etc/init.d/xvfb start"
-  - sleep 3 # give xvfb some time to start
+  - if [[ -z $HEADLESS ]]; then
+      export DISPLAY=:99.0;
+      sh -e /etc/init.d/xvfb start;
+      sleep 1;
+    fi
\ No newline at end of file