Adding healthcheck endpoint

Gemfile

@@ -4,3 +4,7 @@ ruby '2.1.2'
 gem 'sinatra', require: 'sinatra/base'
 gem 'sinatra-contrib'
 gem 'fleet-api', '0.6.1', require: 'fleet'
+
+group :test do
+  gem 'rspec', '~> 3.0'
+end

Gemfile.lock

@@ -2,6 +2,7 @@ GEM
   remote: https://rubygems.org/
   specs:
     backports (3.6.0)
+    diff-lcs (1.2.5)
     faraday (0.8.9)
       multipart-post (~> 1.2.0)
     faraday_middleware (0.9.0)
@@ -16,6 +17,18 @@ GEM
       rack
     rack-test (0.6.2)
       rack (>= 1.0)
+    rspec (3.1.0)
+      rspec-core (~> 3.1.0)
+      rspec-expectations (~> 3.1.0)
+      rspec-mocks (~> 3.1.0)
+    rspec-core (3.1.7)
+      rspec-support (~> 3.1.0)
+    rspec-expectations (3.1.2)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.1.0)
+    rspec-mocks (3.1.3)
+      rspec-support (~> 3.1.0)
+    rspec-support (3.1.2)
     sinatra (1.4.5)
       rack (~> 1.4)
       rack-protection (~> 1.4)
@@ -34,5 +47,6 @@ PLATFORMS
 
 DEPENDENCIES
   fleet-api (= 0.6.1)
+  rspec (~> 3.0)
   sinatra
   sinatra-contrib

app.rb

@@ -1,10 +1,15 @@
 require 'rubygems'
 require 'bundler'
+require 'fleet'
 
 # Setup load paths
 Bundler.require
 $: << File.expand_path('../', __FILE__)
 
+Fleet.configure do |config|
+  config.fleet_api_url = ENV['FLEETCTL_ENDPOINT']
+end
+
 # Require base
 require 'sinatra/base'
 
@@ -18,6 +23,7 @@ class App < Sinatra::Application
       disable :static
     end
 
+    use FleetAdapter::Routes::Healthcheck
     use FleetAdapter::Routes::Services
   end
 end

app/routes.rb

@@ -1,6 +1,7 @@
 module FleetAdapter
   module Routes
     autoload :Base, 'app/routes/base'
+    autoload :Healthcheck, 'app/routes/healthcheck'
     autoload :Services, 'app/routes/services'
   end
 end

app/routes/healthcheck.rb

@@ -0,0 +1,18 @@
+module FleetAdapter
+  module Routes
+    class Healthcheck < Base
+
+      get '/healthcheck' do
+        headers 'Content-Type' => 'text/plain'
+
+        begin
+          Fleet.new.list_machines
+          'true'
+        rescue
+          'false'
+        end
+      end
+
+    end
+  end
+end

spec/api/healthcheck_spec.rb

@@ -0,0 +1,41 @@
+require 'spec_helper'
+
+describe FleetAdapter::Routes::Healthcheck do
+
+  describe 'GET /healthcheck' do
+
+    before do
+      allow_any_instance_of(Fleet::Client).to receive(:list_machines)
+    end
+
+    it 'has a text/plain Content-Type' do
+      get '/healthcheck'
+      expect(last_response.headers['Content-Type']).to eq 'text/plain'
+    end
+
+    it 'returns a 200 status' do
+      get '/healthcheck'
+      expect(last_response.status).to eq 200
+    end
+
+    context 'when Fleet is healthy' do
+      it 'returns true' do
+        get '/healthcheck'
+        expect(last_response.body).to eq 'true'
+      end
+    end
+
+    context 'when Fleet is NOT healthy' do
+
+      before do
+        allow_any_instance_of(Fleet::Client).to receive(:list_machines)
+          .and_raise('oops')
+      end
+
+      it 'returns false' do
+        get '/healthcheck'
+        expect(last_response.body).to eq 'false'
+      end
+    end
+  end
+end

spec/spec_helper.rb

@@ -0,0 +1,17 @@
+require 'rack/test'
+
+require File.expand_path '../../app.rb', __FILE__
+
+ENV['RACK_ENV'] = 'test'
+
+module ApiType
+  include Rack::Test::Methods
+
+  def app
+    FleetAdapter::App
+  end
+end
+
+RSpec.configure do |c|
+  c.include ApiType, type: :api, file_path: %r(spec/api)
+end