tengine_core 0.5.28
Sign up to get free protection for your applications and to get access to all the features.
- data/.document +5 -0
- data/.rspec +1 -0
- data/Gemfile +40 -0
- data/Gemfile.lock +95 -0
- data/README.md +54 -0
- data/Rakefile +44 -0
- data/VERSION +1 -0
- data/bin/tengine_atd +8 -0
- data/bin/tengine_heartbeat_watchd +8 -0
- data/bin/tengined +182 -0
- data/examples/VERSION +1 -0
- data/examples/uc01_execute_processing_for_event.rb +11 -0
- data/examples/uc02_fire_another_event.rb +16 -0
- data/examples/uc03_2handlers_for_1event.rb +16 -0
- data/examples/uc08_if_both_a_and_b_occurs.rb +11 -0
- data/examples/uc10_if_the_event_occurs_at_the_server.rb +15 -0
- data/examples/uc50_commit_event_at_first.rb +17 -0
- data/examples/uc51_commit_event_at_first_submit.rb +29 -0
- data/examples/uc52_commit_event_after_all_handler_submit.rb +31 -0
- data/examples/uc52_never_commit_event_unless_all_handler_submit.rb +31 -0
- data/examples/uc60_event_in_handler.rb +18 -0
- data/examples/uc62_session_in_driver.rb +16 -0
- data/examples/uc64_safety_countup.rb +14 -0
- data/examples/uc70_driver_enabled_on_activation.rb +13 -0
- data/examples/uc71_driver_disabled_on_activation.rb +14 -0
- data/examples/uc72_setup_eventmachine.rb +17 -0
- data/examples/uc80_raise_io_error.rb +10 -0
- data/examples/uc81_raise_runtime_error.rb +10 -0
- data/examples2/driver01.rb +18 -0
- data/examples2/driver02.rb +19 -0
- data/examples2/uc08_if_both_a_and_b_occurs.rb +13 -0
- data/examples2/uc10_if_the_event_occurs_at_the_server.rb +18 -0
- data/examples2/uc51_commit_event_at_first_submit_1.rb +16 -0
- data/examples2/uc51_commit_event_at_first_submit_2.rb +17 -0
- data/examples2/uc51_commit_event_at_first_submit_3.rb +17 -0
- data/examples2/uc62_session_in_driver.rb +16 -0
- data/examples2/uc71_driver_disabled_on_activation.rb +16 -0
- data/failure_examples/VERSION +1 -0
- data/failure_examples/uc53_submit_outside_of_handler.rb +15 -0
- data/failure_examples/uc61_event_outside_of_handler.rb +12 -0
- data/failure_examples/uc63_session_outside_of_driver.rb +13 -0
- data/lib/tengine/core.rb +74 -0
- data/lib/tengine/core/bootstrap.rb +123 -0
- data/lib/tengine/core/collection_accessible.rb +34 -0
- data/lib/tengine/core/config.rb +10 -0
- data/lib/tengine/core/config/atd.rb +225 -0
- data/lib/tengine/core/config/core.rb +319 -0
- data/lib/tengine/core/config/heartbeat_watcher.rb +229 -0
- data/lib/tengine/core/connection_test/.gitignore +1 -0
- data/lib/tengine/core/connection_test/fire_bar_on_foo.rb +16 -0
- data/lib/tengine/core/driveable.rb +213 -0
- data/lib/tengine/core/driver.rb +69 -0
- data/lib/tengine/core/driver/finder.rb +42 -0
- data/lib/tengine/core/dsl_evaluator.rb +110 -0
- data/lib/tengine/core/dsl_filter_def.rb +11 -0
- data/lib/tengine/core/dsl_loader.rb +108 -0
- data/lib/tengine/core/event.rb +145 -0
- data/lib/tengine/core/event/finder.rb +82 -0
- data/lib/tengine/core/event_exception_reportable.rb +88 -0
- data/lib/tengine/core/event_wrapper.rb +21 -0
- data/lib/tengine/core/find_by_name.rb +31 -0
- data/lib/tengine/core/handler.rb +152 -0
- data/lib/tengine/core/handler_path.rb +33 -0
- data/lib/tengine/core/heartbeat_watcher.rb +161 -0
- data/lib/tengine/core/io_to_logger.rb +22 -0
- data/lib/tengine/core/kernel.rb +510 -0
- data/lib/tengine/core/kernel_runtime.rb +91 -0
- data/lib/tengine/core/method_traceable.rb +38 -0
- data/lib/tengine/core/mongoid_fix.rb +19 -0
- data/lib/tengine/core/mutex.rb +177 -0
- data/lib/tengine/core/optimistic_lock.rb +69 -0
- data/lib/tengine/core/plugins.rb +54 -0
- data/lib/tengine/core/schedule.rb +21 -0
- data/lib/tengine/core/scheduler.rb +156 -0
- data/lib/tengine/core/selectable_attr.rb +29 -0
- data/lib/tengine/core/session.rb +21 -0
- data/lib/tengine/core/session_wrapper.rb +68 -0
- data/lib/tengine/core/setting.rb +21 -0
- data/lib/tengine/core/validation.rb +36 -0
- data/lib/tengine/errors.rb +18 -0
- data/lib/tengine/rspec.rb +8 -0
- data/lib/tengine/rspec/context_wrapper.rb +51 -0
- data/lib/tengine/rspec/extension.rb +53 -0
- data/lib/tengine_core.rb +23 -0
- data/spec/factories/tengine_core_drivers.rb +10 -0
- data/spec/factories/tengine_core_events.rb +14 -0
- data/spec/factories/tengine_core_handler_paths.rb +9 -0
- data/spec/factories/tengine_core_handlers.rb +9 -0
- data/spec/factories/tengine_core_sessions.rb +9 -0
- data/spec/mongoid.yml +35 -0
- data/spec/spec_helper.rb +48 -0
- data/spec/support/mongo_index_key_log.rb +91 -0
- data/spec/tengine/core/bootstrap_spec.rb +278 -0
- data/spec/tengine/core/bugfix/bind_dsl_file_in_multi_byte_dir_spec.rb +21 -0
- data/spec/tengine/core/bugfix/enabled_on_activation_spec.rb +112 -0
- data/spec/tengine/core/bugfix/receive_event_spec.rb +133 -0
- data/spec/tengine/core/bugfix/use_dsl_version_method.rb +12 -0
- data/spec/tengine/core/bugfix/use_dsl_version_method_spec.rb +28 -0
- data/spec/tengine/core/bugfix/use_event_in_handler_dsl.rb +11 -0
- data/spec/tengine/core/bugfix//351/235/236ACSII/343/201/256/343/203/206/343/202/231/343/202/243/343/203/254/343/202/257/343/203/210/343/203/252/345/220/215/source_location_encoding.rb +35 -0
- data/spec/tengine/core/bugfix//351/235/236ACSII/343/201/256/343/203/206/343/202/231/343/202/243/343/203/254/343/202/257/343/203/210/343/203/252/345/220/215//351/235/236ASCII/343/201/256/343/203/225/343/202/241/343/202/244/343/203/253/345/220/215_dsl.rb +38 -0
- data/spec/tengine/core/bugfix//351/235/236ACSII/343/201/256/343/203/207/343/202/243/343/203/254/343/202/257/343/203/210/343/203/252/345/220/215/source_location_encoding.rb +35 -0
- data/spec/tengine/core/bugfix//351/235/236ACSII/343/201/256/343/203/207/343/202/243/343/203/254/343/202/257/343/203/210/343/203/252/345/220/215//351/235/236ASCII/343/201/256/343/203/225/343/202/241/343/202/244/343/203/253/345/220/215_dsl.rb +38 -0
- data/spec/tengine/core/config/atd_spec.rb +62 -0
- data/spec/tengine/core/config/core_spec.rb +479 -0
- data/spec/tengine/core/config/heartbeat_watcher_spec.rb +62 -0
- data/spec/tengine/core/config/syntax_error_in_erb.yml.erb +13 -0
- data/spec/tengine/core/config/wrong_category_name.yml.erb +13 -0
- data/spec/tengine/core/config/wrong_field_name.yml.erb +12 -0
- data/spec/tengine/core/config/wrong_yaml.yml.erb +13 -0
- data/spec/tengine/core/config_spec/another_port.yml +54 -0
- data/spec/tengine/core/config_spec/config_with_dir_absolute_load_path.yml +16 -0
- data/spec/tengine/core/config_spec/config_with_dir_relative_load_path.yml +16 -0
- data/spec/tengine/core/config_spec/config_with_file_absolute_load_path.yml +16 -0
- data/spec/tengine/core/config_spec/config_with_file_relative_load_path.yml +16 -0
- data/spec/tengine/core/config_spec/log_config_spec.rb +235 -0
- data/spec/tengine/core/driveable_spec.rb +240 -0
- data/spec/tengine/core/driver_spec.rb +159 -0
- data/spec/tengine/core/dsl_loader_spec.rb +172 -0
- data/spec/tengine/core/dsls/uc08_if_both_a_and_b_occurs_spec.rb +35 -0
- data/spec/tengine/core/dsls/uc10_if_the_event_occurs_at_the_server_spec.rb +58 -0
- data/spec/tengine/core/dsls/uc50_commit_event_at_first_spec.rb +29 -0
- data/spec/tengine/core/dsls/uc52_commit_event_after_all_handler_submit_spec.rb +33 -0
- data/spec/tengine/core/dsls/uc52_never_commit_event_unless_all_handler_submit_spec.rb +37 -0
- data/spec/tengine/core/dsls/uc53_submit_outside_of_handler_spec.rb +37 -0
- data/spec/tengine/core/dsls/uc60_event_in_handler_spec.rb +31 -0
- data/spec/tengine/core/dsls/uc61_event_outside_of_handler_spec.rb +37 -0
- data/spec/tengine/core/dsls/uc62_session_in_driver_spec.rb +36 -0
- data/spec/tengine/core/dsls/uc63_session_outside_of_driver_spec.rb +35 -0
- data/spec/tengine/core/dsls/uc64_safety_countup_spec.rb +134 -0
- data/spec/tengine/core/dsls/uc70_driver_enabled_on_activation_spec.rb +39 -0
- data/spec/tengine/core/dsls/uc71_driver_disabled_on_activation_spec.rb +36 -0
- data/spec/tengine/core/dsls/uc72_setup_eventmachine_spec.rb +39 -0
- data/spec/tengine/core/dsls/uc80_raise_io_error_spec.rb +53 -0
- data/spec/tengine/core/dsls/uc81_raise_runtime_error_spec.rb +49 -0
- data/spec/tengine/core/event/finder_spec.rb +136 -0
- data/spec/tengine/core/event_exception_reportable_spec.rb +33 -0
- data/spec/tengine/core/event_spec.rb +161 -0
- data/spec/tengine/core/event_wrapper_spec.rb +35 -0
- data/spec/tengine/core/handler_path_spec.rb +87 -0
- data/spec/tengine/core/handler_spec.rb +190 -0
- data/spec/tengine/core/heartbeat_watcher_spec.rb +131 -0
- data/spec/tengine/core/io_to_logger_spec.rb +30 -0
- data/spec/tengine/core/kernel_spec.rb +885 -0
- data/spec/tengine/core/mutex_spec.rb +184 -0
- data/spec/tengine/core/optimistic_lock_spec.rb +55 -0
- data/spec/tengine/core/scheculer_spec.rb +121 -0
- data/spec/tengine/core/selectable_attr_spec.rb +30 -0
- data/spec/tengine/core/session_spec.rb +104 -0
- data/spec/tengine/core/setting_spec.rb +79 -0
- data/spec/tengine/core_spec.rb +13 -0
- data/spec/tengine_spec.rb +14 -0
- data/tengine_core.gemspec +248 -0
- data/tmp/log/.gitignore +1 -0
- data/tmp/tengined_status/.gitignore +1 -0
- metadata +421 -0
@@ -0,0 +1,885 @@
|
|
1
|
+
# -*- coding: utf-8 -*-
|
2
|
+
require 'spec_helper'
|
3
|
+
require 'amqp'
|
4
|
+
require 'eventmachine'
|
5
|
+
|
6
|
+
# ログを黙らせたり喋らせたりする
|
7
|
+
require 'amq/client'
|
8
|
+
require 'mongoid'
|
9
|
+
if $DEBUG
|
10
|
+
require 'logger'
|
11
|
+
AMQP::Session.logger = Tengine.logger = Mongoid.logger = Logger.new(STDERR)
|
12
|
+
else
|
13
|
+
AMQP::Session.logger = Tengine.logger = Mongoid.logger = Tengine::NullLogger.new
|
14
|
+
end
|
15
|
+
|
16
|
+
describe Tengine::Core::Kernel do
|
17
|
+
before do
|
18
|
+
Tengine::Core::Driver.delete_all
|
19
|
+
Tengine::Core::HandlerPath.delete_all
|
20
|
+
Tengine::Core::Event.delete_all
|
21
|
+
end
|
22
|
+
|
23
|
+
describe :start do
|
24
|
+
# describe :bind, "handlerのblockをメモリ上で保持" do
|
25
|
+
# before do
|
26
|
+
# config = Tengine::Core::Config::Core.new({
|
27
|
+
# :tengined => {
|
28
|
+
# :load_path => File.expand_path('../../../examples/uc01_execute_processing_for_event.rb', File.dirname(__FILE__)),
|
29
|
+
# },
|
30
|
+
# })
|
31
|
+
# @kernel = Tengine::Core::Kernel.new(config)
|
32
|
+
# @driver = Tengine::Core::Driver.new(:name => "driver01", :version => config.dsl_version, :enabled => true)
|
33
|
+
# @handler1 = @driver.handlers.new(:filepath => "uc01_execute_processing_for_event.rb", :lineno => 7, :event_type_names => ["event01"])
|
34
|
+
# @driver.save!
|
35
|
+
# end
|
36
|
+
|
37
|
+
# it "event_type_nameからblockを検索することができる" do
|
38
|
+
# @kernel.bind
|
39
|
+
# @kernel.context.__block_for__(@handler1).should_not be_nil
|
40
|
+
# end
|
41
|
+
|
42
|
+
# context "拡張モジュールあり" do
|
43
|
+
# before(:all) do
|
44
|
+
# @ext_mod1 = Module.new{}
|
45
|
+
# @ext_mod1.instance_eval do
|
46
|
+
# def dsl_binder; self; end
|
47
|
+
# end
|
48
|
+
# Tengine.plugins.add(@ext_mod1)
|
49
|
+
# end
|
50
|
+
|
51
|
+
# it "Kernel#contextに拡張モジュールがextendされる" do
|
52
|
+
# @kernel.bind
|
53
|
+
# @kernel.context.__block_for__(@handler1).should_not be_nil
|
54
|
+
# @kernel.context.should be_a(Tengine::Core::DslBinder)
|
55
|
+
# @kernel.context.should be_a(@ext_mod1)
|
56
|
+
# end
|
57
|
+
# end
|
58
|
+
|
59
|
+
# end
|
60
|
+
|
61
|
+
describe :wait_for_activation, "activate待ち" do
|
62
|
+
before do
|
63
|
+
config = Tengine::Core::Config::Core.new({
|
64
|
+
:tengined => {
|
65
|
+
:load_path => File.expand_path('../../../examples/uc01_execute_processing_for_event.rb', File.dirname(__FILE__)),
|
66
|
+
:wait_activation => true,
|
67
|
+
:activation_timeout => 3,
|
68
|
+
:activation_dir => File.expand_path('.', File.dirname(__FILE__)),
|
69
|
+
},
|
70
|
+
})
|
71
|
+
@kernel = Tengine::Core::Kernel.new(config)
|
72
|
+
@driver = Tengine::Core::Driver.new(:name => "driver01", :version => config.dsl_version, :enabled => true)
|
73
|
+
@handler1 = @driver.handlers.new(:filepath => "uc01_execute_processing_for_event.rb", :lineno => 7, :event_type_names => ["event01"])
|
74
|
+
@driver.save!
|
75
|
+
@activation_file_path = "#{@kernel.config[:tengined][:activation_dir]}\/tengined_#{Process.pid}.activation"
|
76
|
+
end
|
77
|
+
|
78
|
+
after do
|
79
|
+
FileUtils.rm_f(@activation_file_path)
|
80
|
+
end
|
81
|
+
|
82
|
+
it "activationファイルが生成されたらactivateされる" do
|
83
|
+
@kernel.should_receive(:activate)
|
84
|
+
t1 = Thread.new {
|
85
|
+
@kernel.start
|
86
|
+
}
|
87
|
+
t2 = Thread.new {
|
88
|
+
FileUtils.touch(@activation_file_path)
|
89
|
+
}
|
90
|
+
t1.join
|
91
|
+
t2.join
|
92
|
+
end
|
93
|
+
|
94
|
+
it "activationファイルが生成されないままならタイムアウトになる" do
|
95
|
+
lambda {
|
96
|
+
@kernel.should_not_receive(:activate)
|
97
|
+
@kernel.start
|
98
|
+
}.should raise_error(Tengine::Core::ActivationTimeoutError, "activation file found timeout error.")
|
99
|
+
end
|
100
|
+
end
|
101
|
+
|
102
|
+
describe :activate, "メッセージの受信を開始" do
|
103
|
+
before do
|
104
|
+
@header = AMQP::Header.new(@mock_channel, nil, {
|
105
|
+
:routing_key => "",
|
106
|
+
:content_type => "application/octet-stream",
|
107
|
+
:priority => 0,
|
108
|
+
:headers => { },
|
109
|
+
:timestamp => Time.now,
|
110
|
+
:type => "",
|
111
|
+
:delivery_tag => 1,
|
112
|
+
:redelivered => false,
|
113
|
+
:exchange => "tengine_event_exchange",
|
114
|
+
})
|
115
|
+
|
116
|
+
config = Tengine::Core::Config::Core.new({
|
117
|
+
:tengined => {
|
118
|
+
:load_path => File.expand_path('../../../examples/uc01_execute_processing_for_event.rb', File.dirname(__FILE__)),
|
119
|
+
:wait_activation => false,
|
120
|
+
:confirmation_threshold => 'info'
|
121
|
+
},
|
122
|
+
:heartbeat => {
|
123
|
+
:core => {
|
124
|
+
:interval => 1024,
|
125
|
+
:expire => 32768,
|
126
|
+
},
|
127
|
+
},
|
128
|
+
})
|
129
|
+
@kernel = Tengine::Core::Kernel.new(config)
|
130
|
+
@driver = Tengine::Core::Driver.new(
|
131
|
+
:name => "driver01", :version => config.dsl_version, :enabled => true)
|
132
|
+
@handler1 = @driver.handlers.new(
|
133
|
+
:filepath => "uc01_execute_processing_for_event.rb", :lineno => 7, :event_type_names => ["event01"])
|
134
|
+
@driver.save!
|
135
|
+
@event1 = Tengine::Core::Event.new(:event_type_name => :event01, :key => "uuid1", :sender_name => "localhost")
|
136
|
+
@event1.save!
|
137
|
+
end
|
138
|
+
|
139
|
+
context "イベントの受信待ち状態になる" do
|
140
|
+
before do
|
141
|
+
# eventmachine と mq の mock を生成
|
142
|
+
mock_mq = mock("mq")
|
143
|
+
mock_sender = mock("sender")
|
144
|
+
@kernel.stub(:mq).and_return(mock_mq)
|
145
|
+
@kernel.stub(:sender).and_return(mock_sender)
|
146
|
+
mock_sender.stub(:fire)
|
147
|
+
mock_mq.stub(:initiate_termination).and_yield
|
148
|
+
mock_mq.stub(:unsubscribe).and_yield
|
149
|
+
mock_mq.stub(:stop).and_yield
|
150
|
+
mock_mq.stub(:add_hook)
|
151
|
+
mock_mq.stub(:subscribe).with(nil).with(:ack => true, :nowait => false, :confirm => an_instance_of(Proc)) do |h, b|
|
152
|
+
h[:confirm].yield(mock("confirm-ok"))
|
153
|
+
#b.yield(@header, :message)
|
154
|
+
end
|
155
|
+
# subscribe されていることを検証
|
156
|
+
mock_mq.should_receive(:subscribe).with(:ack => true, :nowait => false, :confirm => an_instance_of(Proc))
|
157
|
+
end
|
158
|
+
|
159
|
+
it "heartbeatは有効にならない" do
|
160
|
+
@kernel.config[:heartbeat][:core][:interval] = -1
|
161
|
+
@kernel.should_receive(:setup_mq_connection)
|
162
|
+
sender = mock(:sender)
|
163
|
+
@kernel.stub(:sender).and_return(sender)
|
164
|
+
sender.stub(:fire).with("finished.process.core.tengine", an_instance_of(Hash)).and_yield
|
165
|
+
@kernel.start
|
166
|
+
end
|
167
|
+
|
168
|
+
it "heartbeatは有効になる" do
|
169
|
+
@kernel.config[:heartbeat][:core][:interval] = 65535
|
170
|
+
EM.should_receive(:add_periodic_timer).with(65535).and_yield
|
171
|
+
sender = mock(:sender)
|
172
|
+
@kernel.stub(:sender).and_return(sender)
|
173
|
+
sender.stub(:fire).with("finished.process.core.tengine", an_instance_of(Hash)).and_yield
|
174
|
+
sender.stub(:fire).with("core.heartbeat.tengine", an_instance_of(Hash))
|
175
|
+
@kernel.start do
|
176
|
+
@kernel.stop
|
177
|
+
end
|
178
|
+
end
|
179
|
+
|
180
|
+
it "heartbeatが送られる" do
|
181
|
+
@kernel.should_receive(:setup_mq_connection)
|
182
|
+
EM.should_receive(:add_periodic_timer).with(1024).and_yield
|
183
|
+
mock_sender = mock(:sender)
|
184
|
+
@kernel.stub(:sender).and_return(mock_sender)
|
185
|
+
mock_sender.stub(:fire).with("finished.process.core.tengine", an_instance_of(Hash)).and_yield
|
186
|
+
mock_sender.stub(:fire).with("core.heartbeat.tengine", an_instance_of(Hash))
|
187
|
+
@kernel.start
|
188
|
+
end
|
189
|
+
end
|
190
|
+
|
191
|
+
context "発火されたイベントを登録できる" do
|
192
|
+
before do
|
193
|
+
# eventmachine と mq の mock を生成
|
194
|
+
mock_mq = mock("mq")
|
195
|
+
mock_sender = mock("sender")
|
196
|
+
@kernel.stub(:mq).and_return(mock_mq)
|
197
|
+
@kernel.stub(:sender).and_return(mock_sender)
|
198
|
+
mock_sender.stub(:fire)
|
199
|
+
mock_mq.stub(:initiate_termination).and_yield
|
200
|
+
mock_mq.stub(:unsubscribe).and_yield
|
201
|
+
mock_mq.stub(:stop).and_yield
|
202
|
+
mock_mq.stub(:add_hook)
|
203
|
+
mock_mq.stub(:subscribe).with(nil).with(:ack => true, :nowait => false, :confirm => an_instance_of(Proc)) do |h, b|
|
204
|
+
h[:confirm].yield(mock("confirm-ok"))
|
205
|
+
b.yield(@header, :message)
|
206
|
+
end
|
207
|
+
|
208
|
+
# subscribe してみる
|
209
|
+
@mock_raw_event = mock(:row_event)
|
210
|
+
Tengine::Event.should_receive(:parse).with(:message).and_return(@mock_raw_event)
|
211
|
+
|
212
|
+
@header.should_receive(:ack)
|
213
|
+
end
|
214
|
+
|
215
|
+
it "confirmation_threshold以下なら登録されたイベントはconfirmedがtrue" do
|
216
|
+
@mock_raw_event.stub!(:key).and_return("uniq_key")
|
217
|
+
@mock_raw_event.stub!(:sender_name).and_return("localhost")
|
218
|
+
@mock_raw_event.stub!(:attributes).and_return(:event_type_name => :foo, :key => "uniq_key", :level => Tengine::Event::LEVELS_INV[:info])
|
219
|
+
@mock_raw_event.stub!(:level).and_return(Tengine::Event::LEVELS_INV[:info])
|
220
|
+
@mock_raw_event.stub!(:event_type_name).and_return("foo")
|
221
|
+
count = lambda{ Tengine::Core::Event.where(:event_type_name => :foo, :confirmed => true).count }
|
222
|
+
expect{ @kernel.start { @kernel.stop } }.should change(count, :call).by(1) # イベントが登録されていることを検証
|
223
|
+
end
|
224
|
+
|
225
|
+
it "confirmation_threshold以下なら登録されたイベントはconfirmedがfalse" do
|
226
|
+
@mock_raw_event.stub!(:key).and_return("uniq_key")
|
227
|
+
@mock_raw_event.stub!(:sender_name).and_return("localhost")
|
228
|
+
@mock_raw_event.stub!(:attributes).and_return(:event_type_name => :foo, :key => "uniq_key", :level => Tengine::Event::LEVELS_INV[:warn])
|
229
|
+
@mock_raw_event.stub!(:level).and_return(Tengine::Event::LEVELS_INV[:warn])
|
230
|
+
@mock_raw_event.stub!(:event_type_name).and_return("foo")
|
231
|
+
count = lambda{ Tengine::Core::Event.where(:event_type_name => :foo, :confirmed => false).count }
|
232
|
+
@kernel.should_receive(:setup_mq_connection)
|
233
|
+
expect{ @kernel.start }.should change(count, :call).by(1) # イベントが登録されていることを検証
|
234
|
+
end
|
235
|
+
end
|
236
|
+
|
237
|
+
context "イベントストアへの登録有無" do
|
238
|
+
it "不正なフォーマットのメッセージの場合、イベントストアへ登録を行わずACKを返却" do
|
239
|
+
@header.should_receive(:ack)
|
240
|
+
@kernel.process_message(@header, "invalid format message").should_not be_true
|
241
|
+
end
|
242
|
+
|
243
|
+
it "https://www.pivotaltracker.com/story/show/22698533" do
|
244
|
+
ev = Tengine::Event.new :key => "2498e870-11cd-012f-f8c0-48bcc89f84e1", :source_name => "localhost/8110", :sender_name => "localhost/8110", :level => 2, :occurred_at => Time.now, :properties => {}, :event_type_name => ""
|
245
|
+
@header.should_receive(:ack)
|
246
|
+
@kernel.process_message(@header, ev.to_json).should_not be_true
|
247
|
+
end
|
248
|
+
|
249
|
+
it "keyがnilのイベント場合、イベントストアへ登録を行わずACKを返却" do
|
250
|
+
raw_event = Tengine::Event.new(:key => "", :sender_name => "another_host", :event_type_name => "event1")
|
251
|
+
@header.should_receive(:ack)
|
252
|
+
@kernel.process_message(@header, raw_event.to_json).should_not be_true
|
253
|
+
end
|
254
|
+
|
255
|
+
it "keyが同じ、sender_nameが異なる場合は、イベントストアへ登録を行わずACKを返却" do
|
256
|
+
@header.should_receive(:ack)
|
257
|
+
raw_event = Tengine::Event.new(:key => "uuid1", :sender_name => "another_host", :event_type_name => "event1")
|
258
|
+
lambda {
|
259
|
+
Tengine::Core::Event.create!(raw_event.attributes.update(:confirmed => (raw_event.level <= @kernel.config.confirmation_threshold)))
|
260
|
+
}.should raise_error(Mongo::OperationFailure)
|
261
|
+
@kernel.process_message(@header, raw_event.to_json).should_not be_true
|
262
|
+
end
|
263
|
+
|
264
|
+
it "keyが異なる場合は、イベントストアへ登録を行い、ACKを返却" do
|
265
|
+
@header.should_receive(:ack)
|
266
|
+
raw_event = Tengine::Event.new(:key => "uuid99", :sender_name => "another_host", :event_type_name => "event1")
|
267
|
+
Tengine::Core::Event.should_receive(:create!).and_return(Tengine::Core::Event.new(raw_event.attributes))
|
268
|
+
@kernel.process_message(@header, raw_event.to_json).should be_true
|
269
|
+
end
|
270
|
+
end
|
271
|
+
|
272
|
+
context "イベント処理失敗イベントの発火" do
|
273
|
+
before do
|
274
|
+
# eventmachine と mq の mock を生成
|
275
|
+
mock_mq = mock("mq")
|
276
|
+
@kernel.stub(:mq).and_return(mock_mq)
|
277
|
+
mock_mq.stub(:initiate_termination).and_yield
|
278
|
+
mock_mq.stub(:unsubscribe).and_yield
|
279
|
+
mock_mq.stub(:stop).and_yield
|
280
|
+
mock_mq.stub(:add_hook)
|
281
|
+
mock_mq.stub(:subscribe).with(nil).with(:ack => true, :nowait => false, :confirm => an_instance_of(Proc)) do |h, b|
|
282
|
+
h[:confirm].yield(mock("confirm-ok"))
|
283
|
+
b.yield(@header, :message)
|
284
|
+
end
|
285
|
+
|
286
|
+
# subscribe してみる
|
287
|
+
@raw_event = Tengine::Event.new(:key => "uuid1", :sender_name => "localhost", :event_type_name => "event1")
|
288
|
+
Tengine::Event.should_receive(:parse).with(:message).and_return(@raw_event)
|
289
|
+
@header.should_receive(:ack)
|
290
|
+
end
|
291
|
+
|
292
|
+
it "既に登録されているイベントとkey, sender_nameが一致するイベントを受信した場合、発火" do
|
293
|
+
mock_sender = mock(:sender)
|
294
|
+
Tengine::Event::Sender.should_receive(:new).with(@kernel.mq).and_return(mock_sender)
|
295
|
+
mock_sender.should_receive(:default_keep_connection=).with(true)
|
296
|
+
mock_sender.should_receive(:fire).with("#{@raw_event.event_type_name}.failed.tengined",
|
297
|
+
{
|
298
|
+
:level => Tengine::Event::LEVELS_INV[:error],
|
299
|
+
:properties => { :original_event => @raw_event }
|
300
|
+
})
|
301
|
+
mock_sender.stub(:fire).with("finished.process.core.tengine", an_instance_of(Hash)).and_yield
|
302
|
+
@kernel.start { @kernel.stop }
|
303
|
+
events = Tengine::Core::Event.where(:key => @raw_event.key, :sender_name => @raw_event.sender_name)
|
304
|
+
events.count.should == 1
|
305
|
+
end
|
306
|
+
end
|
307
|
+
|
308
|
+
it "イベント種別に対応したハンドラの処理を実行することができる" do
|
309
|
+
# eventmachine と mq の mock を生成
|
310
|
+
mock_mq = mock("mq")
|
311
|
+
mock_sender = mock("sender")
|
312
|
+
@kernel.stub(:mq).and_return(mock_mq)
|
313
|
+
@kernel.stub(:sender).and_return(mock_sender)
|
314
|
+
mock_sender.stub(:fire)
|
315
|
+
mock_mq.stub(:initiate_termination).and_yield
|
316
|
+
mock_mq.stub(:unsubscribe).and_yield
|
317
|
+
mock_mq.stub(:stop).and_yield
|
318
|
+
mock_mq.stub(:add_hook)
|
319
|
+
mock_mq.stub(:subscribe).with(nil).with(:ack => true, :nowait => false, :confirm => an_instance_of(Proc)) do |h, b|
|
320
|
+
h[:confirm].yield(mock("confirm-ok"))
|
321
|
+
b.yield(@header, :message)
|
322
|
+
end
|
323
|
+
|
324
|
+
# subscribe してみる
|
325
|
+
mock_raw_event = mock(:row_event)
|
326
|
+
mock_raw_event.stub!(:key).and_return("uuid")
|
327
|
+
mock_raw_event.stub!(:sender_name).and_return("localhost")
|
328
|
+
mock_raw_event.should_receive(:attributes).and_return(:event_type_name => :event01, :key => "uuid")
|
329
|
+
mock_raw_event.stub!(:level).and_return(Tengine::Event::LEVELS_INV[:info])
|
330
|
+
mock_raw_event.stub!(:event_type_name).and_return("event01")
|
331
|
+
Tengine::Event.should_receive(:parse).with(:message).and_return(mock_raw_event)
|
332
|
+
# イベントの登録
|
333
|
+
Tengine::Core::Event.should_receive(:create!).with(:event_type_name => :event01, :key => "uuid", :confirmed => true).and_return(@event1)
|
334
|
+
|
335
|
+
# ハンドラの実行を検証
|
336
|
+
Tengine::Core::HandlerPath.should_receive(:find_handlers).with("event01").and_return([@handler1])
|
337
|
+
@handler1.should_receive(:match?).with(@event1).and_return(true)
|
338
|
+
|
339
|
+
# 仕様変更のためイベントハンドラの処理を確認するのは一旦コメントアウトしました
|
340
|
+
# @kernel.context.should_receive(:puts).with("handler01")
|
341
|
+
|
342
|
+
@header.should_receive(:ack)
|
343
|
+
|
344
|
+
# 実行
|
345
|
+
@kernel.start
|
346
|
+
end
|
347
|
+
|
348
|
+
context "*.failed.tengine" do
|
349
|
+
before do
|
350
|
+
@uuid = ::UUID.new
|
351
|
+
@header.stub(:ack)
|
352
|
+
@sender = mock(:sender)
|
353
|
+
@kernel.stub(:sender).and_return(@sender)
|
354
|
+
@sender.should_not_receive(:fire)
|
355
|
+
end
|
356
|
+
|
357
|
+
context "正常系" do
|
358
|
+
it "メッセージストアに保存" do
|
359
|
+
e = Tengine::Event.new key: @uuid.generate, event_type_name: "something.failed.tengine"
|
360
|
+
|
361
|
+
@kernel.process_message @header, e.to_json
|
362
|
+
|
363
|
+
Tengine::Core::Event.where(key: e.key).count.should == 1
|
364
|
+
Tengine::Core::Event.where(key: e.key).first.event_type_name.should =~ /failed\.tengine$/
|
365
|
+
end
|
366
|
+
end
|
367
|
+
|
368
|
+
context "異常系" do
|
369
|
+
it "無限地獄の回避" do
|
370
|
+
Tengine::Core::Event.stub(:create!).and_raise(Mongo::OperationFailure.new)
|
371
|
+
e = Tengine::Event.new key: @uuid.generate, event_type_name: "something.failed.tengine"
|
372
|
+
|
373
|
+
@kernel.process_message @header, e.to_json
|
374
|
+
end
|
375
|
+
end
|
376
|
+
end
|
377
|
+
|
378
|
+
context "heartbeat" do
|
379
|
+
before do
|
380
|
+
@uuid = ::UUID.new
|
381
|
+
@header.stub(:ack)
|
382
|
+
@sender = mock(:sender)
|
383
|
+
@kernel.stub(:sender).and_return(@sender)
|
384
|
+
@sender.should_not_receive(:fire)
|
385
|
+
end
|
386
|
+
|
387
|
+
shared_examples "generic heartbeats" do
|
388
|
+
context "正常系" do
|
389
|
+
context "初回" do
|
390
|
+
before do
|
391
|
+
@u = @uuid.generate
|
392
|
+
Tengine::Core::Event.where(key: @u).delete_all
|
393
|
+
end
|
394
|
+
it "beatは保存される" do
|
395
|
+
@kernel.process_message(@header, Tengine::Event.new(key: @u, event_type_name: "#{kind}.heartbeat.tengine").to_json).should be_true
|
396
|
+
Tengine::Core::Event.where(key: @u).count.should == 1
|
397
|
+
Tengine::Core::Event.where(key: @u).first.event_type_name.should =~ /^#{kind}/
|
398
|
+
end
|
399
|
+
|
400
|
+
it "finishedは保存される" do
|
401
|
+
@kernel.process_message(@header, Tengine::Event.new(key: @u, event_type_name: "finished.process.#{kind}.tengine").to_json).should be_true
|
402
|
+
Tengine::Core::Event.where(key: @u).count.should == 1
|
403
|
+
Tengine::Core::Event.where(key: @u).first.event_type_name.should =~ /^finished/
|
404
|
+
end
|
405
|
+
|
406
|
+
it "expiredは保存される" do
|
407
|
+
@kernel.process_message(@header, Tengine::Event.new(key: @u, event_type_name: "expired.#{kind}.heartbeat.tengine").to_json).should be_true
|
408
|
+
Tengine::Core::Event.where(key: @u).count.should == 1
|
409
|
+
Tengine::Core::Event.where(key: @u).first.event_type_name.should =~ /^expired/
|
410
|
+
end
|
411
|
+
end
|
412
|
+
|
413
|
+
it "beat -> beat -> beat" do
|
414
|
+
e = Tengine::Event.new key: @uuid.generate, event_type_name: "#{kind}.heartbeat.tengine"
|
415
|
+
|
416
|
+
@kernel.process_message @header, e.to_json
|
417
|
+
@kernel.process_message @header, e.to_json
|
418
|
+
@kernel.process_message @header, e.to_json
|
419
|
+
|
420
|
+
Tengine::Core::Event.where(key: e.key).count.should == 1
|
421
|
+
Tengine::Core::Event.where(key: e.key).first.event_type_name.should =~ /^#{kind}/
|
422
|
+
end
|
423
|
+
|
424
|
+
it "beat -> finished (finishedが勝つ)" do
|
425
|
+
u = @uuid.generate
|
426
|
+
@kernel.process_message @header, Tengine::Event.new(key: u, event_type_name: "#{kind}.heartbeat.tengine").to_json
|
427
|
+
@kernel.process_message @header, Tengine::Event.new(key: u, event_type_name: "finished.process.#{kind}.tengine").to_json
|
428
|
+
|
429
|
+
Tengine::Core::Event.where(key: u).count.should == 1
|
430
|
+
Tengine::Core::Event.where(key: u).first.event_type_name.should =~ /finished/
|
431
|
+
end
|
432
|
+
|
433
|
+
it "beat -> expired (expiredが勝つ)" do
|
434
|
+
u = @uuid.generate
|
435
|
+
@kernel.process_message @header, Tengine::Event.new(key: u, event_type_name: "#{kind}.heartbeat.tengine").to_json
|
436
|
+
@kernel.process_message @header, Tengine::Event.new(key: u, event_type_name: "expired.#{kind}.heartbeat.tengine").to_json
|
437
|
+
|
438
|
+
Tengine::Core::Event.where(key: u).count.should == 1
|
439
|
+
Tengine::Core::Event.where(key: u).first.event_type_name.should =~ /expired/
|
440
|
+
end
|
441
|
+
|
442
|
+
it "beat -> finish -> expired (expiredが勝つ)" do
|
443
|
+
u = @uuid.generate
|
444
|
+
@kernel.process_message @header, Tengine::Event.new(key: u, event_type_name: "#{kind}.heartbeat.tengine").to_json
|
445
|
+
@kernel.process_message @header, Tengine::Event.new(key: u, event_type_name: "finished.process.#{kind}.tengine").to_json
|
446
|
+
@kernel.process_message @header, Tengine::Event.new(key: u, event_type_name: "expired.#{kind}.heartbeat.tengine").to_json
|
447
|
+
|
448
|
+
Tengine::Core::Event.where(key: u).count.should == 1
|
449
|
+
Tengine::Core::Event.where(key: u).first.event_type_name.should =~ /expired/
|
450
|
+
end
|
451
|
+
|
452
|
+
it "finished -> beat (finishedが勝つ)" do
|
453
|
+
u = @uuid.generate
|
454
|
+
@kernel.process_message @header, Tengine::Event.new(key: u, event_type_name: "finished.process.#{kind}.tengine").to_json
|
455
|
+
@kernel.process_message @header, Tengine::Event.new(key: u, event_type_name: "#{kind}.heartbeat.tengine").to_json
|
456
|
+
|
457
|
+
Tengine::Core::Event.where(key: u).count.should == 1
|
458
|
+
Tengine::Core::Event.where(key: u).first.event_type_name.should =~ /finished/
|
459
|
+
end
|
460
|
+
|
461
|
+
it "finished -> finished (上書き)" do
|
462
|
+
e = Tengine::Event.new key: @uuid.generate, event_type_name: "finished.process.#{kind}.tengine"
|
463
|
+
|
464
|
+
@kernel.process_message @header, e.to_json
|
465
|
+
@kernel.process_message @header, e.to_json
|
466
|
+
|
467
|
+
Tengine::Core::Event.where(key: e.key).count.should == 1
|
468
|
+
Tengine::Core::Event.where(key: e.key).first.event_type_name.should =~ /finished/
|
469
|
+
end
|
470
|
+
|
471
|
+
it "finished -> expired (expiredが勝つ)" do
|
472
|
+
u = @uuid.generate
|
473
|
+
@kernel.process_message @header, Tengine::Event.new(key: u, event_type_name: "finished.process.#{kind}.tengine").to_json
|
474
|
+
@kernel.process_message @header, Tengine::Event.new(key: u, event_type_name: "expired.#{kind}.heartbeat.tengine").to_json
|
475
|
+
|
476
|
+
Tengine::Core::Event.where(key: u).count.should == 1
|
477
|
+
Tengine::Core::Event.where(key: u).first.event_type_name.should =~ /expired/
|
478
|
+
end
|
479
|
+
|
480
|
+
it "expired -> beat (expiredが勝つ)" do
|
481
|
+
u = @uuid.generate
|
482
|
+
@kernel.process_message @header, Tengine::Event.new(key: u, event_type_name: "expired.#{kind}.heartbeat.tengine").to_json
|
483
|
+
@kernel.process_message @header, Tengine::Event.new(key: u, event_type_name: "#{kind}.heartbeat.tengine").to_json
|
484
|
+
|
485
|
+
Tengine::Core::Event.where(key: u).count.should == 1
|
486
|
+
Tengine::Core::Event.where(key: u).first.event_type_name.should =~ /expired/
|
487
|
+
end
|
488
|
+
|
489
|
+
it "expired -> finished (expiredが勝つ)" do
|
490
|
+
u = @uuid.generate
|
491
|
+
@kernel.process_message @header, Tengine::Event.new(key: u, event_type_name: "expired.#{kind}.heartbeat.tengine").to_json
|
492
|
+
@kernel.process_message @header, Tengine::Event.new(key: u, event_type_name: "finished.process.#{kind}.tengine").to_json
|
493
|
+
|
494
|
+
Tengine::Core::Event.where(key: u).count.should == 1
|
495
|
+
Tengine::Core::Event.where(key: u).first.event_type_name.should =~ /expired/
|
496
|
+
end
|
497
|
+
|
498
|
+
it "expired -> expired (上書き)" do
|
499
|
+
e = Tengine::Event.new key: @uuid.generate, event_type_name: "expired.#{kind}.heartbeat.tengine"
|
500
|
+
|
501
|
+
@kernel.process_message @header, e.to_json
|
502
|
+
@kernel.process_message @header, e.to_json
|
503
|
+
|
504
|
+
Tengine::Core::Event.where(key: e.key).count.should == 1
|
505
|
+
Tengine::Core::Event.where(key: e.key).first.event_type_name.should =~ /expired/
|
506
|
+
end
|
507
|
+
end
|
508
|
+
|
509
|
+
context "異常系" do
|
510
|
+
['"#{kind}.heartbeat.tengine"',
|
511
|
+
'"finished.process.#{kind}.tengine"',
|
512
|
+
'"expired.#{kind}.heartbeat.tengine"'
|
513
|
+
].each do |name|
|
514
|
+
context name do
|
515
|
+
it "Mongo::OperationFailureの場合、failed eventを連鎖" do
|
516
|
+
@sender.stub(:fire).with("#{kind}.heartbeat.tengine.failed.tengine", an_instance_of(Hash))
|
517
|
+
Tengine::Core::Event.stub(:create!).and_raise(Mongo::OperationFailure.new)
|
518
|
+
@kernel.stub(:upsert).and_raise Mongo::OperationFailure
|
519
|
+
|
520
|
+
@kernel.process_message @header, Tengine::Event.new(key: @uuid.generate, event_type_name: eval(name)).to_json
|
521
|
+
end
|
522
|
+
|
523
|
+
it "Mongoid::Errors::Validationの場合、failed eventを連鎖" do
|
524
|
+
x = Tengine::Core::Event.new
|
525
|
+
x.valid? # false
|
526
|
+
@sender.stub(:fire).with("#{kind}.heartbeat.tengine.failed.tengine", an_instance_of(Hash))
|
527
|
+
Tengine::Core::Event.stub(:create!).and_raise(Mongoid::Errors::Validations.new(x))
|
528
|
+
@kernel.stub(:upsert).and_raise Mongoid::Errors::Validations.new(x)
|
529
|
+
|
530
|
+
@kernel.process_message @header, Tengine::Event.new(key: @uuid.generate, event_type_name: eval(name)).to_json
|
531
|
+
end
|
532
|
+
|
533
|
+
it "その他の場合、例外を外に伝播しない" do
|
534
|
+
Tengine::Core::Event.stub(:find_or_create_by_key_then_update_with_block).and_raise StandardError
|
535
|
+
expect do
|
536
|
+
@kernel.process_message @header, Tengine::Event.new(key: @uuid.generate, event_type_name: eval(name)).to_json
|
537
|
+
end.to_not raise_exception(StandardError)
|
538
|
+
@kernel.process_message @header, Tengine::Event.new(key: @uuid.generate, event_type_name: eval(name)).to_json
|
539
|
+
end
|
540
|
+
end
|
541
|
+
end
|
542
|
+
end
|
543
|
+
end
|
544
|
+
|
545
|
+
describe "job heartbeat" do
|
546
|
+
let(:kind) {"job"}
|
547
|
+
it_behaves_like "generic heartbeats"
|
548
|
+
end
|
549
|
+
|
550
|
+
describe "core heartbeat" do
|
551
|
+
let(:kind) {"core"}
|
552
|
+
it_behaves_like "generic heartbeats"
|
553
|
+
end
|
554
|
+
|
555
|
+
describe "heartbeat watcher's heartbeat" do
|
556
|
+
let(:kind) {"hbw"}
|
557
|
+
it_behaves_like "generic heartbeats"
|
558
|
+
end
|
559
|
+
|
560
|
+
describe "atd heartbeat" do
|
561
|
+
let(:kind) {"atd"}
|
562
|
+
it_behaves_like "generic heartbeats"
|
563
|
+
end
|
564
|
+
end
|
565
|
+
|
566
|
+
describe "schedule" do
|
567
|
+
it "tenginedが調停する" do
|
568
|
+
@header.stub(:ack)
|
569
|
+
n = "alert.execution.job.tengine"
|
570
|
+
s = "test test"
|
571
|
+
e = Tengine::Event.new event_type_name: n, source_name: s
|
572
|
+
|
573
|
+
@kernel.process_message @header, e.to_json
|
574
|
+
@kernel.process_message @header, e.to_json
|
575
|
+
@kernel.process_message @header, e.to_json
|
576
|
+
|
577
|
+
Tengine::Core::Event.where(event_type_name: n, source_name: s).count.should == 1
|
578
|
+
end
|
579
|
+
|
580
|
+
it "tenginedが調停する #2" do
|
581
|
+
@header.stub(:ack)
|
582
|
+
n = "alert.execution.job.tengine"
|
583
|
+
s = "test test"
|
584
|
+
e = Tengine::Event.new event_type_name: n, source_name: s, key: "k1"
|
585
|
+
f = Tengine::Event.new event_type_name: n, source_name: s, key: "k2"
|
586
|
+
g = Tengine::Event.new event_type_name: n, source_name: s, key: "k3"
|
587
|
+
|
588
|
+
@kernel.process_message @header, e.to_json
|
589
|
+
@kernel.process_message @header, f.to_json
|
590
|
+
@kernel.process_message @header, g.to_json
|
591
|
+
|
592
|
+
Tengine::Core::Event.where(event_type_name: n, source_name: s).count.should == 1
|
593
|
+
end
|
594
|
+
end
|
595
|
+
end
|
596
|
+
end
|
597
|
+
|
598
|
+
describe :setup_mq_connection do
|
599
|
+
if RUBY_VERSION >= "1.9.2"
|
600
|
+
before do
|
601
|
+
EM.instance_eval do
|
602
|
+
@timers.each {|i| EM.cancel_timer i } if @timers
|
603
|
+
@next_tick_queue = nil
|
604
|
+
end
|
605
|
+
trigger
|
606
|
+
config = Tengine::Core::Config::Core.new({
|
607
|
+
:tengined => {
|
608
|
+
:load_path => File.expand_path('../../../examples/uc01_execute_processing_for_event.rb', File.dirname(__FILE__)),
|
609
|
+
},
|
610
|
+
:event_queue => { :connection => { :port => @port } }
|
611
|
+
})
|
612
|
+
@kernel = Tengine::Core::Kernel.new(config)
|
613
|
+
end
|
614
|
+
|
615
|
+
let(:rabbitmq) do
|
616
|
+
ret = nil
|
617
|
+
ENV["PATH"].split(/:/).find do |dir|
|
618
|
+
Dir.glob("#{dir}/rabbitmq-server") do |path|
|
619
|
+
if File.executable?(path)
|
620
|
+
ret = path
|
621
|
+
break
|
622
|
+
end
|
623
|
+
end
|
624
|
+
end
|
625
|
+
|
626
|
+
pending "rabbitmq が見つかりません" unless ret
|
627
|
+
ret
|
628
|
+
end
|
629
|
+
|
630
|
+
def trigger port = rand(32768)
|
631
|
+
require 'tmpdir'
|
632
|
+
@dir = Dir.mktmpdir
|
633
|
+
# 指定したポートはもう使われているかもしれないので、その際は
|
634
|
+
# rabbitmqが起動に失敗するので、何回かポートを変えて試す。
|
635
|
+
n = 0
|
636
|
+
begin
|
637
|
+
envp = {
|
638
|
+
"RABBITMQ_NODENAME" => "rspec",
|
639
|
+
"RABBITMQ_NODE_PORT" => port.to_s,
|
640
|
+
"RABBITMQ_NODE_IP_ADDRESS" => "auto",
|
641
|
+
"RABBITMQ_MNESIA_BASE" => @dir.to_s,
|
642
|
+
"RABBITMQ_LOG_BASE" => @dir.to_s,
|
643
|
+
}
|
644
|
+
@pid = Process.spawn(envp, rabbitmq, :chdir => @dir, :in => :close, :out => '/dev/null', :err => '/dev/null')
|
645
|
+
x = Time.now
|
646
|
+
while Time.now < x + 64 do # まあこんくらい待てばいいでしょ
|
647
|
+
sleep 0.1
|
648
|
+
Process.waitpid2(@pid, Process::WNOHANG)
|
649
|
+
Process.kill 0, @pid
|
650
|
+
# netstat -an は Linux / BSD ともに有効
|
651
|
+
# どちらかに限ればもう少し効率的な探し方はある。たとえば Linux 限定でよければ netstat -lnt ...
|
652
|
+
y = `netstat -an | fgrep LISTEN | fgrep #{port}`
|
653
|
+
if y.lines.to_a.size > 1
|
654
|
+
@port = port
|
655
|
+
return
|
656
|
+
end
|
657
|
+
end
|
658
|
+
pending "failed to invoke rabbitmq in 16 secs."
|
659
|
+
rescue Errno::ECHILD, Errno::ESRCH
|
660
|
+
pending "10 attempts to invoke rabbitmq failed." if (n += 1) > 10
|
661
|
+
port = rand(32768)
|
662
|
+
retry
|
663
|
+
end
|
664
|
+
end
|
665
|
+
|
666
|
+
def finish
|
667
|
+
if @pid
|
668
|
+
begin
|
669
|
+
Process.kill "INT", @pid
|
670
|
+
Process.waitpid @pid
|
671
|
+
rescue Errno::ECHILD, Errno::ESRCH
|
672
|
+
ensure
|
673
|
+
require 'fileutils'
|
674
|
+
FileUtils.remove_entry_secure @dir, :force
|
675
|
+
end
|
676
|
+
end
|
677
|
+
end
|
678
|
+
|
679
|
+
after do
|
680
|
+
finish
|
681
|
+
end
|
682
|
+
|
683
|
+
it "MQ接続時にエラーなどのイベントハンドリングを行います" do
|
684
|
+
EM.run do
|
685
|
+
mq = @kernel.mq
|
686
|
+
|
687
|
+
@kernel.setup_mq_connection
|
688
|
+
|
689
|
+
# ここではイベント発生時の振る舞いもチェックします
|
690
|
+
@kernel.subscribe_queue do
|
691
|
+
Tengine::Core.stderr_logger.should_receive(:warn).with('mq.connection.on_tcp_connection_loss.').at_least(1).times
|
692
|
+
finish
|
693
|
+
EM.add_timer(1) do
|
694
|
+
Tengine::Core.stderr_logger.should_receive(:info).with('mq.connection.after_recovery: recovered successfully.')
|
695
|
+
EM.defer(
|
696
|
+
lambda { trigger @port; sleep 2; true },
|
697
|
+
lambda do |a|
|
698
|
+
Tengine::Core.stderr_logger.should_receive(:error).with('mq.channel.on_error channel_close: "channel close reason object"')
|
699
|
+
mq.channel.exec_callback_once_yielding_self(:error, "channel close reason object")
|
700
|
+
|
701
|
+
Tengine::Core.stderr_logger.should_receive(:error).with('mq.connection.on_error connection_close: "connection close reason object"')
|
702
|
+
mq.connection.exec_callback_yielding_self(:error, "connection close reason object")
|
703
|
+
|
704
|
+
EM.next_tick do
|
705
|
+
mq.stop
|
706
|
+
end
|
707
|
+
end
|
708
|
+
)
|
709
|
+
end
|
710
|
+
end
|
711
|
+
end
|
712
|
+
end
|
713
|
+
end
|
714
|
+
end
|
715
|
+
|
716
|
+
describe :status do
|
717
|
+
describe :starting do
|
718
|
+
before do
|
719
|
+
config = Tengine::Core::Config::Core.new({
|
720
|
+
:tengined => {
|
721
|
+
:load_path => File.expand_path('../../../examples/uc01_execute_processing_for_event.rb', File.dirname(__FILE__)),
|
722
|
+
},
|
723
|
+
})
|
724
|
+
@kernel = Tengine::Core::Kernel.new(config)
|
725
|
+
end
|
726
|
+
|
727
|
+
it "カーネルのインスタンス生成直後は「初期化済み」の状態を返す" do
|
728
|
+
@kernel.status.should == :initialized
|
729
|
+
end
|
730
|
+
|
731
|
+
it "起動要求を受け取った直後は「起動中」の状態を返す", :start => true do
|
732
|
+
@kernel.should_receive(:bind)
|
733
|
+
@kernel.should_receive(:activate)
|
734
|
+
|
735
|
+
@kernel.start
|
736
|
+
@kernel.status.should == :starting
|
737
|
+
end
|
738
|
+
|
739
|
+
it "内部で使用されるupdate_statusにおかしな値を入れるとArgumentError" do
|
740
|
+
expect {
|
741
|
+
@kernel.send(:update_status, :invalid_status)
|
742
|
+
}.to raise_error(ArgumentError)
|
743
|
+
end
|
744
|
+
end
|
745
|
+
|
746
|
+
describe :waiting_for_activation do
|
747
|
+
before do
|
748
|
+
config = Tengine::Core::Config::Core.new({
|
749
|
+
:tengined => {
|
750
|
+
:load_path => File.expand_path('../../../examples/uc01_execute_processing_for_event.rb', File.dirname(__FILE__)),
|
751
|
+
:wait_activation => true,
|
752
|
+
:activation_timeout => 3,
|
753
|
+
:activation_dir => File.expand_path('.', File.dirname(__FILE__)),
|
754
|
+
},
|
755
|
+
})
|
756
|
+
@kernel = Tengine::Core::Kernel.new(config)
|
757
|
+
end
|
758
|
+
|
759
|
+
it "起動処理が終了した直後に「稼働要求待ち」の状態を返す" do
|
760
|
+
@kernel.should_receive(:bind)
|
761
|
+
@kernel.should_receive(:wait_for_activation)
|
762
|
+
|
763
|
+
@kernel.start
|
764
|
+
@kernel.status.should == :waiting_activation
|
765
|
+
end
|
766
|
+
end
|
767
|
+
|
768
|
+
describe :running do
|
769
|
+
before do
|
770
|
+
config = Tengine::Core::Config::Core.new({
|
771
|
+
:tengined => {
|
772
|
+
:load_path => File.expand_path('../../../examples/uc01_execute_processing_for_event.rb', File.dirname(__FILE__)),
|
773
|
+
},
|
774
|
+
})
|
775
|
+
@kernel = Tengine::Core::Kernel.new(config)
|
776
|
+
@kernel.should_receive(:bind)
|
777
|
+
end
|
778
|
+
|
779
|
+
it "稼働要求を受け取った直後では「稼働中」の状態を返す" do
|
780
|
+
mock_mq = mock("mq")
|
781
|
+
mock_sender = mock("sender")
|
782
|
+
@kernel.stub(:mq).and_return(mock_mq)
|
783
|
+
@kernel.stub(:sender).and_return(mock_sender)
|
784
|
+
mock_sender.stub(:fire).and_yield
|
785
|
+
mock_mq.stub(:initiate_termination).and_yield
|
786
|
+
mock_mq.stub(:unsubscribe).and_yield
|
787
|
+
mock_mq.stub(:stop).and_yield
|
788
|
+
mock_mq.stub(:add_hook)
|
789
|
+
mock_mq.stub(:subscribe).with(nil).with(:ack => true, :nowait => false, :confirm => an_instance_of(Proc)) do |h, b|
|
790
|
+
h[:confirm].yield(mock("confirm-ok"))
|
791
|
+
end
|
792
|
+
|
793
|
+
@kernel.should_receive(:setup_mq_connection)
|
794
|
+
@kernel.start do
|
795
|
+
@kernel.status.should == :running
|
796
|
+
@kernel.stop
|
797
|
+
end
|
798
|
+
end
|
799
|
+
end
|
800
|
+
|
801
|
+
describe :stop do
|
802
|
+
it "停止要求を受け取った直後では「停止中」および「停止済」の状態を返す(稼働中)" do
|
803
|
+
config = Tengine::Core::Config::Core.new({
|
804
|
+
:tengined => {
|
805
|
+
:load_path => File.expand_path('../../../examples/uc01_execute_processing_for_event.rb', File.dirname(__FILE__)),
|
806
|
+
},
|
807
|
+
})
|
808
|
+
kernel = Tengine::Core::Kernel.new(config)
|
809
|
+
kernel.should_receive(:bind)
|
810
|
+
|
811
|
+
mock_mq = mock("mq")
|
812
|
+
mock_sender = mock("sender")
|
813
|
+
kernel.stub(:mq).and_return(mock_mq)
|
814
|
+
kernel.stub(:sender).and_return(mock_sender)
|
815
|
+
mock_sender.stub(:fire).and_yield
|
816
|
+
mock_mq.stub(:initiate_termination).and_yield
|
817
|
+
mock_mq.stub(:unsubscribe).and_yield
|
818
|
+
mock_mq.stub(:stop).and_yield
|
819
|
+
mock_mq.stub(:add_hook)
|
820
|
+
mock_mq.stub(:subscribe).with(nil).with(:ack => true, :nowait => false, :confirm => an_instance_of(Proc)) do |h, b|
|
821
|
+
h[:confirm].yield(mock("confirm-ok"))
|
822
|
+
end
|
823
|
+
|
824
|
+
kernel.start do
|
825
|
+
kernel.status.should == :running
|
826
|
+
|
827
|
+
kernel.stop do
|
828
|
+
kernel.status.should == :terminated
|
829
|
+
end
|
830
|
+
end
|
831
|
+
end
|
832
|
+
|
833
|
+
it "停止要求を受け取った直後では「停止中」および「停止済」の状態を返す(稼働要求待ち)" do
|
834
|
+
config = Tengine::Core::Config::Core.new({
|
835
|
+
:tengined => {
|
836
|
+
:load_path => File.expand_path('../../../examples/uc01_execute_processing_for_event.rb', File.dirname(__FILE__)),
|
837
|
+
:wait_activation => true,
|
838
|
+
:activation_timeout => 3,
|
839
|
+
:activation_dir => File.expand_path('.', File.dirname(__FILE__)),
|
840
|
+
},
|
841
|
+
})
|
842
|
+
kernel = Tengine::Core::Kernel.new(config)
|
843
|
+
kernel.should_receive(:bind)
|
844
|
+
|
845
|
+
lambda {
|
846
|
+
kernel.start
|
847
|
+
# kernel.stop
|
848
|
+
}.should raise_error(Tengine::Core::ActivationTimeoutError, "activation file found timeout error.")
|
849
|
+
kernel.status.should == :shutting_down
|
850
|
+
end
|
851
|
+
|
852
|
+
it "heartbeatの停止" do
|
853
|
+
kernel = Tengine::Core::Kernel.new(Tengine::Core::Config::Core.new({
|
854
|
+
:tengined => {
|
855
|
+
:load_path => File.expand_path('../../../examples/uc01_execute_processing_for_event.rb', File.dirname(__FILE__)),
|
856
|
+
:wait_activation => true,
|
857
|
+
:activation_timeout => 3,
|
858
|
+
:activation_dir => File.expand_path('.', File.dirname(__FILE__)),
|
859
|
+
},
|
860
|
+
}))
|
861
|
+
kernel.instance_eval do
|
862
|
+
@status = :running
|
863
|
+
@heartbeat_timer = true
|
864
|
+
end
|
865
|
+
mock_mq = mock("mq")
|
866
|
+
kernel.stub(:mq).and_return(mock_mq)
|
867
|
+
mock_mq.stub(:initiate_termination).and_yield
|
868
|
+
mock_mq.stub(:unsubscribe).and_yield
|
869
|
+
mock_mq.stub(:stop).and_yield
|
870
|
+
mock_mq.stub(:add_hook)
|
871
|
+
mock_mq.stub(:subscribe).with(nil).with(:ack => true, :nowait => false, :confirm => an_instance_of(Proc)) do |h, b|
|
872
|
+
h[:confirm].yield(mock("confirm-ok"))
|
873
|
+
end
|
874
|
+
sender = mock(:sender)
|
875
|
+
kernel.stub(:sender).and_return(sender)
|
876
|
+
|
877
|
+
EM.should_receive(:cancel_timer)
|
878
|
+
sender.should_receive(:fire).with("finished.process.core.tengine", an_instance_of(Hash))
|
879
|
+
|
880
|
+
kernel.stop
|
881
|
+
end
|
882
|
+
end
|
883
|
+
end
|
884
|
+
|
885
|
+
end
|