tengine_core 0.5.28

data/spec/tengine/core/dsls/uc70_driver_enabled_on_activation_spec.rb

@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+require 'spec_helper'
+
+describe "uc70_driver_enabled_on_activation" do
+  before do
+    Tengine::Core::Driver.delete_all
+    Tengine::Core::Session.delete_all
+    @config = Tengine::Core::Config::Core.new({
+        :tengined => {
+          :load_path => File.expand_path('../../../../examples/uc70_driver_enabled_on_activation.rb', File.dirname(__FILE__)),
+          :skip_enablement => true,
+        },
+      })
+    @bootstrap = Tengine::Core::Bootstrap.new(@config)
+    @kernel = Tengine::Core::Kernel.new(@config)
+  end
+
+  it "普通に登録されて、起動後は有効になっている" do
+    @bootstrap.load_dsl
+    driver = Tengine::Core::Driver.first
+    driver.enabled.should == false
+    driver.enabled_on_activation.should == true
+    @bootstrap.enable_drivers
+    driver.reload
+    driver.enabled.should == true
+    driver.enabled_on_activation.should == true
+    @kernel.bind
+    #
+    klass = driver.target_class_name.constantize
+    obj = klass.new
+    klass.should_receive(:new).and_return(obj)
+    obj.should_receive(:puts).with("handler70")
+    mock_headers = mock(:headers)
+    mock_headers.should_receive(:ack)
+    raw_event = Tengine::Event.new(:event_type_name => "event70")
+    @kernel.process_message(mock_headers, raw_event.to_json)
+  end
+
+end

data/spec/tengine/core/dsls/uc71_driver_disabled_on_activation_spec.rb

@@ -0,0 +1,36 @@
+# -*- coding: utf-8 -*-
+require 'spec_helper'
+
+describe "uc71_driver_disabled_on_activation" do
+  before do
+    Tengine::Core::Driver.delete_all
+    Tengine::Core::Session.delete_all
+    @config = Tengine::Core::Config::Core.new({
+        :tengined => {
+          :load_path => File.expand_path('../../../../examples/uc71_driver_disabled_on_activation.rb', File.dirname(__FILE__)),
+          :skip_enablement => true,
+        },
+      })
+    @bootstrap = Tengine::Core::Bootstrap.new(@config)
+    @kernel = Tengine::Core::Kernel.new(@config)
+  end
+
+  it "普通に登録されて、起動後は有効になっている" do
+    @bootstrap.load_dsl
+    driver = Tengine::Core::Driver.first
+    driver.enabled.should == false
+    driver.enabled_on_activation.should == false
+    @bootstrap.enable_drivers
+    driver.reload
+    driver.enabled.should == false
+    driver.enabled_on_activation.should == false
+    @kernel.bind
+    #
+    @kernel.context.should_not_receive(:puts).with("handler71")
+    mock_headers = mock(:headers)
+    mock_headers.should_receive(:ack)
+    raw_event = Tengine::Event.new(:event_type_name => "event71")
+    @kernel.process_message(mock_headers, raw_event.to_json)
+  end
+
+end

data/spec/tengine/core/dsls/uc72_setup_eventmachine_spec.rb

@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+require 'spec_helper'
+
+describe "uc72_setup_eventmachine_spec" do
+  before do
+    Tengine::Core::Driver.delete_all
+    Tengine::Core::Session.delete_all
+    @dsl_path = File.expand_path('../../../../examples/uc72_setup_eventmachine.rb', File.dirname(__FILE__))
+    config = Tengine::Core::Config::Core.new({
+        :tengined => {
+          :load_path => @dsl_path,
+        },
+      })
+    @kernel = Tengine::Core::Kernel.new(config)
+    @bootstrap = Tengine::Core::Bootstrap.new(config)
+    @bootstrap.kernel = @kernel
+  end
+
+  it "EM.run実行時にsetup_eventmachineに渡されたブロックが実行されます" do
+    @kernel.em_setup_blocks.length.should == 0
+    expect{
+      @bootstrap.load_dsl
+    }.to change(@kernel.em_setup_blocks, :length).by(1)
+    @kernel.em_setup_blocks.length.should == 1
+    expect{
+      @kernel.bind
+    }.to_not change(@kernel.em_setup_blocks, :length)
+    @kernel.em_setup_blocks.length.should == 1
+    EM.should_receive(:run).and_yield
+    EM.stub(:defer) # #enable_heartbeat
+    mq = mock(:mq, :queue => nil)
+    @kernel.stub(:mq).at_least(1).times.and_return(mq)
+    @kernel.should_receive(:setup_mq_connection)
+    @kernel.should_receive(:subscribe_queue).and_yield
+    @kernel.context.should_receive(:puts).with("setup_eventmachine")
+    EM.should_receive(:add_periodic_timer).with(3)
+    @kernel.activate
+  end
+end

