Adds methods for setting/retrieving a users badge

Author: cakejelly

README.md

@@ -365,6 +365,22 @@ user.destroy_identity
 # => true
 ```
 
+#### Setting a Users Badge
+
+```ruby
+user = platform.users.find("user_id")
+user.set_badge(10)
+# => true
+```
+
+#### Retrieving a Users Badge
+
+```ruby
+user = platform.users.find("user_id")
+user.badge
+# => { "external_unread_count" => 13, "unread_conversation_count" => 10, "unread_message_count" => 50 }
+```
+
 
 #### Generating Identity Tokens ####
 See: [the official authentication guide](https://developer.layer.com/docs/android/guides#authentication)

lib/layer/resources/user.rb

@@ -41,6 +41,15 @@ def destroy_identity
         client.delete(identity_url)
       end
 
+      def set_badge(badge_count)
+        body = { external_unread_count: badge_count }
+        client.put("#{url}/badge", body: body.to_json)
+      end
+
+      def badge
+        client.get("#{url}/badge")
+      end
+
       private
 
       def identity_url

spec/layer/resources/user_spec.rb

@@ -121,4 +121,32 @@
       user.destroy_identity
     end
   end
+
+  describe "#set_badge" do
+    let(:badge_count) { { external_unread_count: 5 } }
+
+    it "should set the badge count for a user" do
+      allow(http_client).to receive(:put).and_return(nil)
+      expect(http_client).to receive(:put).with(
+                               "users/#{user.id}/badge",
+                               body: badge_count.to_json
+                             )
+      user.set_badge(5)
+    end
+  end
+
+  describe "#badge" do
+    before do
+      allow(http_client).to receive(:get).and_return(user_badge)
+      expect(http_client).to receive(:get).with("users/#{user.id}/badge")
+    end
+
+    it "should retrieve badge for a user" do
+      user.badge
+    end
+
+    it "should return hash of badge values" do
+      expect(user.badge).to be_instance_of(Hash)
+    end
+  end
 end

spec/layer_helper.rb

@@ -91,4 +91,12 @@ def user_identity_operations
       {operation: "set", property: "metadata.level", value: "2"}
     ]
   end
+
+  def user_badge
+    {
+      external_unread_count: 13,
+      unread_conversation_count: 10,
+      unread_message_count: 50
+    }
+  end
 end