litmus_paper 0.9.1 → 0.9.2

data/.travis.yml

@@ -3,6 +3,7 @@ rvm:
 - 1.9.2
 - 1.9.3
 - 2.0.0
+sudo: false
 notifications:
   email:
-    - blue@getbraintree.com
+    - blue-ci-notifications@getbraintree.com

data/lib/litmus_paper/metric/available_memory.rb

@@ -19,7 +19,7 @@ module LitmusPaper
       def memory_total
         return @memory_total unless @memory_total.nil?
 
-        size, scale = @facter.value('memorytotal').split(' ')
+        size, scale = @facter.value('memorysize').split(' ')
         @memory_total = (size.to_f * MULTIPLIER[scale]).to_i
       end
 

data/lib/litmus_paper/metric/internet_health.rb

@@ -0,0 +1,39 @@
+module LitmusPaper
+  module Metric
+    class InternetHealth
+      def initialize(hosts, options = {})
+        @hosts = hosts
+        @options = options
+        @timeout = options.fetch(:timeout_seconds, 5)
+      end
+
+      def tcp_connect?(host, port)
+        Timeout.timeout(@timeout) do
+          socket = TCPSocket.new(host, port)
+          socket.close
+        end
+        true
+      rescue Timeout::Error
+        LitmusPaper.logger.info("Timeout connecting to #{host}:#{port}")
+        false
+      rescue => e
+        LitmusPaper.logger.info("TCP connect to #{host}:#{port} failed with #{e.message}")
+        false
+      end
+
+      def current_health
+        health = 100 * @hosts.reduce(Rational(0)) do |memo, host|
+          if tcp_connect?(*host.split(':'))
+            memo += Rational(1) / Rational(@hosts.length)
+          end
+          memo
+        end
+        health.to_i
+      end
+
+      def to_s
+        "Metric::InternetHealth(#{@hosts.inspect}, #{@options.inspect})"
+      end
+    end
+  end
+end

data/lib/litmus_paper/version.rb

@@ -1,3 +1,3 @@
 module LitmusPaper
-  VERSION = "0.9.1"
+  VERSION = "0.9.2"
 end

data/lib/litmus_paper.rb

@@ -37,6 +37,7 @@ require 'litmus_paper/metric/available_memory'
 require 'litmus_paper/metric/big_brother_service'
 require 'litmus_paper/metric/constant_metric'
 require 'litmus_paper/metric/cpu_load'
+require 'litmus_paper/metric/internet_health'
 require 'litmus_paper/metric/script'
 require 'litmus_paper/service'
 require 'litmus_paper/status_file'

data/spec/litmus_paper/metric/available_memory_spec.rb

@@ -3,13 +3,13 @@ require 'spec_helper'
 describe LitmusPaper::Metric::AvailableMemory do
   describe "#current_health" do
     it "multiplies weight by memory available" do
-      facter = StubFacter.new({"memorytotal" => "10 GB", "memoryfree" => "5 GB"})
+      facter = StubFacter.new({"memorysize" => "10 GB", "memoryfree" => "5 GB"})
       memory = LitmusPaper::Metric::AvailableMemory.new(50, facter)
       memory.current_health.should == 25
     end
 
     it "multiplies weight by memory available when handling floating point values" do
-      facter = StubFacter.new({"memorytotal" => "2.0 GB", "memoryfree" => "1.8 GB"})
+      facter = StubFacter.new({"memorysize" => "2.0 GB", "memoryfree" => "1.8 GB"})
       memory = LitmusPaper::Metric::AvailableMemory.new(50, facter)
       memory.current_health.should == 44
     end
@@ -21,7 +21,7 @@ describe LitmusPaper::Metric::AvailableMemory do
       end
 
       it "handles floating point values properly" do
-        facter = StubFacter.new("memorytotal" => "1.80 GB")
+        facter = StubFacter.new("memorysize" => "1.80 GB")
         memory = LitmusPaper::Metric::AvailableMemory.new(50, facter)
         memory.memory_total.should == 1932735283
       end

data/spec/litmus_paper/metric/internet_health_spec.rb

