fcm 1.0.5 → 2.0.1

data/spec/fcm_spec.rb

@@ -1,501 +1,438 @@
 require "spec_helper"
 
 describe FCM do
-  let(:send_url) { "#{FCM::BASE_URI}/fcm/send" }
-  let(:group_notification_base_uri) { "#{FCM::GROUP_NOTIFICATION_BASE_URI}/gcm/notification" }
-  let(:api_key) { "AIzaSyB-1uEai2WiUapxCs2Q0GZYzPu7Udno5aA" }
-  let(:registration_id) { "42" }
-  let(:registration_ids) { ["42"] }
-  let(:key_name) { "appUser-Chris" }
-  let(:project_id) { "123456789" } # https://developers.google.com/cloud-messaging/gcm#senderid
-  let(:notification_key) { "APA91bGHXQBB...9QgnYOEURwm0I3lmyqzk2TXQ" }
-  let(:valid_topic) { "TopicA" }
-  let(:invalid_topic) { "TopicA$" }
-  let(:valid_condition) { "'TopicA' in topics && ('TopicB' in topics || 'TopicC' in topics)" }
-  let(:invalid_condition) { "'TopicA' in topics and some other text ('TopicB' in topics || 'TopicC' in topics)" }
-  let(:invalid_condition_topic) { "'TopicA$' in topics" }
-
-  it "should raise an error if the api key is not provided" do
-    expect { FCM.new }.to raise_error(ArgumentError)
+  let(:project_name) { 'test-project' }
+  let(:json_key_path) { 'path/to/json/key.json' }
+  let(:client) { FCM.new(json_key_path) }
+
+  let(:mock_token) { "access_token" }
+  let(:mock_headers) do
+    {
+      "Content-Type" => "application/json",
+      "Authorization" => "Bearer #{mock_token}",
+    }
   end
 
-  it "should raise error if time_to_live is given" do
-    # ref: https://firebase.google.com/docs/cloud-messaging/http-server-ref#ttl
+  before do
+    allow(client).to receive(:json_key)
+
+    # Mock the Google::Auth::ServiceAccountCredentials
+    allow(Google::Auth::ServiceAccountCredentials).to receive(:make_creds).
+      and_return(double(fetch_access_token!: { 'access_token' => mock_token }))
   end
 
-  describe "#send_v1" do
-    pending "should send message"
+  it "should initialize" do
+    expect { client }.not_to raise_error
   end
 
-  describe "sending notification" do
-    let(:valid_request_body) do
-      { registration_ids: registration_ids }
-    end
-    let(:valid_request_body_with_string) do
-      { registration_ids: registration_id }
+  describe "credentials path" do
+    it "can be a path to a file" do
+      fcm = FCM.new("README.md")
+      expect(fcm.__send__(:json_key).class).to eq(File)
     end
-    let(:valid_request_headers) do
-      {
-        "Content-Type" => "application/json",
-        "Authorization" => "key=#{api_key}",
-      }
+
+    it "can be an IO object" do
+      fcm = FCM.new(StringIO.new("hey"))
+      expect(fcm.__send__(:json_key).class).to eq(StringIO)
     end
+  end
+
+  describe "#send_v1 or #send_notification_v1" do
+    let(:client) { FCM.new(json_key_path, project_name) }
 
