gorgon 0.0.1

data/spec/originator_logger_spec.rb

@@ -0,0 +1,40 @@
+require 'gorgon/originator_logger'
+
+describe OriginatorLogger do
+  before do
+    OriginatorLogger.any_instance.stub(:initialize_logger)
+  end
+
+  let (:originator_logger) { OriginatorLogger.new "" }
+
+  describe "#log_message" do
+    it "prints start messages" do
+      payload = {:action => "start",
+                 :hostname => "host",
+                 :filename => "filename"}
+      originator_logger.should_receive(:log).with("Started running 'filename' at 'host'")
+      originator_logger.log_message(payload)
+    end
+
+    it "prints finish messages" do
+      payload = {:action => "finish",
+                 :hostname => "host",
+                 :filename => "filename"}
+      originator_logger.should_receive(:log).with("Finished running 'filename' at 'host'")
+      originator_logger.log_message(payload)
+    end
+
+    it "prints failure messages when a test finishes with failures" do
+      payload = {:action => "finish",
+                 :type => "fail",
+                 :hostname => "host",
+                 :filename => "filename",
+                 :failures => [
+                   "failure"
+                 ]}
+
+      originator_logger.should_receive(:log).with("Finished running 'filename' at 'host'failure\n")
+      originator_logger.log_message(payload)
+    end
+  end
+end

data/spec/originator_protocol_spec.rb

@@ -0,0 +1,134 @@
+require 'gorgon/originator_protocol'
+
+describe OriginatorProtocol do
+  let(:connection) { stub("Connection", :disconnect => nil, :on_closed => nil)}
+  let(:queue) { stub("Queue", :bind => nil, :subscribe => nil, :name => "queue", :purge => nil,
+                     :delete => nil) }
+  let(:exchange) { stub("Exchange", :publish => nil, :name => "exchange") }
+  let(:channel) { stub("Channel", :queue => queue, :direct => exchange, :fanout => exchange,
+                       :default_exchange => exchange) }
+  let(:logger){ stub("Logger", :log => nil)}
+
+  before do
+    AMQP.stub!(:connect).and_return connection
+    AMQP::Channel.stub!(:new).and_return channel
+    @originator_p = OriginatorProtocol.new logger
+    @conn_information = {:host => "host"}
+  end
+
+  describe "#connect" do
+    it "opens AMQP connection" do
+      AMQP.should_receive(:connect).with(@conn_information)
+      @originator_p.connect(@conn_information)
+    end
+
+    it "opens a new channel" do
+      AMQP::Channel.should_receive(:new).with(connection)
+      @originator_p.connect @conn_information
+    end
+
+    it "sets Connection#on_close callbacks" do
+      on_disconnect_called = false
+      on_disconnect = Proc.new { on_disconnect_called = true }
+      connection.should_receive(:on_closed).and_yield
+
+      @originator_p.connect @conn_information, :on_closed => on_disconnect
+      @originator_p.disconnect
+      on_disconnect_called.should be_true
+    end
+
+    it "opens a reply and exchange queue" do
+      UUIDTools::UUID.stub!(:timestamp_create).and_return 1
+      channel.should_receive(:queue).twice.with("1")
+      @originator_p.connect @conn_information
+    end
+
+    it "opens a reply exchange and binds reply queue to it" do
+      UUIDTools::UUID.stub!(:timestamp_create).and_return 1
+      channel.should_receive(:direct).with("1")
+      queue.should_receive(:bind).with(exchange)
+      @originator_p.connect @conn_information
+    end
+  end
+
+  describe "#publish_files" do
+    before do
+      @originator_p.connect @conn_information
+    end
+
+    it "publish each file using channel's default_exchange" do
+      files = ["file1", "file2"]
+      channel.should_receive(:default_exchange)
+      exchange.should_receive(:publish).once.ordered.with("file1", :routing_key => "queue")
+      exchange.should_receive(:publish).once.ordered.with("file2", :routing_key => "queue")
+      @originator_p.publish_files files
+    end
+  end
+
+  describe "#publish_job" do
+    before do
+      @originator_p.connect @conn_information
+    end
+
+    it "add queue's names to job_definition and fanout using 'gorgon.jobs' exchange" do
+      channel.should_receive(:fanout).with("gorgon.jobs")
+      exp_job_definition = JobDefinition.new
+      exp_job_definition.file_queue_name = "queue"
+      exp_job_definition.reply_exchange_name = "exchange"
+
+      exchange.should_receive(:publish).with(exp_job_definition.to_json)
+      @originator_p.publish_job JobDefinition.new
+    end
+  end
+
+  describe "#receive_payloads" do
+    before do
+      @originator_p.connect @conn_information
+    end
+
+    it "subscribe to reply_queue and yield payload" do
+      payload = {:key => "info"}
+      queue.should_receive(:subscribe).and_yield(payload)
+      yielded = false
+      @originator_p.receive_payloads do |p|
+        yielded = true
+        p.should == payload
+      end
+      yielded.should be_true
+    end
+  end
+
+  describe "#cancel_job" do
+    before do
+      @originator_p.connect @conn_information
+    end
+
+    it "purges file_queue" do
+      queue.should_receive(:purge)
+      @originator_p.cancel_job
+    end
+
+    it "fanout 'cancel' message using 'gorgon.worker_managers' exchange" do
+      msg = Yajl::Encoder.encode({:action => "cancel_job"})
+      channel.should_receive(:fanout).with("gorgon.worker_managers")
+      exchange.should_receive(:publish).with(msg)
+      @originator_p.cancel_job
+    end
+  end
+
+  describe "#disconnect" do
+    before do
+      @originator_p.connect @conn_information
+    end
+
+    it "deletes reply and file queue" do
+      queue.should_receive(:delete).twice
+      @originator_p.disconnect
+    end
+
+    it "disconnects connection" do
+      connection.should_receive(:disconnect)
+      @originator_p.disconnect
+    end
+  end
+end