@@ -0,0 +1,107 @@
+require 'spec_helper'
+
+describe LitmusPaper::Dependency::TCP do
+
+  describe "#current_health" do
+    before(:all) do
+      @server_3000 = TCPServer.new 3000
+      @server_3001 = TCPServer.new 3001
+      @server_3002 = TCPServer.new 3002
+      @server_3003 = TCPServer.new 3003
+      @server_3004 = TCPServer.new 3004
+    end
+
+    after(:all) do
+      @server_3000.close
+      @server_3001.close
+      @server_3002.close
+      @server_3003.close
+      @server_3004.close
+    end
+
+    it "returns 100 when it's able to reach a single host" do
+      internet_health = LitmusPaper::Metric::InternetHealth.new(["127.0.0.1:3000"])
+      internet_health.current_health.should == 100
+    end
+
+    it "returns 0 when it's unable to reach a single host" do
+      internet_health = LitmusPaper::Metric::InternetHealth.new(["127.0.0.1:6000"])
+      internet_health.current_health.should == 0
+    end
+
+    it "returns 0 when it's request to a single host times out" do
+      TCPSocket.stub(:new) do
+        sleep(5)
+      end
+      internet_health = LitmusPaper::Metric::InternetHealth.new(
+        ["127.0.0.1:6000"],
+        { :timeout_seconds => 2 },
+      )
+      internet_health.current_health.should == 0
+    end
+
+    it "returns 100 when it's able to reach multiple hosts" do
+      internet_health = LitmusPaper::Metric::InternetHealth.new([
+        "127.0.0.1:3000",
+        "127.0.0.1:3001",
+        "127.0.0.1:3002",
+        "127.0.0.1:3003",
+        "127.0.0.1:3004",
+      ])
+      internet_health.current_health.should == 100
+    end
+
+    it "returns 50 when it's unable to reach half the hosts" do
+      internet_health = LitmusPaper::Metric::InternetHealth.new([
+        "127.0.0.1:3000",
+        "127.0.0.1:3001",
+        "127.0.0.1:6002",
+        "127.0.0.1:6003",
+      ])
+      internet_health.current_health.should == 50
+    end
+
+    it "returns 50 when it's request to a single host out of two hosts times out" do
+      TCPSocket.stub(:new) do
+        TCPSocket.unstub(:new)
+        sleep(5)
+      end
+      internet_health = LitmusPaper::Metric::InternetHealth.new(
+        ["127.0.0.1:3000", "127.0.0.1:3001"],
+        { :timeout_seconds => 2 },
+      )
+      internet_health.current_health.should == 50
+    end
+
+    it "returns 0 when it's unable to reach any of the hosts" do
+      internet_health = LitmusPaper::Metric::InternetHealth.new([
+        "127.0.0.1:6000",
+        "127.0.0.1:6001",
+        "127.0.0.1:6002",
+        "127.0.0.1:6003",
+        "127.0.0.1:6004",
+      ])
+      internet_health.current_health.should == 0
+    end
+
+    it "logs exceptions and returns 0" do
+      internet_health = LitmusPaper::Metric::InternetHealth.new(["127.0.0.1:6000"])
+      LitmusPaper.logger.should_receive(:info)
+      internet_health.current_health.should == 0
+    end
+  end
+
+  describe "to_s" do
+    it "is the name of the class and the lists of hosts" do
+      internet_health = LitmusPaper::Metric::InternetHealth.new([
+        "127.0.0.1:6000",
+        "127.0.0.1:6001",
+        "127.0.0.1:6002",
+        "127.0.0.1:6003",
+        "127.0.0.1:6004",
+      ])
+      internet_health.to_s.should == "Metric::InternetHealth([\"127.0.0.1:6000\", \"127.0.0.1:6001\", \"127.0.0.1:6002\", \"127.0.0.1:6003\", \"127.0.0.1:6004\"], {})"
+    end
+  end
+
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: litmus_paper
 version: !ruby/object:Gem::Version
-  version: 0.9.1
+  version: 0.9.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-12-08 00:00:00.000000000 Z
+date: 2017-03-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sinatra
@@ -213,6 +213,7 @@ files:
 - lib/litmus_paper/metric/big_brother_service.rb
 - lib/litmus_paper/metric/constant_metric.rb
 - lib/litmus_paper/metric/cpu_load.rb
+- lib/litmus_paper/metric/internet_health.rb
 - lib/litmus_paper/metric/script.rb
 - lib/litmus_paper/service.rb
 - lib/litmus_paper/status_file.rb
@@ -232,6 +233,7 @@ files:
 - spec/litmus_paper/metric/big_brother_service_spec.rb
 - spec/litmus_paper/metric/constant_metric_spec.rb
 - spec/litmus_paper/metric/cpu_load_spec.rb
+- spec/litmus_paper/metric/internet_health_spec.rb
 - spec/litmus_paper/metric/script_spec.rb
 - spec/litmus_paper/service_spec.rb
 - spec/litmus_paper/status_file_spec.rb
@@ -291,6 +293,7 @@ test_files:
 - spec/litmus_paper/metric/big_brother_service_spec.rb
 - spec/litmus_paper/metric/constant_metric_spec.rb
 - spec/litmus_paper/metric/cpu_load_spec.rb
+- spec/litmus_paper/metric/internet_health_spec.rb
 - spec/litmus_paper/metric/script_spec.rb
 - spec/litmus_paper/service_spec.rb
 - spec/litmus_paper/status_file_spec.rb