right_agent 2.0.7-x86-mingw32

data/spec/clients/spec_helper.rb

@@ -0,0 +1,111 @@
+#
+# Copyright (c) 2009-2011 RightScale Inc
+#
+# Permission is hereby granted, free of charge, to any person obtaining
+# a copy of this software and associated documentation files (the
+# "Software"), to deal in the Software without restriction, including
+# without limitation the rights to use, copy, modify, merge, publish,
+# distribute, sublicense, and/or sell copies of the Software, and to
+# permit persons to whom the Software is furnished to do so, subject to
+# the following conditions:
+#
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+require 'restclient'
+
+require ::File.expand_path('../../spec_helper', __FILE__)
+
+# Mock auth client providing basic support needed by various clients
+class AuthClientMock < RightScale::AuthClient
+  attr_reader :test_url, :expired_called, :redirect_location
+
+  def initialize(url, auth_header, state = nil, account_id = nil, identity = nil)
+    @test_url = @api_url = @router_url = url
+    @auth_header = auth_header
+    @account_id = account_id
+    @identity = identity || "rs-agent-1-1"
+    @state = :authorized if state
+  end
+
+  def identity
+    @identity
+  end
+
+  def headers
+    auth_header
+  end
+
+  def auth_header
+    @auth_header
+  end
+
+  def expired
+    @expired_called = true
+  end
+
+  def redirect(location)
+    @redirect_location = location
+  end
+end
+
+# Mock WebSocket event
+class WebSocketEventMock
+  attr_reader :code, :data, :reason
+
+  def initialize(data, code = nil, reason = nil)
+    @code = code
+    @data = data
+    @reason = reason
+  end
+end
+
+# Mock of WebSocket so that can call on methods
+class WebSocketClientMock
+  attr_reader :sent, :closed, :code, :reason
+
+  def send(event)
+    @sent = @sent.nil? ? event : (@sent.is_a?(Array) ? @sent << event : [@sent, event])
+  end
+
+  def close(code = nil, reason = nil)
+    @code = code
+    @reason = reason
+    @closed = true
+  end
+
+  def onclose=(block)
+    @close_block = block
+  end
+
+  def onclose(code, reason = nil)
+    @event = WebSocketEventMock.new(nil, code, reason)
+    @close_block.call(@event)
+  end
+
+  def onerror=(block)
+    @error_block = block
+  end
+
+  def onerror(data)
+    @event = WebSocketEventMock.new(data)
+    @error_block.call(@event)
+  end
+
+  def onmessage=(block)
+    @message_block = block
+  end
+
+  def onmessage(data)
+    @event = WebSocketEventMock.new(data)
+    @message_block.call(@event)
+  end
+end

data/spec/command/agent_manager_commands_spec.rb

@@ -0,0 +1,51 @@
+#
+# Copyright (c) 2009 RightScale Inc
+#
+# Permission is hereby granted, free of charge, to any person obtaining
+# a copy of this software and associated documentation files (the
+# "Software"), to deal in the Software without restriction, including
+# without limitation the rights to use, copy, modify, merge, publish,
+# distribute, sublicense, and/or sell copies of the Software, and to
+# permit persons to whom the Software is furnished to do so, subject to
+# the following conditions:
+#
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+require File.expand_path(File.join(File.dirname(__FILE__), '..', 'spec_helper'))
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', 'lib', 'right_agent', 'command'))
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', 'lib', 'right_agent', 'command', 'agent_manager_commands'))
+
+describe RightScale::AgentManagerCommands do
+
+  before(:all) do
+    @commands = RightScale::AgentManagerCommands::COMMANDS
+    @agent_identity = RightScale::AgentIdentity.new('rs', 'test', 1).to_s
+  end
+
+  it 'should list commands' do
+    flexmock(RightScale::CommandIO.instance).should_receive(:reply).and_return do |conn, r|
+      conn.should == 42
+      # r is YAML, 2 lines for each command, one less command printed (the list command)
+      # plus one header line
+      r.count("\n").should == (@commands.reject {|k,_| k.to_s =~ /test/}.size - 1) * 2 + 1
+    end
+    RightScale::AgentManagerCommands.new("agent_manager").send(:list_command, {:conn => 42}).should be_true
+  end
+
+  it 'should get commands' do
+    cmds = RightScale::AgentManagerCommands.get("agent_manager")
+    cmds.size.should == @commands.size
+    cmds.keys.map { |k| k.to_s }.sort.should == @commands.keys.map { |k| k.to_s }.sort
+    cmds.values.all? { |v| v.is_a? Proc }.should be_true
+  end
+
+end