data/spec/originator_spec.rb

@@ -0,0 +1,134 @@
+require 'gorgon/originator'
+
+describe Originator do
+  let(:protocol){ stub("Originator Protocol", :connect => nil, :publish_files => nil,
+                       :publish_job => nil, :receive_payloads => nil, :cancel_job => nil,
+                       :disconnect => nil)}
+
+  let(:configuration){ {:files => ["some/file"]}}
+  let(:job_state){ stub("JobState", :is_job_complete? => false, :file_finished => nil,
+                        :add_observer => nil)}
+  let(:progress_bar_view){ stub("Progress Bar View", :show => nil)}
+  let(:originator_logger){ stub("Originator Logger", :log => nil, :log_message => nil)}
+
+  before do
+    OriginatorLogger.stub(:new).and_return originator_logger
+    @originator = Originator.new
+  end
+
+  describe "#publish_job" do
+    before do
+      stub_methods
+    end
+
+    it "creates a JobState instance and passes total files" do
+      @originator.stub!(:files).and_return ["a file", "other file"]
+      JobState.should_receive(:new).with(2).and_return job_state
+
+      @originator.publish
+    end
+
+    it "creates a ProgressBarView and show" do
+      JobState.stub!(:new).and_return job_state
+      ProgressBarView.should_receive(:new).with(job_state).and_return progress_bar_view
+      progress_bar_view.should_receive(:show)
+      @originator.publish
+    end
+  end
+
+  describe "#cancel_job" do
+    before do
+      stub_methods
+    end
+
+    it "call JobState#cancel" do
+      JobState.stub!(:new).and_return job_state
+      job_state.should_receive(:cancel)
+      @originator.publish
+      @originator.cancel_job
+    end
+
+    it "tells @protocol to cancel job and disconnect" do
+      protocol.should_receive(:cancel_job)
+      protocol.should_receive(:disconnect)
+      @originator.publish
+      @originator.cancel_job
+    end
+  end
+
+  describe "#cleanup_if_job_complete" do
+    before do
+      stub_methods
+      JobState.stub!(:new).and_return job_state
+      @originator.publish
+    end
+
+    it "calls JobState#is_job_complete?" do
+      job_state.should_receive(:is_job_complete?).and_return false
+      @originator.cleanup_if_job_complete
+    end
+
+    it "disconnect if job is complete" do
+      job_state.stub!(:is_job_complete?).and_return true
+      protocol.should_receive(:disconnect)
+      @originator.cleanup_if_job_complete
+    end
+  end
+
+  describe "#handle_reply" do
+    before do
+      stub_methods
+      JobState.stub!(:new).and_return job_state
+      @originator.publish
+    end
+
+    it "calls cleanup_if_job_complete" do
+      @originator.should_receive(:cleanup_if_job_complete)
+      @originator.handle_reply finish_payload
+    end
+
+    it "calls JobState#file_started if payload[:action] is 'start'" do
+      payload = Yajl::Parser.new(:symbolize_keys => true).parse(start_payload)
+      job_state.should_receive(:file_started)
+      @originator.handle_reply(start_payload)
+    end
+
+    it "calls JobState#file_finished if payload[:action] is 'finish'" do
+      payload = Yajl::Parser.new(:symbolize_keys => true).parse(finish_payload)
+      job_state.should_receive(:file_finished).with(payload)
+      @originator.handle_reply(finish_payload)
+    end
+  end
+
+  private
+
+  def stub_methods
+    EventMachine.stub!(:run).and_yield
+    ProgressBarView.stub!(:new).and_return progress_bar_view
+    OriginatorProtocol.stub!(:new).and_return protocol
+    @originator.stub!(:configuration).and_return configuration
+    @originator.stub!(:connection_information).and_return 'host'
+    @originator.stub!(:job_definition).and_return JobDefinition.new
+  end
+
+  def start_payload
+    '{
+      "action": "start",
+      "hostname": "host",
+      "worker_id": "1",
+      "filename": "test/file_test.rb"
+    }'
+  end
+
+  def finish_payload
+    '{
+      "action": "finish",
+      "hostname": "host",
+      "worker_id": "1",
+      "filename": "test/file_test.rb",
+      "failures": [],
+      "type": "pass",
+      "time": 3
+    }'
+  end
+end