data/spec/tengine/core/dsls/uc80_raise_io_error_spec.rb

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+require 'spec_helper'
+
+require 'stringio'
+
+describe "uc80_raise_io_error" do
+  before(:all) do
+    @logger = Tengine.logger
+  end
+  after(:all) do
+    Tengine.logger = @logger
+  end
+
+  before do
+    Tengine::Core::Driver.delete_all
+    Tengine::Core::Session.delete_all
+    @dsl_path = File.expand_path('../../../../examples/uc80_raise_io_error.rb', File.dirname(__FILE__))
+    @config = Tengine::Core::Config::Core.new({
+        :tengined => {
+          :load_path => @dsl_path,
+        },
+      })
+    @bootstrap = Tengine::Core::Bootstrap.new(@config)
+    @kernel = Tengine::Core::Kernel.new(@config)
+  end
+
+  it "例外がraiseされると、その例外の内容がログに出力され、イベント処理エラーイベントをfireする" do
+    @bootstrap.load_dsl
+    @kernel.bind
+    mock_headers = mock(:headers)
+    mock_headers.should_receive(:ack)
+    raw_event = Tengine::Event.new(:event_type_name => "event80")
+    @buffer = StringIO.new
+    Tengine.logger = Logger.new(@buffer)
+    Tengine.logger.level = Logger::ERROR
+    @kernel.should_receive(:fire).with("event80.error.tengined",
+      :properties => {
+        :original_event => instance_of(String),
+        :error_class_name => "IOError",
+        :error_message => "by driver80",
+        :error_backtrace => instance_of(Array),
+        # :block_source_location => "#{@dsl_path}:6" # 6はブロックの行番号
+      })
+    Tengine::Core::Kernel.temp_exception_reporter(:except_test) do
+      expect{
+        @kernel.process_message(mock_headers, raw_event.to_json)
+      }.to_not raise_error
+    end
+    @buffer.rewind
+    @buffer.string.should =~ /\[IOError\] by driver80/
+  end
+
+end

data/spec/tengine/core/dsls/uc81_raise_runtime_error_spec.rb

@@ -0,0 +1,49 @@
+# -*- coding: utf-8 -*-
+require 'spec_helper'
+
+describe "uc81_raise_runtime_error" do
+  before do 
+    Tengine::Core::HandlerPath.delete_all
+    Tengine::Core::Driver.delete_all
+    Tengine::Core::Session.delete_all
+    @dsl_path = File.expand_path('../../../../examples/uc81_raise_runtime_error.rb', File.dirname(__FILE__))
+    @config = Tengine::Core::Config::Core.new({
+        :tengined => {
+          :load_path => @dsl_path,
+        },
+      })
+    @bootstrap = Tengine::Core::Bootstrap.new(@config)
+    @kernel = Tengine::Core::Kernel.new(@config)
+  end
+
+  it "例外がraiseされると、イベント処理エラーイベントをfireする" do
+    @bootstrap.load_dsl
+    @kernel.bind
+
+    driver = Tengine::Core::Driver.first
+    driver.enabled.should == true
+    driver.version.should == @config.dsl_version
+    path = Tengine::Core::HandlerPath.first
+    path.event_type_name.should == "event81"
+    path.driver_id.should == driver.id
+    Tengine::Core::HandlerPath.default_driver_version.should == @config.dsl_version
+
+    mock_headers = mock(:headers)
+    mock_headers.should_receive(:ack)
+    raw_event = Tengine::Event.new(:event_type_name => "event81")
+    @kernel.should_receive(:fire).with("event81.error.tengined",
+      :properties => {
+        :original_event => instance_of(String),
+        :error_class_name => "RuntimeError",
+        :error_message => "by driver81",
+        :error_backtrace => instance_of(Array),
+        # :block_source_location => "#{@dsl_path}:6" # 6はブロックの行番号
+      })
+    Tengine::Core::Kernel.temp_exception_reporter(:except_test) do
+      expect{
+        @kernel.process_message(mock_headers, raw_event.to_json)
+      }.to_not raise_error
+    end
+  end
+
+end