data/spec/command/command_io_spec.rb

@@ -0,0 +1,93 @@
+#
+# Copyright (c) 2009-2011 RightScale Inc
+#
+# Permission is hereby granted, free of charge, to any person obtaining
+# a copy of this software and associated documentation files (the
+# "Software"), to deal in the Software without restriction, including
+# without limitation the rights to use, copy, modify, merge, publish,
+# distribute, sublicense, and/or sell copies of the Software, and to
+# permit persons to whom the Software is furnished to do so, subject to
+# the following conditions:
+#
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+require File.expand_path(File.join(File.dirname(__FILE__), '..', 'spec_helper'))
+
+module RightScale
+  module CommandIoSpec
+
+    # ensure uniqueness of handler to avoid confusion.
+    raise "#{ClientOutputHandler.name} is already defined" if defined?(ClientOutputHandler)
+
+    module ClientOutputHandler
+      def initialize(input)
+        @input = input
+       end
+      def post_init
+        send_data(RightScale::CommandSerializer.dump(@input))
+        close_connection_after_writing
+      end
+    end
+  end
+end
+
+describe RightScale::CommandIO do
+
+  # Serialize and send given input to command listener
+  def send_input(input)
+    
+    EM.connect('127.0.0.1', @socket_port, ::RightScale::CommandIoSpec::ClientOutputHandler, input)
+  end
+
+  before(:all) do
+    @socket_port = TEST_SOCKET_PORT
+  end
+
+  it 'should detect missing blocks' do
+    lambda { RightScale::CommandIO.instance.listen(@socket_port) }.should raise_error(ArgumentError)
+  end
+
+  it 'should receive a command' do
+    @input = ''
+    EM.run do
+      RightScale::CommandIO.instance.listen(@socket_port) { |input, _| @input = input; stop }
+      send_input('input')
+      EM.add_timer(2) { stop }
+    end
+    @input.should == 'input'
+  end
+
+  it 'should receive many commands' do
+    @inputs = []
+    EM.run do
+      RightScale::CommandIO.instance.listen(@socket_port) do |input, _|
+        @inputs << input
+        stop if input == 'final'
+      end
+      for i in 1..50 do
+        send_input("input#{i}")
+      end
+      send_input("final")
+      EM.add_timer(2) { stop }
+    end
+
+    @inputs.size.should == 51
+    (0..49).each { |i| @inputs[i].should == "input#{i+1}" }
+    @inputs[50].should == 'final'
+  end
+
+  def stop
+    RightScale::CommandIO.instance.stop_listening
+    EM.stop
+  end
+
+end

data/spec/command/command_parser_spec.rb

