litmus_paper 0.8.9 → 0.9.0

data/lib/litmus_paper/dependency/tcp.rb

@@ -4,14 +4,25 @@ module LitmusPaper
       def initialize(ip, port, options = {})
         @ip, @port = ip, port
         @timeout = options.fetch(:timeout_seconds, 5)
+        @input_data = options[:input_data]
+        @expected_output = options[:expected_output]
       end
 
       def available?
+        response = true
+
         Timeout.timeout(@timeout) do
           socket = TCPSocket.new(@ip, @port)
+          if @input_data || @expected_output
+            socket.puts(@input_data.to_s)
+            data = socket.gets
+            response = data.chomp == @expected_output
+            LitmusPaper.logger.info("Response (#{response}) does not match expected output (#{@expected_output})")
+          end
           socket.close
         end
-        true
+
+        response
       rescue Timeout::Error
         LitmusPaper.logger.info("Timeout connecting #{@ip}:#{@port}")
         false

data/lib/litmus_paper/version.rb

@@ -1,3 +1,3 @@
 module LitmusPaper
-  VERSION = "0.8.9"
+  VERSION = "0.9.0"
 end

data/spec/litmus_paper/dependency/tcp_spec.rb

@@ -1,15 +1,47 @@
 require 'spec_helper'
 
 describe LitmusPaper::Dependency::TCP do
-  before(:all) do
-    @server = TCPServer.new 3333
-  end
+  describe "accepts input data and compares result against an expected output " do
+    before(:each) do
+      @server = TCPServer.new 7333
+      @thread = Thread.start do
+        s = @server.accept
+        s.puts "+PONG"
+      end
+    end
+
+    after(:each) do
+      @server.close
+      @thread.join
+    end
+
+    describe "#available?" do
+      it "is true when expected_output equals response from socket" do
+        check = LitmusPaper::Dependency::TCP.new("127.0.0.1", 7333, :expected_output => "+PONG", :input_data => "PING")
+        check.should be_available
+      end
 
-  after(:all) do
-    @server.close
+      it "is true when expected_output equals response from socket when no input is supplied" do
+        check = LitmusPaper::Dependency::TCP.new("127.0.0.1", 7333, :expected_output => "+PONG")
+        check.should be_available
+      end
+
+      it "is false when expected_output does not equal response from socket" do
+        check = LitmusPaper::Dependency::TCP.new("127.0.0.1", 7333, :expected_output => "+PANG", :input_data => "PING")
+        check.should_not be_available
+      end
+    end
   end
 
   describe "#available?" do
+    before(:all) do
+      @server = TCPServer.new 3333
+    end
+
+    after(:all) do
+      @server.close
+    end
+
     it "is true when it's able to reach the ip and port" do
       check = LitmusPaper::Dependency::TCP.new("127.0.0.1", 3333)
       check.should be_available

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: litmus_paper
 version: !ruby/object:Gem::Version
-  version: 0.8.9
+  version: 0.9.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-07-07 00:00:00.000000000 Z
+date: 2015-08-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sinatra