add some cleanup testing

Fixes elastic#2

Author: Pere Urbon-Bayes

lib/lsit/run.rb

@@ -4,13 +4,13 @@
 
 require 'lsit/stats'
 
-INITIAL_MESSAGE = ">>> lorem ipsum start".freeze
-LAST_MESSAGE = ">>> lorem ipsum stop".freeze
-
 Thread.abort_on_exception = true
 
 class Runner
   LOGSTASH_BIN  = File.join("bin", "logstash").freeze
+
+  INITIAL_MESSAGE = ">>> lorem ipsum start".freeze
+  LAST_MESSAGE = ">>> lorem ipsum stop".freeze
   REFRESH_COUNT = 100
 
   attr_reader :command
@@ -62,11 +62,8 @@ def read_input_file(file_path)
     IO.readlines(file_path).map(&:chomp)
   end
 
-  private
-
   def feed_input_events(io, events_count, lines, last_message)
     loop_count = (events_count / lines.size).ceil # how many time we send the input file over
-
     (1..loop_count).each{lines.each {|line| io.puts(line)}}
 
     io.puts(last_message)
@@ -75,6 +72,8 @@ def feed_input_events(io, events_count, lines, last_message)
     loop_count * lines.size
   end
 
+  private
+
   def feed_input_interval(io, seconds, lines, last_message)
     loop_count = (2000 / lines.size).ceil # check time every ~2000(ceil) input lines
     lines_per_iteration = loop_count * lines.size

spec/lib/runner_spec.rb

@@ -3,20 +3,32 @@
 describe Runner do
 
   let(:config) { 'spec/fixtures/simple.conf' }
-  let(:lines)  { load_fixture 'simple_10.txt' }
+  let(:lines)  { load_fixture('simple_10.txt').split("\n") }
 
   let(:events) { 30 }
   let(:time)   { 10 }
 
   subject (:runner) { Runner.new(config) }
 
+  let(:command) { [Runner::LOGSTASH_BIN, "-f", "spec/fixtures/simple.conf"]}
+
+  it "invokes the logstash command" do
+    Open3.should_receive(:popen3).with(*command).and_return(true)
+    runner.run(events, 0, lines)
+  end
+
   context "#execution with number of events" do
 
-    let(:command) { [Runner::LOGSTASH_BIN, "-f", "spec/fixtures/simple.conf"]}
+    let(:io)       { double("io", :puts => true, :flush => true) }
+    subject(:feed) { runner.feed_input_with(events, 0, lines, io) }
+
+    it "feeds in terms of events" do
+      expect(runner).to receive(:feed_input_events).with(io, events, lines, Runner::LAST_MESSAGE)
+      runner.feed_input_with(events, 0, lines, io)
+    end
 
-    it "should invoke the logstash command" do
-      Open3.should_receive(:popen3).with(*command).and_return(true)
-      runner.run(events, 0, lines)
+    it "feed the input stream with events.size+1 events" do
+      expect(feed).to eq(events.to_i)
     end
 
   end