data/spec/progress_bar_view_spec.rb

@@ -0,0 +1,98 @@
+require 'gorgon/progress_bar_view'
+require 'gorgon/job_state'
+
+describe ProgressBarView do
+  describe "#initialize" do
+    it "adds itself to observers of job_state" do
+      job_state = JobState.new 1
+      job_state.should_receive(:add_observer)
+      ProgressBarView.new job_state
+    end
+  end
+
+  describe "#show" do
+    before do
+      job_state = JobState.new 1
+      @progress_bar_view = ProgressBarView.new job_state
+    end
+
+    it "prints a message in console saying that is loading workers" do
+      $stdout.should_receive(:write).with(/loading .*workers/i)
+      ProgressBar.should_not_receive(:create)
+      @progress_bar_view.show
+    end
+  end
+
+  describe "#update" do
+    let(:progress_bar) { stub("Progress Bar", :title= => nil, :progress= => nil, :format => nil,
+                              :finished? => false)}
+    let(:payload) {
+      { :filename => "path/file.rb",
+        :hostname => "host",
+        :failures => ["Failure messages"]}
+    }
+
+    before do
+      ProgressBar.stub!(:create).and_return progress_bar
+      @job_state = JobState.new 2
+      @job_state.stub!(:state).and_return :running
+      @progress_bar_view = ProgressBarView.new @job_state
+      $stdout.stub!(:write)
+      @progress_bar_view.show
+    end
+
+    it "doesn't create ProgressBar if JobState is not running" do
+      @job_state.should_receive(:state).and_return :starting
+      ProgressBar.should_not_receive(:create).with(hash_including(:total => 2))
+      @progress_bar_view.update
+    end
+
+    it "doesn't create a ProgressBar if one was already created" do
+      @progress_bar_view.update
+      ProgressBar.should_not_receive(:create).with(hash_including(:total => 2))
+      @progress_bar_view.update
+    end
+
+    it "gets total files from JobState and create a ProgressBar once JobState is running" do
+      ProgressBar.should_receive(:create).with(hash_including(:total => 2))
+      @progress_bar_view.update
+    end
+
+    it "gets finished_files_count" do
+      @job_state.should_receive :finished_files_count
+      @progress_bar_view.update
+    end
+
+    it "gets failed_files_count" do
+      @job_state.should_receive(:failed_files_count).and_return 0
+      @progress_bar_view.update
+    end
+
+    it "prints failures and finish progress_bar when job is done" do
+      @progress_bar_view.update
+      @job_state.stub!(:each_failed_test).and_yield(payload)
+      @job_state.stub!(:is_job_complete?).and_return :true
+      $stdout.should_receive(:write).with(/Failure messages/)
+      @progress_bar_view.update
+    end
+
+    context "when job is cancelled" do
+      before do
+        @progress_bar_view.update
+        @job_state.stub!(:is_job_cancelled?).and_return :true
+      end
+
+      it "prints failures and finish progress_bar when job is cancelled" do
+        @job_state.stub!(:each_failed_test).and_yield(payload)
+        $stdout.should_receive(:write).with(/Failure messages/)
+        @progress_bar_view.update
+      end
+
+      it "prints files that were running when the job was cancelled" do
+        @job_state.should_receive(:each_running_file).and_yield("hostname", "file1.rb")
+        $stdout.should_receive(:write).with(/file1\.rb.*hostname/)
+        @progress_bar_view.update
+      end
+    end
+  end
+end