-    let(:stub_fcm_send_request) do
-      stub_request(:post, send_url).with(
-        body: valid_request_body.to_json,
-        headers: valid_request_headers,
+    let(:uri) { "#{FCM::BASE_URI_V1}#{project_name}/messages:send" }
+    let(:status_code) { 200 }
+
+    let(:stub_fcm_send_v1_request) do
+      stub_request(:post, uri).with(
+        body: { 'message' => send_v1_params }.to_json,
+        headers: mock_headers
       ).to_return(
         # ref: https://firebase.google.com/docs/cloud-messaging/http-server-ref#interpret-downstream
         body: "{}",
         headers: {},
-        status: 200,
+        status: status_code,
       )
     end
 
-    let(:stub_fcm_send_request_with_string) do
-      stub_request(:post, send_url).with(
-        body: valid_request_body_with_string.to_json,
-        headers: valid_request_headers,
-      ).to_return(
-        body: "{}",
-        headers: {},
-        status: 200,
-      )
+    before do
+      stub_fcm_send_v1_request
     end
 
-    let(:stub_fcm_send_request_with_basic_auth) do
-      uri = URI.parse(send_url)
-      uri.user = "a"
-      uri.password = "b"
-      stub_request(:post, uri.to_s).to_return(body: "{}", headers: {}, status: 200)
+    shared_examples 'succesfuly send notification' do
+      it 'should send notification of HTTP V1 using POST to FCM server' do
+        client.send_v1(send_v1_params).should eq(
+          response: 'success', body: '{}', headers: {}, status_code: 200
+        )
+        stub_fcm_send_v1_request.should have_been_made.times(1)
+      end
     end
 
-    before(:each) do
-      stub_fcm_send_request
-      stub_fcm_send_request_with_string
-      stub_fcm_send_request_with_basic_auth
-    end
+    describe 'send to token' do
+      let(:token) { '4sdsx' }
+      let(:send_v1_params) do
+        {
+          'token' => token,
+          'notification' => {
+            'title' => 'Breaking News',
+            'body' => 'New news story available.'
+          },
+          'data' => {
+            'story_id' => 'story_12345'
+          },
+          'android' => {
+            'notification' => {
+              'click_action' => 'TOP_STORY_ACTIVITY',
+              'body' => 'Check out the Top Story'
+            }
+          },
+          'apns' => {
+            'payload' => {
+              'aps' => {
+                'category' => 'NEW_MESSAGE_CATEGORY'
+              }
+            }
+          }
+        }
+      end
 
-    it "should send notification using POST to FCM server" do
-      fcm = FCM.new(api_key)
-      fcm.send(registration_ids).should eq(response: "success", body: "{}", headers: {}, status_code: 200, canonical_ids: [], not_registered_ids: [])
-      stub_fcm_send_request.should have_been_made.times(1)
-    end
+      include_examples 'succesfuly send notification'
 
-    it "should send notification using POST to FCM if id provided as string" do
-      fcm = FCM.new(api_key)
-      fcm.send(registration_id).should eq(response: "success", body: "{}", headers: {}, status_code: 200, canonical_ids: [], not_registered_ids: [])
-      stub_fcm_send_request.should have_been_made.times(1)
+      it 'includes all the response' do
+        response = client.send_v1(send_v1_params)
+        expect(response[:status_code]).to eq(status_code)
+        expect(response[:response]).to eq('success')
+        expect(response[:body]).to eq('{}')
+        expect(response[:headers]).to eq({})
+        expect(response[:canonical_ids]).to be_nil
+        expect(response[:not_registered_ids]).to be_nil
+      end
     end
 
-    context "send notification with data" do
-      let!(:stub_with_data) do
-        stub_request(:post, send_url)
-          .with(body: '{"registration_ids":["42"],"data":{"score":"5x1","time":"15:10"}}',
-                headers: valid_request_headers)
-          .to_return(status: 200, body: "", headers: {})
-      end
-      before do
-      end
-      it "should send the data in a post request to fcm" do
-        fcm = FCM.new(api_key)
-        fcm.send(registration_ids, data: { score: "5x1", time: "15:10" })
-        stub_with_data.should have_been_requested
+    describe 'send to multiple tokens' do
+      let(:tokens) { ['4sdsx', '4sdsy'] }
+      let(:send_v1_params) do
+        {
+          'token' => tokens,
+          'notification' => {
+            'title' => 'Breaking News',
+            'body' => 'New news story available.'
+          }
+        }
       end
+
+      include_examples 'succesfuly send notification'
     end
 
-    context "sending notification to a topic" do
-      let!(:stub_with_valid_topic) do
-        stub_request(:post, send_url)
-          .with(body: '{"to":"/topics/TopicA","data":{"score":"5x1","time":"15:10"}}',
-                headers: valid_request_headers)
-          .to_return(status: 200, body: "", headers: {})
-      end
-      let!(:stub_with_invalid_topic) do
-        stub_request(:post, send_url)
-          .with(body: '{"condition":"/topics/TopicA$","data":{"score":"5x1","time":"15:10"}}',
-                headers: valid_request_headers)
-          .to_return(status: 200, body: "", headers: {})
+    describe 'send to topic' do
+      let(:topic) { 'news' }
+      let(:send_v1_params) do
+        {
+          'topic' => topic,
+          'notification' => {
+            'title' => 'Breaking News',
+            'body' => 'New news story available.'
+          }
+        }
       end
 
-      describe "#send_to_topic" do
-        it "should send the data in a post request to fcm" do
-          fcm = FCM.new(api_key)
-          fcm.send_to_topic(valid_topic, data: { score: "5x1", time: "15:10" })
-          stub_with_valid_topic.should have_been_requested
-        end
+      include_examples 'succesfuly send notification'
 
-        it "should not send to invalid topics" do
-          fcm = FCM.new(api_key)
-          fcm.send_to_topic(invalid_topic, data: { score: "5x1", time: "15:10" })
-          stub_with_invalid_topic.should_not have_been_requested
+      context 'when topic is invalid' do
+        let(:topic) { '/topics/news$' }
+
+        it 'should raise error' do
+          stub_fcm_send_v1_request.should_not have_been_requested
         end
       end
     end
 
-    context "sending notification to a topic condition" do
-      let!(:stub_with_valid_condition) do
-        stub_request(:post, send_url)
-          .with(body: '{"condition":"\'TopicA\' in topics && (\'TopicB\' in topics || \'TopicC\' in topics)","data":{"score":"5x1","time":"15:10"}}',
-                headers: valid_request_headers)
-          .to_return(status: 200, body: "", headers: {})
-      end
-      let!(:stub_with_invalid_condition) do
-        stub_request(:post, send_url)
-          .with(body: '{"condition":"\'TopicA\' in topics and some other text (\'TopicB\' in topics || \'TopicC\' in topics)","data":{"score":"5x1","time":"15:10"}}',
-                headers: valid_request_headers)
-          .to_return(status: 200, body: "", headers: {})
+    describe 'send to condition' do
+      let(:condition) { "'foo' in topics" }
+      let(:send_v1_params) do
+        {
+          'condition' => condition,
+          'notification' => {
+            'title' => 'Breaking News',
+            'body' => 'New news story available.'
+          },
+        }
       end
-      let!(:stub_with_invalid_condition_topic) do
-        stub_request(:post, send_url)
-          .with(body: '{"condition":"\'TopicA$\' in topics","data":{"score":"5x1","time":"15:10"}}',
-                headers: valid_request_headers)
-          .to_return(status: 200, body: "", headers: {})
+
+      include_examples 'succesfuly send notification'
+    end
+
+    describe 'send to notification_key' do
+      let(:notification_key) { 'notification_key' }
+      let(:send_v1_params) do
+        {
+          'notification_key' => notification_key,
+          'notification' => {
+            'title' => 'Breaking News',
+            'body' => 'New news story available.'
+          }
+        }
       end
 
-      describe "#send_to_topic_condition" do
-        it "should send the data in a post request to fcm" do
-          fcm = FCM.new(api_key)
-          fcm.send_to_topic_condition(valid_condition, data: { score: "5x1", time: "15:10" })
-          stub_with_valid_condition.should have_been_requested
-        end
+      include_examples 'succesfuly send notification'
+    end
 
-        it "should not send to invalid conditions" do
-          fcm = FCM.new(api_key)
-          fcm.send_to_topic_condition(invalid_condition, data: { score: "5x1", time: "15:10" })
-          stub_with_invalid_condition.should_not have_been_requested
-        end
+    context 'when project_name is empty' do
+      let(:project_name) { '' }
+      let(:send_v1_params) do
+        {
+          'token' => '4sdsx',
+          'notification' => {
+            'title' => 'Breaking News',
+            'body' => 'New news story available.'
+          }
+        }
+      end
 
-        it "should not send to invalid topics in a condition" do
-          fcm = FCM.new(api_key)
-          fcm.send_to_topic_condition(invalid_condition_topic, data: { score: "5x1", time: "15:10" })
-          stub_with_invalid_condition_topic.should_not have_been_requested
-        end
+      it 'should not send notification' do
+        client.send_v1(send_v1_params)
+        stub_fcm_send_v1_request.should_not have_been_requested
       end
     end
 
-    context "when send_notification responds with failure" do
-      let(:mock_request_attributes) do
+    describe 'error handling' do
+      let(:send_v1_params) do
         {
-          body: valid_request_body.to_json,
-          headers: valid_request_headers,
+          'token' => '4sdsx',
+          'notification' => {
+            'title' => 'Breaking News',
+            'body' => 'New news story available.'
+          }
         }
       end
 
-      subject { FCM.new(api_key) }
-
-      context "on failure code 400" do
-        before do
-          stub_request(:post, send_url).with(
-            mock_request_attributes
-          ).to_return(
-            # ref: https://firebase.google.com/docs/cloud-messaging/http-server-ref#interpret-downstream
-            body: "{}",
-            headers: {},
-            status: 400,
-          )
-        end
-        it "should not send notification due to 400" do
-          subject.send(registration_ids).should eq(body: "{}",
-                                                   headers: {},
-                                                   response: "Only applies for JSON requests. Indicates that the request could not be parsed as JSON, or it contained invalid fields.",
-                                                   status_code: 400)
+      context 'when status_code is 400' do
+        let(:status_code) { 400 }
+
+        it 'should raise error' do
+          response = client.send_v1(send_v1_params)
+          expect(response[:status_code]).to eq(status_code)
+          expect(response[:response]).to include('Only applies for JSON requests')
         end
       end
 
-      context "on failure code 401" do
-        before do
-          stub_request(:post, send_url).with(
-            mock_request_attributes
-          ).to_return(
-            # ref: https://firebase.google.com/docs/cloud-messaging/http-server-ref#interpret-downstream
-            body: "{}",
-            headers: {},
-            status: 401,
-          )
-        end
+      context 'when status_code is 401' do
+        let(:status_code) { 401 }
 
-        it "should not send notification due to 401" do
-          subject.send(registration_ids).should eq(body: "{}",
-                                                   headers: {},
-                                                   response: "There was an error authenticating the sender account.",
-                                                   status_code: 401)
+        it 'should raise error' do
+          response = client.send_v1(send_v1_params)
+          expect(response[:status_code]).to eq(status_code)
+          expect(response[:response]).to include('There was an error authenticating')
         end
       end
 
-      context "on failure code 503" do
-        before do
-          stub_request(:post, send_url).with(
-            mock_request_attributes
-          ).to_return(
-            # ref: https://firebase.google.com/docs/cloud-messaging/http-server-ref#interpret-downstream
-            body: "{}",
-            headers: {},
-            status: 503,
-          )
-        end
+      context 'when status_code is 500' do
+        let(:status_code) { 500 }
 
-        it "should not send notification due to 503" do
-          subject.send(registration_ids).should eq(body: "{}",
-                                                   headers: {},
-                                                   response: "Server is temporarily unavailable.",
-                                                   status_code: 503)
+        it 'should raise error' do
+          response = client.send_v1(send_v1_params)
+          expect(response[:status_code]).to eq(status_code)
+          expect(response[:response]).to include('There was an internal error')
         end
       end
 
-      context "on failure code 5xx" do
-        before do
-          stub_request(:post, send_url).with(
-            mock_request_attributes
-          ).to_return(
-            # ref: https://firebase.google.com/docs/cloud-messaging/http-server-ref#interpret-downstream
-            body: '{"body-key" => "Body value"}',
-            headers: { "header-key" => "Header value" },
-            status: 599,
-          )
-        end
+      context 'when status_code is 503' do
+        let(:status_code) { 503 }
 
-        it "should not send notification due to 599" do
-          subject.send(registration_ids).should eq(body: '{"body-key" => "Body value"}',
-                                                   headers: { "header-key" => "Header value" },
-                                                   response: "There was an internal error in the FCM server while trying to process the request.",
-                                                   status_code: 599)
+        it 'should raise error' do
+          response = client.send_v1(send_v1_params)
+          expect(response[:status_code]).to eq(status_code)
+          expect(response[:response]).to include('Server is temporarily unavailable')
         end
       end
     end
+  end
 
-    context "when send_notification responds canonical_ids" do
-      let(:mock_request_attributes) do
-        {
-          body: valid_request_body.to_json,
-          headers: valid_request_headers,
-        }
-      end
+  describe '#send_to_topic' do
+    let(:client) { FCM.new(json_key_path, project_name) }
 
-      let(:valid_response_body_with_canonical_ids) do
-        {
-          failure: 0, canonical_ids: 1, results: [{ registration_id: "43", message_id: "0:1385025861956342%572c22801bb3" }],
-        }
-      end
+    let(:uri) { "#{FCM::BASE_URI_V1}#{project_name}/messages:send" }
 
-      subject { FCM.new(api_key) }
+    let(:topic) { 'news' }
+    let(:params) do
+      {
+        'topic' => topic
+      }.merge(options)
+    end
+    let(:options) do
+      {
+        'data' => {
+          'story_id' => 'story_12345'
+        }
+      }
+    end
 
-      before do
-        stub_request(:post, send_url).with(
-          mock_request_attributes
-        ).to_return(
-          # ref: https://firebase.google.com/docs/cloud-messaging/http-server-ref#interpret-downstream
-          body: valid_response_body_with_canonical_ids.to_json,
-          headers: {},
-          status: 200,
-        )
-      end
+    let(:stub_fcm_send_to_topic_request) do
+      stub_request(:post, uri).with(
+        body: { 'message' => params }.to_json,
+        headers: mock_headers
+      ).to_return(
+        body: "{}",
+        headers: {},
+        status: 200,
+      )
+    end
 
-      it "should contain canonical_ids" do
-        response = subject.send(registration_ids)
+    before do
+      stub_fcm_send_to_topic_request
+    end
 
-        response.should eq(headers: {},
-                           canonical_ids: [{ old: "42", new: "43" }],
-                           not_registered_ids: [],
-                           status_code: 200,
-                           response: "success",
-                           body: '{"failure":0,"canonical_ids":1,"results":[{"registration_id":"43","message_id":"0:1385025861956342%572c22801bb3"}]}')
-      end
+    it 'should send notification to topic using POST to FCM server' do
+      client.send_to_topic(topic, options).should eq(
+        response: 'success', body: '{}', headers: {}, status_code: 200
+      )
+      stub_fcm_send_to_topic_request.should have_been_made.times(1)
     end
 
-    context "when send_notification responds with NotRegistered" do
-      subject { FCM.new(api_key) }
+    context 'when topic is invalid' do
+      let(:topic) { '/topics/news$' }
 
-      let(:mock_request_attributes) do
-        {
-          body: valid_request_body.to_json,
-          headers: valid_request_headers,
-        }
+      it 'should raise error' do
+        client.send_to_topic(topic, options)
+        stub_fcm_send_to_topic_request.should_not have_been_requested
       end
+    end
+  end
 
-      let(:valid_response_body_with_not_registered_ids) do
-        {
-          canonical_ids: 0, failure: 1, results: [{ error: "NotRegistered" }],
-        }
-      end
+  describe "#send_to_topic_condition" do
+    let(:client) { FCM.new(json_key_path, project_name) }
 
-      before do
-        stub_request(:post, send_url).with(
-          mock_request_attributes
-        ).to_return(
-          body: valid_response_body_with_not_registered_ids.to_json,
-          headers: {},
-          status: 200,
-        )
-      end
+    let(:uri) { "#{FCM::BASE_URI_V1}#{project_name}/messages:send" }
 
-      it "should contain not_registered_ids" do
-        response = subject.send(registration_ids)
-        response.should eq(
-          headers: {},
-          canonical_ids: [],
-          not_registered_ids: registration_ids,
-          status_code: 200,
-          response: "success",
-          body: '{"canonical_ids":0,"failure":1,"results":[{"error":"NotRegistered"}]}',
-        )
-      end
+    let(:topic_condition) { "'foo' in topics" }
+    let(:params) do
+      {
+        'condition' => topic_condition
+      }.merge(options)
     end
-  end
-
-  describe "sending group notifications" do
-    # TODO: refactor to should_behave_like
-    let(:valid_request_headers) do
+    let(:options) do
       {
-        "Authorization" => "key=#{api_key}",
-        "Content-Type" => "application/json",
-        "Project-Id" => project_id,
+        'data' => {
+          'story_id' => 'story_12345'
+        }
       }
     end
-    let(:valid_response_body) do
-      { notification_key: "APA91bGHXQBB...9QgnYOEURwm0I3lmyqzk2TXQ" }
+
+    let(:stub_fcm_send_to_topic_condition_request) do
+      stub_request(:post, uri).with(
+        body: { 'message' => params }.to_json,
+        headers: mock_headers
+      ).to_return(
+        body: "{}",
+        headers: {},
+        status: 200,
+      )
     end
 
-    let(:default_valid_request_body) do
-      {
-        registration_ids: registration_ids,
-        operation: "create",
-        notification_key_name: key_name,
-      }
+    before do
+      stub_fcm_send_to_topic_condition_request
     end
 
-    subject { FCM.new(api_key) }
+    it 'should send notification to topic_condition using POST to FCM server' do
+      client.send_to_topic_condition(topic_condition, options).should eq(
+        response: 'success', body: '{}', headers: {}, status_code: 200
+      )
+      stub_fcm_send_to_topic_condition_request.should have_been_made.times(1)
+    end
 
-    # ref: https://firebase.google.com/docs/cloud-messaging/notifications#managing-device-groups-on-the-app-server
-    context "create" do
-      let(:valid_request_body) do
-        default_valid_request_body.merge({
-          operation: "create",
-        })
-      end
+    context 'when topic_condition is invalid' do
+      let(:topic_condition) { "'foo' in topics$" }
 
-      let(:mock_request_attributes) do
-        {
-          body: valid_request_body.to_json,
-          headers: valid_request_headers,
-        }
+      it 'should raise error' do
+        client.send_to_topic_condition(topic_condition, options)
+        stub_fcm_send_to_topic_condition_request.should_not have_been_requested
       end
+    end
+  end
 
-      before do
-        stub_request(:post, group_notification_base_uri).with(
-          mock_request_attributes
-        ).to_return(
-          body: valid_response_body.to_json,
-          headers: {},
-          status: 200,
-        )
-      end
+  describe "#get_instance_id_info" do
+    subject(:get_info) { client.get_instance_id_info(registration_token, options) }
 
-      it "should send a post request" do
-        response = subject.create(key_name, project_id, registration_ids)
-        response.should eq(
-          headers: {},
-          status_code: 200,
-          response: "success",
-          body: valid_response_body.to_json,
-        )
-      end
-    end # create context
-
-    context "add" do
-      let(:valid_request_body) do
-        default_valid_request_body.merge({
-          operation: "add",
-          notification_key: notification_key,
-        })
-      end
+    let(:options) { nil }
+    let(:base_uri) { "#{FCM::INSTANCE_ID_API}/iid/info" }
+    let(:uri) { "#{base_uri}/#{registration_token}" }
+    let(:registration_token) { "42" }
 
-      let(:mock_request_attributes) do
-        {
-          body: valid_request_body.to_json,
-          headers: valid_request_headers,
-        }
+    context 'without options' do
+      it 'calls info endpoint' do
+        endpoint = stub_request(:get, uri).with(headers: mock_headers)
+        get_info
+        expect(endpoint).to have_been_requested
       end
+    end
 
-      before do
-        stub_request(:post, group_notification_base_uri).with(
-          mock_request_attributes
-        ).to_return(
-          body: valid_response_body.to_json,
-          headers: {},
-          status: 200,
-        )
-      end
+    context 'with detail option' do
+      let(:uri) { "#{base_uri}/#{registration_token}?details=true" }
+      let(:options) { { details: true } }
 
-      it "should send a post request" do
-        response = subject.add(key_name, project_id, notification_key, registration_ids)
-        response.should eq(
-          headers: {},
-          status_code: 200,
-          response: "success",
-          body: valid_response_body.to_json,
-        )
-      end
-    end # add context
-
-    context "remove" do
-      let(:valid_request_body) do
-        default_valid_request_body.merge({
-          operation: "remove",
-          notification_key: notification_key,
-        })
+      it 'calls info endpoint' do
+        endpoint = stub_request(:get, uri).with(headers: mock_headers)
+        get_info
+        expect(endpoint).to have_been_requested
       end
+    end
+  end
 
-      let(:mock_request_attributes) do
-        {
-          body: valid_request_body.to_json,
-          headers: valid_request_headers,
-        }
-      end
+  describe "topic subscriptions" do
+    let(:topic) { 'news' }
+    let(:registration_token) { "42" }
+    let(:registration_token_2) { "43" }
+    let(:registration_tokens) { [registration_token, registration_token_2] }
 
-      before do
-        stub_request(:post, group_notification_base_uri).with(
-          mock_request_attributes
-        ).to_return(
-          body: valid_response_body.to_json,
-          headers: {},
-          status: 200,
-        )
+    describe "#topic_subscription" do
+      subject(:subscribe) { client.topic_subscription(topic, registration_token) }
+
+      let(:uri) { "#{FCM::INSTANCE_ID_API}/iid/v1/#{registration_token}/rel/topics/#{topic}" }
+
+      it 'subscribes to a topic' do
+        endpoint = stub_request(:post, uri).with(headers: mock_headers)
+        subscribe
+        expect(endpoint).to have_been_requested
       end
+    end
 
-      it "should send a post request" do
-        response = subject.remove(key_name, project_id, notification_key, registration_ids)
-        response.should eq(
-          headers: {},
-          status_code: 200,
-          response: "success",
-          body: valid_response_body.to_json,
-        )
+    describe "#topic_unsubscription" do
+      subject(:unsubscribe) { client.topic_unsubscription(topic, registration_token) }
+
+      let(:uri) { "#{FCM::INSTANCE_ID_API}/iid/v1:batchRemove" }
+      let(:params) { { to: "/topics/#{topic}", registration_tokens: [registration_token] } }
+
+      it 'unsubscribes from a topic' do
+        endpoint = stub_request(:post, uri).with(body: params.to_json, headers: mock_headers)
+        unsubscribe
+        expect(endpoint).to have_been_requested
       end
-    end # remove context
-  end
+    end
 
-  describe "#recover_notification_key" do
-    it "sends a 'retrieve notification key' request" do
-      uri = "#{FCM::GROUP_NOTIFICATION_BASE_URI}/gcm/notification"
-      endpoint = stub_request(:get, uri).with(
-        headers: {
-          "Content-Type" => "application/json",
-          "Authorization" => "key=TEST_SERVER_KEY",
-          "project_id" => "TEST_PROJECT_ID",
-        },
-        query: { notification_key_name: "TEST_KEY_NAME" },
-      )
-      client = FCM.new("TEST_SERVER_KEY")
+    describe "#batch_topic_subscription" do
+      subject(:batch_subscribe) { client.batch_topic_subscription(topic, registration_tokens) }
 
-      client.recover_notification_key("TEST_KEY_NAME", "TEST_PROJECT_ID")
+      let(:uri) { "#{FCM::INSTANCE_ID_API}/iid/v1:batchAdd" }
+      let(:params) { { to: "/topics/#{topic}", registration_tokens: registration_tokens } }
 
-      expect(endpoint).to have_been_requested
+      it 'subscribes to a topic' do
+        endpoint = stub_request(:post, uri).with(body: params.to_json, headers: mock_headers)
+        batch_subscribe
+        expect(endpoint).to have_been_requested
+      end
     end
-  end
 
-  describe "subscribing to a topic" do
-    # TODO
+    describe "#batch_topic_unsubscription" do
+      subject(:batch_unsubscribe) { client.batch_topic_unsubscription(topic, registration_tokens) }
+
+      let(:uri) { "#{FCM::INSTANCE_ID_API}/iid/v1:batchRemove" }
+      let(:params) { { to: "/topics/#{topic}", registration_tokens: registration_tokens } }
+
+      it 'unsubscribes from a topic' do
+        endpoint = stub_request(:post, uri).with(body: params.to_json, headers: mock_headers)
+        batch_unsubscribe
+        expect(endpoint).to have_been_requested
+      end
+    end
   end
 end