right_agent 2.0.7-x86-mingw32

data/spec/enrollment_result_spec.rb

@@ -0,0 +1,53 @@
+# Copyright (c) 2009-2011 RightScale, Inc, All Rights Reserved Worldwide.
+#
+# THIS PROGRAM IS CONFIDENTIAL AND PROPRIETARY TO RIGHTSCALE
+# AND CONSTITUTES A VALUABLE TRADE SECRET.  Any unauthorized use,
+# reproduction, modification, or disclosure of this program is
+# strictly prohibited.  Any use of this program by an authorized
+# licensee is strictly subject to the terms and conditions,
+# including confidentiality obligations, set forth in the applicable
+# License Agreement between RightScale.com, Inc. and
+# the licensee.
+
+require File.expand_path(File.join(File.dirname(__FILE__), 'spec_helper'))
+require File.expand_path(File.join(File.dirname(__FILE__), '..' , 'lib', 'right_agent', 'enrollment_result'))
+
+describe RightScale::EnrollmentResult do
+  before(:each) do
+    @key = 'topsecret'
+    @result = RightScale::EnrollmentResult.new(6, Time.now, 'router cert', 'my cert', 'my private key', @key)
+    @message  = RightScale::EnrollmentResult.dump(@result)
+  end
+
+  it 'should serialize and unserialize correctly' do
+    r2 = RightScale::EnrollmentResult.load(@message, @key)
+    @result.should == r2
+  end
+
+  context "supporting different versions" do
+    RightScale::EnrollmentResult::SUPPORTED_VERSIONS.each do |v|
+      it "should support version #{v}" do
+        @result = RightScale::EnrollmentResult.new(v, Time.now, 'router cert', 'my cert', 'my private key', @key)
+        serialized = RightScale::EnrollmentResult.dump(@result)
+        @result2 = RightScale::EnrollmentResult.load(serialized, @key)
+        @result.should == @result2
+      end
+    end
+  end
+
+  it 'should fail to decrypt if tampered with' do
+    #Simulate some ciphertext tampering.
+    @message.gsub! /[0-9]/, '1'
+    @message.gsub! /"r_s_version":"[0-9]+"/, '"r_s_version":"6"'
+
+    lambda do
+      RightScale::EnrollmentResult.load(@message, @key)
+    end.should raise_error(RightScale::EnrollmentResult::IntegrityFailure)
+  end
+
+  it 'should fail to decrypt if the key is wrong' do
+    lambda do
+      RightScale::EnrollmentResult.load(@message, @key + "evil")
+    end.should raise_error(RightScale::EnrollmentResult::IntegrityFailure)
+  end
+end

data/spec/history_spec.rb