data/spec/tengine/core/event/finder_spec.rb

@@ -0,0 +1,136 @@
+# -*- coding: utf-8 -*-
+require 'spec_helper'
+
+describe Tengine::Core::Event::Finder do
+
+  context "検索" do
+    before(:all) do
+      Tengine::Core::Event.delete_all
+      occured_at_base = Time.now
+      base = {
+        :sender_name => "agent:server1/99326/tengine_job_agent",
+        :level => 2,
+      }
+      uuid_gen = Tengine::Event.uuid_gen
+      [
+        {:event_type_name => "start.execution.job.tengine", :source_name => "execution:localhost/123/1111/1111"}.update(base),
+        {:event_type_name => "start.jobnet.job.tengine", :source_name => "job:localhost/123/2222/3333"}.update(base),
+        {:event_type_name => "start.job.job.tengine", :source_name => "job:localhost/123/2222/4444"}.update(base),
+        {:event_type_name => "finished.process.job.tengine", :source_name => "job:localhost/123/2222/4444"}.update(base),
+        {:event_type_name => "success.job.job.tengine", :source_name => "job:localhost/123/2222/4444"}.update(base),
+        {:event_type_name => "success.jobnet.job.tengine", :source_name => "job:localhost/123/2222/3333"}.update(base),
+        {:event_type_name => "success.execution.job.tengine", :source_name => "execution:localhost/123/1111/1111"}.update(base),
+      ].each_with_index do |attrs, i|
+        Tengine::Core::Event.create!(attrs.update(:key => uuid_gen.generate, :occurred_at => occured_at_base + (i*10)))
+      end
+    end
+
+
+    context "event_type_name" do
+      it ".jobnet.job.tengine" do
+        f = Tengine::Core::Event::Finder.new(:event_type_name => "/.jobnet.job.tengine$/")
+        result = f.paginate
+        result.map{|h| h[:event_type_name]}.should == [
+          "success.jobnet.job.tengine",
+          "start.jobnet.job.tengine",
+        ]
+      end
+
+      it "start.job." do
+        f = Tengine::Core::Event::Finder.new(:event_type_name => "start.job.")
+        result = f.paginate
+        result.map{|h| h[:event_type_name]}.should == [
+          "start.job.job.tengine",
+        ]
+      end
+
+      it "/start.job./" do
+        f = Tengine::Core::Event::Finder.new(:event_type_name => "/start.job./")
+        result = f.paginate
+        result.map{|h| h[:event_type_name]}.should == [
+          "start.job.job.tengine",
+          "start.jobnet.job.tengine",
+        ]
+      end
+    end
+
+    context "source_name" do
+      it "execution" do
+        f = Tengine::Core::Event::Finder.new(:source_name => "execution:localhost/123")
+        result = f.paginate
+        result.map{|h| h[:event_type_name]}.should == [
+          "success.execution.job.tengine",
+          "start.execution.job.tengine",
+        ]
+      end
+
+      it "job" do
+        f = Tengine::Core::Event::Finder.new(:source_name => "job:localhost/123/2222")
+        result = f.paginate
+        result.map{|h| h[:event_type_name]}.should == [
+          "success.jobnet.job.tengine",
+          "success.job.job.tengine",
+          "finished.process.job.tengine",
+          "start.job.job.tengine",
+          "start.jobnet.job.tengine",
+        ]
+      end
+    end
+
+    context "sender_name" do
+      it "string" do
+        f = Tengine::Core::Event::Finder.new(:sender_name => "server1")
+        result = f.paginate
+        result.map{|h| h[:event_type_name]}.should == []
+      end
+
+      it "regexp" do
+        f = Tengine::Core::Event::Finder.new(:sender_name => "/server1/")
+        result = f.paginate
+        result.map{|h| h[:event_type_name]}.should == [
+          "success.execution.job.tengine",
+          "success.jobnet.job.tengine",
+          "success.job.job.tengine",
+          "finished.process.job.tengine",
+          "start.job.job.tengine",
+          "start.jobnet.job.tengine",
+          "start.execution.job.tengine",
+        ]
+      end
+    end
+
+    context "scope" do
+      it "ソート順は発生時刻の降順であること" do
+        f = Tengine::Core::Event::Finder.new
+        result = f.paginate
+        result.map{|h| h[:event_type_name]}.should == [
+          "success.execution.job.tengine",
+          "success.jobnet.job.tengine",
+          "success.job.job.tengine",
+          "finished.process.job.tengine",
+          "start.job.job.tengine",
+          "start.jobnet.job.tengine",
+          "start.execution.job.tengine",
+        ]
+
+        occurred_at_base = Time.now
+        result.each_with_index do |event, i|
+          event.occurred_at = occurred_at_base + i
+          event.save!
+        end
+
+        f = Tengine::Core::Event::Finder.new
+        result = f.paginate
+        result.map{|h| h[:event_type_name]}.should == [
+          "start.execution.job.tengine",
+          "start.jobnet.job.tengine",
+          "start.job.job.tengine",
+          "finished.process.job.tengine",
+          "success.job.job.tengine",
+          "success.jobnet.job.tengine",
+          "success.execution.job.tengine",
+        ]
+      end
+    end
+  end
+end