@@ -0,0 +1,79 @@
+#
+# Copyright (c) 2009-2011 RightScale Inc
+#
+# Permission is hereby granted, free of charge, to any person obtaining
+# a copy of this software and associated documentation files (the
+# "Software"), to deal in the Software without restriction, including
+# without limitation the rights to use, copy, modify, merge, publish,
+# distribute, sublicense, and/or sell copies of the Software, and to
+# permit persons to whom the Software is furnished to do so, subject to
+# the following conditions:
+#
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+require File.expand_path(File.join(File.dirname(__FILE__), '..', 'spec_helper'))
+
+describe RightScale::CommandParser do
+
+  it 'should detect missing block' do
+    lambda { RightScale::CommandParser.new }.should raise_error(ArgumentError)
+  end
+
+  it 'should parse' do
+    @parser = RightScale::CommandParser.new { |cmd| @command = cmd; EM.stop }
+    EM.run do
+      @parser.parse_chunk(RightScale::CommandSerializer.dump(42)).should be_true
+      EM.add_timer(0.5) { EM.stop }
+    end
+    @command.should == 42
+  end
+
+  it 'should parse in chunks' do
+    data = RightScale::CommandSerializer.dump({ :some => 'random', :and => 'long', :serialized => 'data' })
+    data.size.should > 6
+    parts = data[0..2], data[3..4], data[5..6], data[7..data.size - 1]
+    command = ''
+    parser = RightScale::CommandParser.new { |cmd| command = cmd; EM.stop }
+    EM.run do
+      parts.each { |p| parser.parse_chunk(p) }
+      EM.add_timer(0.5) { EM.stop }
+    end
+    command.should == { :some => 'random', :and => 'long', :serialized => 'data' }
+  end
+
+  it 'should parse multiple commands' do
+    commands = []
+    sample_data = [42, {:question => 'why?', :answer => 'fourty-two'}]
+    parser = RightScale::CommandParser.new { |cmd| commands << cmd; EM.stop if commands.size == sample_data.size }
+    serialized = sample_data.inject('') { |s, cmd| s << RightScale::CommandSerializer.dump(cmd) }
+    EM.run do
+      parser.parse_chunk(serialized).should be_true
+      EM.add_timer(0.5) { EM.stop }
+    end
+    commands.should == sample_data
+  end
+
+  it 'should parse multiple commands in chunks' do
+    commands = []
+    sample_data = [42, {:question => 'why?', :answer => 'fourty-two'}]
+    parser = RightScale::CommandParser.new { |cmd| commands << cmd; EM.stop if commands.size == sample_data.size }
+    serialized = sample_data.inject('') { |s, cmd| s << RightScale::CommandSerializer.dump(cmd) }
+    serialized.size.should > 10
+    parts = serialized[0..2], serialized[3..4], serialized[5..6], serialized[7..serialized.size - 5], serialized[serialized.size - 4..serialized.size - 1]
+    EM.run do
+      parts.each { |p| parser.parse_chunk(p) }
+      EM.add_timer(0.5) { EM.stop }
+    end
+    commands.should == sample_data
+  end
+
+end

data/spec/command/command_runner_spec.rb

