DRY up tests.

This commit is contained in:
Guo Xiang Tan 2016-04-07 15:43:33 +08:00
parent 7f6fe8fe13
commit 48e118a054
1 changed files with 4 additions and 2 deletions

View File

@ -83,8 +83,10 @@ describe DiscourseRedis do
begin
fallback_handler.master = false
Redis::Client.any_instance.expects(:call).with([:info]).returns(DiscourseRedis::FallbackHandler::MASTER_LINK_STATUS)
Redis::Client.any_instance.expects(:call).with([:client, [:kill, 'type', 'normal']])
Redis::Client.any_instance.expects(:call).with([:client, [:kill, 'type', 'pubsub']])
%w{normal pubsub}.each do |connection_type|
Redis::Client.any_instance.expects(:call).with([:client, [:kill, 'type', connection_type]])
end
fallback_handler.initiate_fallback_to_master