@@ -0,0 +1,246 @@
+#
+# 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 'tmpdir'
+
+describe RightScale::History do
+
+  before(:each) do
+    @identity = "rs-agent-1-1"
+    @pid = Process.pid
+    now = Time.at(1000000)
+    flexmock(Time).should_receive(:now).and_return { now += 10 }
+    FileUtils.mkdir_p(@test_dir = File.join(RightScale::Platform.filesystem.temp_dir, 'history_test'))
+    RightScale::AgentConfig.pid_dir = @test_dir
+    @history = RightScale::History.new(@identity)
+  end
+
+  after(:each) do
+    FileUtils.remove_dir(@test_dir)
+  end
+
+  describe "update" do
+
+    it "should store event in history file" do
+      @history.update("some event").should be_true
+      @history.load.should == [{"time" => 1000010, "pid" => @pid, "event" => "some event"}]
+    end
+
+    it "should store event in history file following previous event" do
+      @history.update("some event").should be_true
+      @history.update("another event").should be_true
+      @history.load.should == [{"time" => 1000010, "pid" => @pid, "event" => "some event"},
+                               {"time" => 1000020, "pid" => @pid, "event" => "another event"}]
+    end
+
+  end
+
+  describe "load" do
+
+    it "should load no events if there is no history" do
+      @history.load.should == []
+    end
+
+    it "should load events from history file" do
+      @history.update("some event").should be_true
+      @history.load.should == [{"time" => 1000010, "pid" => @pid, "event" => "some event"}]
+    end
+
+  end
+
+  describe "analyze_service" do
+
+    it "should indicate no uptime if history empty" do
+      @history.analyze_service.should == {:uptime => 0, :total_uptime => 0}
+    end
+
+    it "should indicate no uptime if not yet running" do
+      @history.update("start")
+      @history.analyze_service.should == {:uptime => 0, :total_uptime => 0}
+    end
+
+    it "should measure uptime starting from last run time" do
+      @history.update("start")
+      @history.update("run")
+      @history.analyze_service.should == {:uptime => 10, :total_uptime => 10}
+    end
+
+    it "should count restarts" do
+      @history.update("start")
+      @history.update("stop")
+      @history.update("start")
+      @history.analyze_service[:restarts].should == 1
+    end
+
+    it "should not count initial start as a restart" do
+      @history.update("start")
+      @history.update("run")
+      @history.analyze_service[:restarts].should be_nil
+    end
+
+    it "should ignore repeated stops when counting restarts" do
+      @history.update("start")
+      @history.update("stop")
+      @history.update("stop")
+      @history.update("start")
+      @history.analyze_service[:restarts].should == 1
+    end
+
+    it "should record number of graceful exits if there are restarts" do
+      @history.update("start")
+      @history.update("run")
+      @history.update("stop")
+      @history.update("graceful exit")
+      @history.update("start")
+      @history.update("stop")
+      @history.update("start")
+      @history.update("run")
+      @history.update("stop")
+      @history.update("graceful exit")
+      @history.update("start")
+      @history.update("run")
+      @history.analyze_service[:graceful_exits].should == 2
+      @history.analyze_service[:restarts].should == 3
+    end
+
+    it "should measure total uptime across restarts and include current uptime" do
+      @history.update("start")
+      @history.update("run")
+      @history.update("stop")
+      @history.update("graceful exit")
+      @history.update("start")
+      @history.update("stop")
+      @history.update("start")
+      @history.update("run")
+      @history.update("stop")
+      @history.update("graceful exit")
+      @history.update("start")
+      @history.update("run")
+      @history.analyze_service.should == {:uptime => 10, :total_uptime => 30, :restarts => 3, :graceful_exits => 2}
+    end
+
+    it "should count crashes" do
+      @history.update("start")
+      @history.update("start")
+      @history.analyze_service[:crashes].should == 1
+      @history.analyze_service[:last_crash_time].should == 1000020
+      @history.analyze_service[:crashed_last].should be_true
+    end
+
+    it "should record last crash age if there are crashes" do
+      @history.update("start")
+      @history.update("start")
+      @history.analyze_service[:crashed_last].should be_true
+    end
+
+    it "should determine whether crashed last time was started" do
+      @history.update("start")
+      @history.update("start")
+      @history.analyze_service[:crashed_last].should be_true
+      @history.update("run")
+      @history.analyze_service[:crashed_last].should be_true
+      @history.update("stop")
+      @history.analyze_service[:crashed_last].should be_false
+      @history.update("start")
+      @history.analyze_service[:crashed_last].should be_false
+      @history.update("stop")
+      @history.update("start")
+      @history.analyze_service[:crashed_last].should be_false
+      @history.update("run")
+      @history.update("stop")
+      @history.update("graceful exit")
+      @history.analyze_service[:crashed_last].should be_false
+      @history.update("start")
+      @history.analyze_service[:crashed_last].should be_false
+      @history.update("start")
+      @history.analyze_service[:crashed_last].should be_true
+    end
+
+    it "should count restarts and crashes" do
+      @history.update("start")
+      @history.update("run")
+      @history.update("stop")
+      @history.update("graceful exit")
+      @history.update("start")
+      @history.update("start")
+      @history.update("stop")
+      @history.update("start")
+      @history.update("run")
+      @history.update("stop")
+      @history.update("graceful exit")
+      @history.update("start")
+      @history.update("run")
+      @history.update("start")
+      @history.update("run")
+      @history.analyze_service[:crashes].should == 2
+      @history.analyze_service[:restarts].should == 3
+      @history.analyze_service[:last_crash_time].should == 1000140
+      @history.analyze_service[:crashed_last].should be_true
+    end
+
+    it "should ignore unrecognized events" do
+      @history.update("other event")
+      @history.update("start")
+      @history.update("other event")
+      @history.update("stop")
+      @history.update("start")
+      @history.update("other event")
+      @history.analyze_service.should == {:uptime => 0, :total_uptime => 0, :restarts => 1, :graceful_exits => 0}
+    end
+
+    it "should not re-analyze if there are no new events" do
+      @history.update("start")
+      @history.update("run")
+      @history.analyze_service.should == {:uptime => 10, :total_uptime => 10}
+      flexmock(@history).should_receive(:load).never
+      @history.analyze_service
+    end
+
+    it "should update uptime and total_uptime even if do not do re-analyze" do
+      @history.update("start")
+      @history.update("run")
+      @history.update("start")
+      @history.update("run")
+      @history.update("stop")
+      @history.update("graceful exit")
+      @history.update("start")
+      @history.update("run")
+      @history.analyze_service.should == {:uptime => 10, :total_uptime => 30, :restarts => 1, :graceful_exits => 1,
+                                          :crashes => 1, :last_crash_time => 1000030, :crashed_last => false}
+      @history.analyze_service.should == {:uptime => 20, :total_uptime => 40, :restarts => 1, :graceful_exits => 1,
+                                          :crashes => 1, :last_crash_time => 1000030, :crashed_last => false}
+    end
+
+    it "should re-analyze if there was a new event since last analysis" do
+      @history.update("start")
+      @history.update("run")
+      @history.analyze_service.should == {:uptime => 10, :total_uptime => 10}
+      @history.instance_variable_set(:@pid, -1) # Simulate new process id following crash
+      @history.update("start")
+      @history.analyze_service.should == {:uptime => 0, :total_uptime => 20, :crashes => 1, :last_crash_time => 1000040,
+                                          :crashed_last => true}
+    end
+
+  end
+
+end

