Chatwoot/spec/services/notification/push_notification_service_spec.rb

require 'rails_helper'

describe Notification::PushNotificationService do
  let!(:account) { create(:account) }
  let!(:user) { create(:user, account: account) }
  let!(:notification) { create(:notification, user: user, account: user.accounts.first) }
  let(:fcm_double) { double }

  before do
    allow(Webpush).to receive(:payload_send).and_return(true)
    allow(FCM).to receive(:new).and_return(fcm_double)
    allow(fcm_double).to receive(:send).and_return({ body: { 'results': [] }.to_json })
  end

  describe '#perform' do
    it 'sends webpush notifications for webpush subscription' do
      ENV['VAPID_PUBLIC_KEY'] = 'test'
      create(:notification_subscription, user: notification.user)

      described_class.new(notification: notification).perform
      expect(Webpush).to have_received(:payload_send)
      expect(FCM).not_to have_received(:new)
      ENV['VAPID_PUBLIC_KEY'] = nil
    end

    it 'sends a fcm notification for firebase subscription' do
      ENV['FCM_SERVER_KEY'] = 'test'
      ENV['ENABLE_PUSH_RELAY_SERVER'] = 'false'
      create(:notification_subscription, user: notification.user, subscription_type: 'fcm')

      described_class.new(notification: notification).perform
      expect(FCM).to have_received(:new)
      expect(Webpush).not_to have_received(:payload_send)

      ENV['FCM_SERVER_KEY'] = nil
      ENV['ENABLE_PUSH_RELAY_SERVER'] = nil
    end
  end
end