right_agent 0.5.1

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(RightScale::Exceptions::Argument)
+  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(RightScale::Exceptions::Argument)
+  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,72 @@
+#
+# 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 Singleton
+
+    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
+
+  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
+
+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

data/spec/core_payload_types/dev_repositories_spec.rb

@@ -0,0 +1,64 @@
+#
+# 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'))
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', 'lib', 'right_agent', 'core_payload_types'))
+
+module RightScale
+  describe DevRepositories do
+    before(:each) do
+      @expected_repo1 = {"111" => {:repo => {:url => "bunk"}, :positions => [1,2,3]}}
+      @expected_repo2 = {"222" => {:repo => {:url => "err"},  :positions => [4,5,6]}}
+    end
+
+    context 'empty?' do
+      it 'should be empty by default' do
+        DevRepositories.new.empty?.should be_true
+      end
+
+      it 'should not be empty when initialized with data' do
+        DevRepositories.new(@expected_repo1).empty?.should be_false
+      end
+
+      it 'should should not be empty when a repo is added' do
+        dev_repos = DevRepositories.new
+        dev_repos.add_repo("111", {:url=>"bunk"}, [1,2,3])
+        dev_repos.empty?.should be_false
+      end
+    end
+
+    context 'add_repo' do
+      it 'when initially empty, should add without error' do
+        dev_repos = DevRepositories.new
+        dev_repos.add_repo("111", {:url=>"bunk"}, [1,2,3])
+        dev_repos.serialized_members.first.should == @expected_repo1
+      end
+
+      it 'when initialized with data, should add without error' do
+        dev_repos = DevRepositories.new(@expected_repo1)
+        dev_repos.add_repo("222", {:url=>"err"}, [4,5,6])
+        dev_repos.serialized_members.first.should == @expected_repo1.merge(@expected_repo2)
+      end
+    end
+  end
+end

data/spec/core_payload_types/executable_bundle_spec.rb

@@ -0,0 +1,59 @@
+#
+# 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'))
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', 'lib', 'right_agent', 'core_payload_types'))
+
+module RightScale
+  describe ExecutableBundle do
+    context 'serialized_members' do
+      it 'contains 8 elements' do
+        ExecutableBundle.new.serialized_members.count.should == 8
+      end
+    end
+    context 'dev_cookbooks' do
+      context 'when set' do
+        let(:expected_dev_cookbooks) { [{ :foo=>'bar' }] }
+        let(:bundle) { ExecutableBundle.new(nil, nil, nil, nil, nil, nil, expected_dev_cookbooks.clone) }
+
+        it 'should serialize in as the seventh parameter' do
+          bundle.dev_cookbooks.should == expected_dev_cookbooks
+        end
+
+        it 'should serialize out as the seventh parameter' do
+          bundle.serialized_members[6].should == expected_dev_cookbooks
+        end
+      end
+      context 'when not set' do
+        let(:bundle) { ExecutableBundle.new }
+        it 'should be nil' do
+          bundle.dev_cookbooks.should be_nil
+        end
+
+        it "should serialize to nil" do
+          bundle.serialized_members[6].should be_nil
+        end
+      end
+    end
+  end
+end