@@ -0,0 +1,107 @@
+#
+# Copyright (c) 2009-2011 RightScale Inc
+#
+# Permission is hereby granted, free of charge, to any person obtaining
+# a copy of this software and associated documentation files (the
+# "Software"), to deal in the Software without restriction, including
+# without limitation the rights to use, copy, modify, merge, publish,
+# distribute, sublicense, and/or sell copies of the Software, and to
+# permit persons to whom the Software is furnished to do so, subject to
+# the following conditions:
+#
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+require File.expand_path(File.join(File.dirname(__FILE__), '..', 'spec_helper'))
+
+module RightScale
+
+  class CommandIOMock < CommandIO
+
+    include RightSupport::Ruby::EasySingleton
+
+    def trigger_listen(payload)
+      @test_callback.call(payload)
+    end
+
+    def listen(socket_port, &block)
+      @test_callback = block
+      true
+    end
+
+  end
+
+end
+
+describe RightScale::CommandRunner do
+
+  before(:all) do
+    @command_payload = { :name => 'test', :options => 'options' }
+    @socket_port = TEST_SOCKET_PORT
+  end
+
+  context :start do
+    before(:each) do
+      @pid_file = flexmock(RightScale::PidFile)
+      @pid_file.should_receive(:exists?).and_return(true)
+      @pid_file.should_receive(:set_command_options).and_return(true)
+      flexmock(RightScale::PidFile).should_receive(:new).and_return(@pid_file)
+    end
+
+    context 'when a block is provided' do
+      before(:each) do
+        @block = Proc.new do |pid_file|
+          @callback_pid_file = pid_file
+        end
+      end
+
+      it 'should yield its PidFile' do
+        commands = { :test => lambda { |opt, _| } }
+        flexmock(RightScale::CommandIO).should_receive(:instance).and_return(RightScale::CommandIOMock.instance)
+        RightScale::CommandRunner.start(@socket_port, RightScale::AgentIdentity.generate, commands, &@block)
+        @callback_pid_file.should == @pid_file
+      end
+    end
+  end
+
+  it 'should handle invalid formats' do
+    flexmock(RightScale::CommandIO.instance).should_receive(:listen).and_yield(['invalid yaml'])
+    flexmock(RightScale::Log).should_receive(:info).once
+    RightScale::CommandRunner.start(@socket_port, RightScale::AgentIdentity.generate, commands={})
+  end
+
+  it 'should handle non-existent commands' do
+    flexmock(RightScale::CommandIO.instance).should_receive(:listen).and_yield(@command_payload)
+    flexmock(RightScale::Log).should_receive(:info).once
+    RightScale::CommandRunner.start(@socket_port, RightScale::AgentIdentity.generate, commands={})
+  end
+
+  it 'should run commands' do
+    commands = { :test => lambda { |opt, _| @opt = opt } }
+    flexmock(RightScale::CommandIO).should_receive(:instance).and_return(RightScale::CommandIOMock.instance)
+    cmd_options = RightScale::CommandRunner.start(@socket_port, RightScale::AgentIdentity.generate, commands)
+    payload = @command_payload.merge(cmd_options)
+    RightScale::CommandIOMock.instance.trigger_listen(payload)
+    @opt.should == payload
+  end
+
+  it 'should run commands using fiber pool if provided' do
+    commands = { :test => lambda { |opt, _| @opt = opt } }
+    fiber_pool = flexmock("fiber pool")
+    fiber_pool.should_receive(:spawn).and_return(true).and_yield.once
+    flexmock(RightScale::CommandIO).should_receive(:instance).and_return(RightScale::CommandIOMock.instance)
+    cmd_options = RightScale::CommandRunner.start(@socket_port, RightScale::AgentIdentity.generate, commands, fiber_pool)
+    payload = @command_payload.merge(cmd_options)
+    RightScale::CommandIOMock.instance.trigger_listen(payload)
+    @opt.should == payload
+  end
+
+end

data/spec/command/command_serializer_spec.rb

@@ -0,0 +1,51 @@
+#
+# Copyright (c) 2009-2011 RightScale Inc
+#
+# Permission is hereby granted, free of charge, to any person obtaining
+# a copy of this software and associated documentation files (the
+# "Software"), to deal in the Software without restriction, including
+# without limitation the rights to use, copy, modify, merge, publish,
+# distribute, sublicense, and/or sell copies of the Software, and to
+# permit persons to whom the Software is furnished to do so, subject to
+# the following conditions:
+#
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+require File.expand_path(File.join(File.dirname(__FILE__), '..', 'spec_helper'))
+
+describe RightScale::CommandSerializer do
+
+  before(:all) do
+    @sample_data = [ 42, 'fourty two', { :haha => 42, 'hoho' => 'fourty_two' }]
+  end
+
+  it 'should serialize' do
+    @sample_data.each do |data|
+      RightScale::CommandSerializer.dump(data)
+    end
+  end
+
+  it 'should deserialize' do
+    @sample_data.each do |data|
+      RightScale::CommandSerializer.load(RightScale::CommandSerializer.dump(data)).should == data
+    end
+  end
+
+  it 'should add separators' do
+    serialized = ''
+    @sample_data.each do |data|
+      serialized << RightScale::CommandSerializer.dump(data)
+    end
+    serialized.split(RightScale::CommandSerializer::SEPARATOR).size.should == @sample_data.size
+  end
+
+end