data/spec/tengine/core/event_exception_reportable_spec.rb

@@ -0,0 +1,33 @@
+# -*- coding: utf-8 -*-
+require 'spec_helper'
+
+require 'tengine/event'
+
+describe Tengine::Core::EventExceptionReportable do
+
+  describe :to_reporter do
+    context "Symbolで指定" do
+      Tengine::Core::EventExceptionReportable::EVENT_EXCEPTION_REPORTERS.keys.each do |reporter_name|
+        it reporter_name do
+          Tengine::Core::EventExceptionReportable.to_reporter(reporter_name).should_not == nil
+        end
+      end
+      it "存在しないreporter名" do
+        expect{
+          Tengine::Core::EventExceptionReportable.to_reporter(:unexist_reporter)
+        }.to raise_error(NameError, "Unknown reporter: :unexist_reporter")
+      end
+    end
+
+    it "Procを指定" do
+      proc1 = Proc.new{  puts "foo" }
+      Tengine::Core::EventExceptionReportable.to_reporter(proc1).should == proc1
+    end
+
+    it "不正なreporterを指定" do
+      expect{
+        Tengine::Core::EventExceptionReportable.to_reporter(100)
+      }.to raise_error(ArgumentError, "Invalid reporter: 100")
+    end
+  end
+end

data/spec/tengine/core/event_spec.rb

