Merge pull request #123 from yossi-eynav/add_method_to_set_users

Add a method for setting an array of users

lib/rollout.rb

@@ -203,6 +203,13 @@ def deactivate_users(feature, users)
     end
   end
 
+  def set_users(feature, users)
+    with_feature(feature) do |f|
+      f.users = []
+      users.each{|user| f.add_user(user)}
+    end
+  end
+
   def define_group(group, &block)
     @groups[group.to_sym] = block
   end

spec/rollout_spec.rb

@@ -226,6 +226,16 @@
     end
   end
 
+
+  describe 'set a group of users' do
+    it 'should replace the users with the given array' do
+      users = %w(1 2 3 4)
+      @rollout.activate_users(:chat, %w(10 20 30))
+      @rollout.set_users(:chat, users)
+      expect(@rollout.get(:chat).users).to eq(users)
+    end
+  end
+
   describe "activating a feature globally" do
     before do
       @rollout.activate(:chat)