From f5019cd462723d2eb470333f2d55bb4cfac31fd5 Mon Sep 17 00:00:00 2001
From: Dawn Richardson <dawn.richardson@abletech.co.nz>
Date: Tue, 22 Dec 2015 11:36:45 +1300
Subject: [PATCH] Add optional group parameter when setting a push audience
 with tags

---
 lib/urbanairship/push/audience.rb           | 6 ++++--
 spec/lib/urbanairship/push/audience_spec.rb | 7 ++++++-
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/lib/urbanairship/push/audience.rb b/lib/urbanairship/push/audience.rb
index 513cd5f3..188cbfcc 100644
--- a/lib/urbanairship/push/audience.rb
+++ b/lib/urbanairship/push/audience.rb
@@ -37,8 +37,10 @@ def device_pin(pin)
       end
 
       # Select a single tag
-      def tag(tag)
-        { tag: tag }
+      def tag(tag, group: nil)
+        tag_params = { tag: tag }
+        tag_params[:group] = group unless group.nil?
+        tag_params
       end
 
       # Select a single alias
diff --git a/spec/lib/urbanairship/push/audience_spec.rb b/spec/lib/urbanairship/push/audience_spec.rb
index e8923942..dfb27ffa 100644
--- a/spec/lib/urbanairship/push/audience_spec.rb
+++ b/spec/lib/urbanairship/push/audience_spec.rb
@@ -57,6 +57,11 @@
         'test',
         { tag: 'test' }
       ],
+      [
+        :tag,
+        ['test', group: 'test-group'],
+        { tag: 'test', group: 'test-group' }
+      ],
       [
         :alias,
         'test',
@@ -69,7 +74,7 @@
       ]
     ].each do |selector, value, expected_result|
       it "can filter for '#{selector}'" do
-        actual_payload = UA.send(selector, value)
+        actual_payload = UA.send(selector, *value)
         expect(actual_payload).to eq expected_result
       end
     end