data/spec/source_tree_syncer_spec.rb

@@ -0,0 +1,65 @@
+require 'gorgon/source_tree_syncer'
+
+describe SourceTreeSyncer.new("") do
+  it { should respond_to :exclude= }
+  it { should respond_to :sync }
+  it { should respond_to :sys_command }
+  it { should respond_to :remove_temp_dir }
+
+  describe "#sync" do
+    before do
+      @syncer = SourceTreeSyncer.new "path/to/source"
+      stub_utilities_methods
+    end
+
+    it "makes tempdir and changes current dir to temdir" do
+      Dir.should_receive(:mktmpdir).and_return("tmp/dir")
+      Dir.should_receive(:chdir).with("tmp/dir")
+      @syncer.sync
+    end
+
+    it "runs rsync system command with appropriate options" do
+      cmd = /rsync.*-az.*-r --rsh=ssh path\/to\/source\/\* \./
+      @syncer.should_receive(:system).with(cmd)
+      @syncer.sync
+    end
+
+    it "exclude files when they are specified" do
+      @syncer.exclude = ["log", ".git"]
+      @syncer.should_receive(:system).with(/--exclude log --exclude .git/)
+      @syncer.sync
+    end
+
+    it "returns true if sys command execution was successful" do
+      $?.stub!(:exitstatus).and_return 0
+      @syncer.sync.should be_true
+    end
+
+    it "returns false if sys command execution failed" do
+      $?.stub!(:exitstatus).and_return 1
+      @syncer.sync.should be_false
+    end
+  end
+
+  describe "#remove_temp_dir" do
+    before do
+      @syncer = SourceTreeSyncer.new "path/to/source"
+      stub_utilities_methods
+      @syncer.sync
+    end
+
+    it "remove temporary dir" do
+      FileUtils.should_receive(:remove_entry_secure).with("tmp/dir")
+      @syncer.remove_temp_dir
+    end
+  end
+
+  private
+
+  def stub_utilities_methods
+    Dir.stub!(:mktmpdir).and_return("tmp/dir")
+    Dir.stub!(:chdir)
+    FileUtils.stub!(:remove_entry_secure)
+    @syncer.stub!(:system)
+  end
+end