litmus_paper 0.9.2 → 0.9.3

data/lib/litmus_paper/metric/internet_health.rb

@@ -1,7 +1,8 @@
 module LitmusPaper
   module Metric
     class InternetHealth
-      def initialize(hosts, options = {})
+      def initialize(weight, hosts, options = {})
+        @weight = weight
         @hosts = hosts
         @options = options
         @timeout = options.fetch(:timeout_seconds, 5)
@@ -22,7 +23,7 @@ module LitmusPaper
       end
 
       def current_health
-        health = 100 * @hosts.reduce(Rational(0)) do |memo, host|
+        health = @weight * @hosts.reduce(Rational(0)) do |memo, host|
           if tcp_connect?(*host.split(':'))
             memo += Rational(1) / Rational(@hosts.length)
           end
@@ -32,7 +33,7 @@ module LitmusPaper
       end
 
       def to_s
-        "Metric::InternetHealth(#{@hosts.inspect}, #{@options.inspect})"
+        "Metric::InternetHealth(#{@weight}, #{@hosts.inspect}, #{@options.inspect})"
       end
     end
   end

data/lib/litmus_paper/service.rb

@@ -24,6 +24,10 @@ module LitmusPaper
       @checks << metric_class.new(options[:weight])
     end
 
+    def measure_health_with_args(metric_class, *args)
+      @checks << metric_class.new(*args)
+    end
+
     def depends(dependency_class, *args)
       @dependencies << dependency_class.new(*args)
     end

data/lib/litmus_paper/version.rb

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

data/spec/litmus_paper/metric/internet_health_spec.rb

@@ -20,12 +20,12 @@ describe LitmusPaper::Dependency::TCP do
     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 = LitmusPaper::Metric::InternetHealth.new(100, ["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 = LitmusPaper::Metric::InternetHealth.new(100, ["127.0.0.1:6000"])
       internet_health.current_health.should == 0
     end
 
@@ -34,6 +34,7 @@ describe LitmusPaper::Dependency::TCP do
         sleep(5)
       end
       internet_health = LitmusPaper::Metric::InternetHealth.new(
+        100,
         ["127.0.0.1:6000"],
         { :timeout_seconds => 2 },
       )
@@ -41,23 +42,29 @@ describe LitmusPaper::Dependency::TCP do
     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 = LitmusPaper::Metric::InternetHealth.new(
+        100,
+        [
+          "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 = LitmusPaper::Metric::InternetHealth.new(
+        100,
+        [
+          "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
 
@@ -67,6 +74,7 @@ describe LitmusPaper::Dependency::TCP do
         sleep(5)
       end
       internet_health = LitmusPaper::Metric::InternetHealth.new(
+        100,
         ["127.0.0.1:3000", "127.0.0.1:3001"],
         { :timeout_seconds => 2 },
       )
@@ -74,18 +82,21 @@ describe LitmusPaper::Dependency::TCP do
     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 = LitmusPaper::Metric::InternetHealth.new(
+        100,
+        [
+          "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"])
+      internet_health = LitmusPaper::Metric::InternetHealth.new(100, ["127.0.0.1:6000"])
       LitmusPaper.logger.should_receive(:info)
       internet_health.current_health.should == 0
     end
@@ -93,14 +104,17 @@ describe LitmusPaper::Dependency::TCP do
 
   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\"], {})"
+      internet_health = LitmusPaper::Metric::InternetHealth.new(
+        100,
+        [
+          "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(100, [\"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
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: litmus_paper
 version: !ruby/object:Gem::Version
-  version: 0.9.2
+  version: 0.9.3
   prerelease: 
 platform: ruby
 authors: