Added support for localOrShuffleGrouping

This commit is contained in:
David Tollmyr 2012-06-07 10:06:39 +02:00
parent 9c923b819e
commit b72e444360
2 changed files with 14 additions and 0 deletions

View File

@ -56,6 +56,8 @@ module RedStorm
declarer.globalGrouping(source_id)
when :shuffle
declarer.shuffleGrouping(source_id)
when :local_or_shuffle
declarer.localOrShuffleGrouping(source_id)
when :none
declarer.noneGrouping(source_id)
when :all

View File

@ -384,6 +384,18 @@ describe RedStorm::SimpleTopology do
Topology1.new.start("base_path", :cluster)
end
it "should support local_or_shuffle" do
class Topology1 < RedStorm::SimpleTopology
spout SpoutClass1, :id => 1
bolt BoltClass1 do
source 1, :local_or_shuffle
end
end
@declarer.should_receive("localOrShuffleGrouping").with('1')
Topology1.new.start("base_path", :cluster)
end
it "should support none" do
class Topology1 < RedStorm::SimpleTopology
spout SpoutClass1, :id => 1