@@ -0,0 +1,161 @@
+# -*- coding: utf-8 -*-
+require 'spec_helper'
+
+describe Tengine::Core::Event do
+
+  valid_attributes1 = {
+    :event_type_name => "valid.event-type_name1",
+    :key => "some_unique_key1",
+  }.freeze
+
+  context "event_type_nameとkeyは必須" do
+    it "正常系" do
+      Tengine::Core::Event.delete_all
+      driver1 = Tengine::Core::Event.new(valid_attributes1)
+      driver1.valid?.should == true
+    end
+
+    [:event_type_name, :key].each do |key|
+      it "#{key}なし" do
+        attrs = valid_attributes1.dup
+        attrs.delete(key)
+        driver1 = Tengine::Core::Event.new(attrs)
+        driver1.valid?.should == false
+      end
+    end
+  end
+
+  context "keyはバージョン毎にユニーク" do
+    before do
+      Tengine::Core::Event.delete_all
+      Tengine::Core::Event.create!(valid_attributes1)
+    end
+
+    it "同じ名前で登録されているものが存在する場合エラー、しかしunique indexによるエラーが発生します" do
+      begin
+        Tengine::Core::Event.create!(valid_attributes1)
+        fail
+      rescue Mongo::OperationFailure => e
+        e.message.should =~ /E11000/
+        e.message.should =~ /duplicate key error/
+        e.message.should =~ /tengine_core_events/
+        e.message.should =~ /some_unique_key1/
+      end
+    end
+  end
+
+  context "must be unique with key and sender_name" do
+    it "raise an exception when violate unique consistent" do
+      Mongoid.persist_in_safe_mode.should == true
+
+      unique_key_name = "key1"
+      Tengine::Core::Event.delete_all
+      Tengine::Core::Event.create!(:event_type_name => "foo", :key => unique_key_name, :sender_name => "server1")
+      expect {
+        Tengine::Core::Event.create!(:event_type_name => "foo", :key => unique_key_name, :sender_name => "server2")
+      }.to raise_error # (Mongo::OperationFailure, /duplicate key error/)
+    end
+  end
+
+  context "event_type_name" do
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "foo").should be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "Foo").should be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "F00").should be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "foo.bar.baz").should be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "foo."*1024+"bar").should be_valid }
+
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "\0").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => " ").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "　").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "\t").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "\n").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => ";").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => ",").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "#").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "-->").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => """).should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "javascript:alert('hello')").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "javascript:alert(String.fromCharCode(88,83,83))").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "<script>alert('hello')</script>").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => '<img src="javascript:alert(\'hello\')">').should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => " foo").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "foo ").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => " foo ").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "'foo'").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => '"foo"').should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "foo bar baz").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "(foo(bar(baz)))").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "ふぅ").should_not be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "נד בייקר").should_not be_valid } # hebrew
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => ".error.tengined").should_not be_valid }
+
+    # 実在のイベント種別名
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "start.job.job.tengine").should be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "job.heartbeat.tengine").should be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "alert.execution.job.tengine").should be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "finished.process.job.tengine.failed.tengined").should be_valid }
+    it { Tengine::Core::Event.new(:key => "k", :event_type_name => "Tengine::Resource::VirtualServer.created.tengine_resource_watchd").should be_valid }
+  end
+
+
+  describe :level do
+
+    it "i18n_scopeが設定される" do
+      Tengine::Core::Event.level_enum.i18n_scope.should == ['selectable_attrs', 'tengine/core/event', 'level']
+    end
+
+    context :i18n_scope do
+      before(:all) do
+        @default_locale_backup = I18n.default_locale
+        @locale_backup = I18n.locale
+      end
+
+      after(:all) do
+        I18n.locale = @locale_backup
+        I18n.default_locale = @default_locale_backup
+      end
+
+      before do
+        I18n.backend = I18n::Backend::Simple.new
+        I18n.backend.store_translations 'en', 'selectable_attrs' => {'tengine/core/event' => {'level' => {
+            'debug' => 'DEBUG',
+            'info'  => 'INFO',
+            'warn'  => 'WARN',
+            'error' => 'ERROR',
+            'fatal' => 'FATAL',
+            } } }
+        I18n.backend.store_translations 'ja', 'selectable_attrs' => {'tengine/core/event' => {'level' => {
+            'debug' => 'デバッグ',
+            'info'  => '情報',
+            'warn'  => '警告',
+            'error' => 'エラー',
+            'fatal' => '致命的なエラー',
+            } } }
+      end
+
+      context "#level_name" do
+        {
+          :debug => {:en => "DEBUG", :ja => "デバッグ"},
+          :info  => {:en => "INFO" , :ja => "情報"},
+          :warn  => {:en => "WARN" , :ja => "警告"},
+          :error => {:en => "ERROR", :ja => "エラー"},
+          :fatal => {:en => "FATAL", :ja => "致命的なエラー"},
+        }.each do |level_key, hash|
+          context level_key.inspect do
+            subject{ Tengine::Core::Event.new(:level_key => level_key)}
+            hash.each do |locale, level_name|
+              it do
+                I18n.locale = locale.to_s
+                subject.level_name.should == level_name
+              end
+            end
+          end
+
+        end
+      end
+
+    end
+  end
+
+end