data/spec/log_spec.rb

@@ -0,0 +1,192 @@
+#
+# 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::Log do
+
+  # Count number of lines logged with the specified text
+  # Search is case sensitive and regular expression sensitive
+  #
+  # === Parameters
+  # text(String):: Text to search for
+  #
+  # === Return
+  # (Integer):: Number of lines where text found
+  def log_count(text)
+    # no such animal as egrep in windows, so hacked up an equivalent here.
+    pattern = Regexp.compile(text)
+    File.read(@log_file).each_line.select { |x| x =~ pattern }.count
+  end
+
+  before(:all) do
+    ENV['RS_LOG'] = 'true'
+
+    class RightScale::Platform::Filesystem
+      alias original_log_dir log_dir
+      def log_dir
+        # For specs, write all logs to temp_dir
+        # since log_dir may not be writable
+        temp_dir
+      end
+    end
+  end
+
+  after(:all) do
+    class RightScale::Platform::Filesystem
+      alias log_dir original_log_dir
+    end
+  end
+
+  before(:each) do
+    Singleton.__init__(RightScale::Log)
+  end
+
+  after(:all) do
+    Singleton.__init__(RightScale::Log)
+    ENV['RS_LOG'] = nil
+  end
+
+  it 'should set default level to info' do
+    RightScale::Log.level.should == :info
+  end
+
+  it 'should change level to debug when force_debug' do
+    RightScale::Log.level.should == :info
+    RightScale::Log.force_debug
+    RightScale::Log.level.should == :debug
+  end
+
+  it 'should notify of log level change' do
+    RightScale::Log.level.should == :info
+    notified = nil
+    RightScale::Log.notify(lambda{ |l| notified = l })
+    RightScale::Log.level = :debug
+    notified.should == Logger::DEBUG
+  end
+
+  context "logging" do
+
+    before(:each) do
+      # note that log directory doesn't necessarily exist in Windows dev/test
+      # environment.
+      log_dir = RightScale::Platform.filesystem.log_dir
+      FileUtils.mkdir_p(log_dir) unless File.directory?(log_dir)
+
+      # use a unique name for the test because the file cannot be deleted after
+      # the test in the Windows case (until the process exits) and we need to
+      # avoid any potential conflicts with other tests, etc.
+      @log_name = "log_test-9D9A9CB7-24A1-4093-9F75-D462D373A0D8"
+      @log_file = File.join(log_dir, "#{@log_name}.log")
+      RightScale::Log.program_name = "tester"
+      RightScale::Log.log_to_file_only(true)
+      RightScale::Log.init(@log_name, log_dir)
+    end
+
+    after(:each) do
+      # note that the log is held open in the Windows case by Log so we
+      # cannot delete it after each test. on the other hand, we can truncate the
+      # log to zero size and continue to the next test (ultimately leaving an
+      # empty file after the test, which is acceptable).
+      File.delete(@log_file) if File.file?(@log_file) rescue File.truncate(@log_file, 0)
+    end
+
+    it 'should log info but not debug by default' do
+      RightScale::Log.debug("Test debug")
+      RightScale::Log.info("Test info")
+      log_count("Test debug$").should == 0
+      log_count("Test info$").should == 1
+    end
+
+    it 'should log debug after adjust level to :debug' do
+      RightScale::Log.level = :debug
+      RightScale::Log.debug("Test debug")
+      RightScale::Log.info("Test info")
+      log_count("Test debug$").should == 1
+      log_count("Test info$").should == 1
+    end
+
+    it 'should log additional error string when given' do
+      RightScale::Log.warning("Test warning", "Error")
+      RightScale::Log.error("Test error", "Error")
+      log_count("Test warning \\\(Error\\\)$").should == 1
+      log_count("Test error \\\(Error\\\)$").should == 1
+    end
+
+    it 'should log without exception when none given' do
+      begin
+        nil + "string"
+      rescue Exception => e
+        RightScale::Log.warning("Test warning")
+        RightScale::Log.error("Test error")
+      end
+      log_count("Test warning$").should == 1
+      log_count("Test error$").should == 1
+    end
+
+    it 'should log with exception class, message, and caller appended by default when exception given' do
+      begin
+        nil + "string"
+      rescue Exception => e
+        RightScale::Log.warning("Test warning", e)
+        RightScale::Log.error("Test error", e)
+      end
+      log_count("Test warning \\\(NoMethodError: undefined method \\\`\\\+\' for nil:NilClass in .*log_spec.*\\\)$").should == 1
+      log_count("Test error \\\(NoMethodError: undefined method \\\`\\\+\' for nil:NilClass in .*log_spec.*\\\)$").should == 1
+    end
+
+    it 'should log with exception class, message, and caller appended when use :caller' do
+      begin
+        nil + "string"
+      rescue Exception => e
+        RightScale::Log.warning("Test warning", e, :caller)
+        RightScale::Log.error("Test error", e, :caller)
+      end
+      log_count("Test warning \\\(NoMethodError: undefined method \\\`\\\+\' for nil:NilClass in .*log_spec.*\\\)$").should == 1
+      log_count("Test error \\\(NoMethodError: undefined method \\\`\\\+\' for nil:NilClass in .*log_spec.*\\\)$").should == 1
+    end
+
+    it 'should log with exception class and message appended when use :no_trace' do
+      begin
+        nil + "string"
+      rescue Exception => e
+        RightScale::Log.warning("Test warning", e, :no_trace)
+        RightScale::Log.error("Test error", e, :no_trace)
+      end
+      log_count("Test warning \\\(NoMethodError: undefined method \\\`\\\+\' for nil:NilClass\\\)$").should == 1
+      log_count("Test error \\\(NoMethodError: undefined method \\\`\\\+\' for nil:NilClass\\\)$").should == 1
+    end
+
+    it 'should log with exception class, message, and full backtrace appended when use :trace' do
+      begin
+        nil + "string"
+      rescue Exception => e
+        RightScale::Log.warning("Test warning", e, :trace)
+        RightScale::Log.error("Test error", e, :trace)
+      end
+      log_count("Test warning \\\(NoMethodError: undefined method \\\`\\\+\' for nil:NilClass in$").should == 1
+      log_count("Test error \\\(NoMethodError: undefined method \\\`\\\+\' for nil:NilClass in$").should == 1
+    end
+
+  end
+
+end