rpush 8.0.0 → 9.0.0

data/lib/rpush/daemon/tcp_connection.rb

@@ -1,190 +0,0 @@
-module Rpush
-  module Daemon
-    class TcpConnectionError < StandardError; end
-
-    class TcpConnection
-      include Reflectable
-      include Loggable
-
-      OSX_TCP_KEEPALIVE = 0x10 # Defined in <netinet/tcp.h>
-      KEEPALIVE_INTERVAL = 5
-      KEEPALIVE_IDLE = 5
-      KEEPALIVE_MAX_FAIL_PROBES = 1
-      TCP_ERRORS = [SystemCallError, OpenSSL::OpenSSLError, IOError]
-
-      attr_accessor :last_touch
-      attr_reader :host, :port
-
-      def self.idle_period
-        30.minutes
-      end
-
-      def initialize(app, host, port)
-        @app = app
-        @host = host
-        @port = port
-        @certificate = app.certificate
-        @password = app.password
-        @connected = false
-        @connection_callbacks = []
-        touch
-      end
-
-      def on_connect(&blk)
-        raise 'already connected' if @connected
-        @connection_callbacks << blk
-      end
-
-      def connect
-        @ssl_context = setup_ssl_context
-        @tcp_socket, @ssl_socket = connect_socket
-        @connected = true
-
-        @connection_callbacks.each do |blk|
-          begin
-            blk.call
-          rescue StandardError => e
-            log_error(e)
-          end
-        end
-
-        @connection_callbacks.clear
-      end
-
-      def close
-        @ssl_socket.close if @ssl_socket
-        @tcp_socket.close if @tcp_socket
-      rescue IOError # rubocop:disable HandleExceptions
-      end
-
-      def read(num_bytes)
-        @ssl_socket.read(num_bytes) if @ssl_socket
-      end
-
-      def select(timeout)
-        IO.select([@ssl_socket], nil, nil, timeout) if @ssl_socket
-      end
-
-      def write(data)
-        connect unless @connected
-        reconnect_idle if idle_period_exceeded?
-
-        retry_count = 0
-
-        begin
-          write_data(data)
-        rescue *TCP_ERRORS => e
-          retry_count += 1
-
-          if retry_count == 1
-            log_error("Lost connection to #{@host}:#{@port} (#{e.class.name}, #{e.message}), reconnecting...")
-            reflect(:tcp_connection_lost, @app, e)
-          end
-
-          if retry_count <= 3
-            reconnect_with_rescue
-            sleep 1
-            retry
-          else
-            raise TcpConnectionError, "#{@app.name} tried #{retry_count - 1} times to reconnect but failed (#{e.class.name}, #{e.message})."
-          end
-        end
-      end
-
-      def reconnect_with_rescue
-        reconnect
-      rescue StandardError => e
-        log_error(e)
-      end
-
-      def reconnect
-        close
-        @tcp_socket, @ssl_socket = connect_socket
-      end
-
-      protected
-
-      def reconnect_idle
-        log_info("Idle period exceeded, reconnecting...")
-        reconnect
-      end
-
-      def idle_period_exceeded?
-        Time.now - last_touch > self.class.idle_period
-      end
-
-      def write_data(data)
-        @ssl_socket.write(data)
-        @ssl_socket.flush
-        touch
-      end
-
-      def touch
-        self.last_touch = Time.now
-      end
-
-      def setup_ssl_context
-        ssl_context = OpenSSL::SSL::SSLContext.new
-        ssl_context.key = OpenSSL::PKey::RSA.new(@certificate, @password)
-        ssl_context.cert = OpenSSL::X509::Certificate.new(@certificate)
-        ssl_context
-      end
-
-      def connect_socket
-        touch
-        check_certificate_expiration
-
-        tcp_socket = TCPSocket.new(@host, @port)
-        tcp_socket.setsockopt(Socket::SOL_SOCKET, Socket::SO_KEEPALIVE, true)
-        tcp_socket.setsockopt(Socket::IPPROTO_TCP, Socket::TCP_NODELAY, true)
-
-        # Linux
-        if [:SOL_TCP, :TCP_KEEPIDLE, :TCP_KEEPINTVL, :TCP_KEEPCNT].all? { |c| Socket.const_defined?(c) }
-          tcp_socket.setsockopt(Socket::SOL_TCP, Socket::TCP_KEEPIDLE, KEEPALIVE_IDLE)
-          tcp_socket.setsockopt(Socket::SOL_TCP, Socket::TCP_KEEPINTVL, KEEPALIVE_INTERVAL)
-          tcp_socket.setsockopt(Socket::SOL_TCP, Socket::TCP_KEEPCNT, KEEPALIVE_MAX_FAIL_PROBES)
-        end
-
-        # OSX
-        if RUBY_PLATFORM =~ /darwin/
-          tcp_socket.setsockopt(Socket::IPPROTO_TCP, OSX_TCP_KEEPALIVE, KEEPALIVE_IDLE)
-        end
-
-        ssl_socket = OpenSSL::SSL::SSLSocket.new(tcp_socket, @ssl_context)
-        ssl_socket.sync = true
-        ssl_socket.connect
-        [tcp_socket, ssl_socket]
-      rescue *TCP_ERRORS => error
-        if error.message =~ /certificate revoked/i
-          log_error('Certificate has been revoked.')
-          reflect(:ssl_certificate_revoked, @app, error)
-        end
-        raise TcpConnectionError, "#{error.class.name}, #{error.message}"
-      end
-
-      def check_certificate_expiration
-        cert = @ssl_context.cert
-        if certificate_expired?
-          log_error(certificate_msg('expired'))
-          raise Rpush::CertificateExpiredError.new(@app, cert.not_after)
-        elsif certificate_expires_soon?
-          log_warn(certificate_msg('will expire'))
-          reflect(:ssl_certificate_will_expire, @app, cert.not_after)
-        end
-      end
-
-      def certificate_msg(msg)
-        time = @ssl_context.cert.not_after.utc.strftime('%Y-%m-%d %H:%M:%S UTC')
-        "Certificate #{msg} at #{time}."
-      end
-
-      def certificate_expired?
-        @ssl_context.cert.not_after && @ssl_context.cert.not_after.utc < Time.now.utc
-      end
-
-      def certificate_expires_soon?
-        @ssl_context.cert.not_after && @ssl_context.cert.not_after.utc < (Time.now + 1.month).utc
-      end
-    end
-  end
-end

data/spec/functional/apns_spec.rb

@@ -1,162 +0,0 @@
-require 'functional_spec_helper'
-
-describe 'APNs' do
-  let(:app) { create_app }
-  let(:tcp_socket) { double(TCPSocket, setsockopt: nil, close: nil) }
-  let(:ssl_socket) { double(OpenSSL::SSL::SSLSocket, :sync= => nil, connect: nil, write: nil, flush: nil, read: nil, close: nil) }
-  let(:io_double) { double(select: nil) }
-  let(:delivered_ids) { [] }
-  let(:failed_ids) { [] }
-  let(:retry_ids) { [] }
-
-  before do
-    Rpush.config.push_poll = 0.5
-    stub_tcp_connection(tcp_socket, ssl_socket, io_double)
-  end
-
-  def create_app
-    app = Rpush::Apns::App.new
-    app.certificate = TEST_CERT
-    app.name = 'test'
-    app.environment = 'sandbox'
-    app.save!
-    app
-  end
-
-  def create_notification
-    notification = Rpush::Apns::Notification.new
-    notification.app = app
-    notification.alert = 'test'
-    notification.device_token = 'a' * 108
-    notification.save!
-    notification
-  end
-
-  def wait
-    sleep 0.1
-  end
-
-  def wait_for_notification_to_deliver(notification)
-    timeout { wait until delivered_ids.include?(notification.id) }
-  end
-
-  def wait_for_notification_to_fail(notification)
-    timeout { wait until failed_ids.include?(notification.id) }
-  end
-
-  def wait_for_notification_to_retry(notification)
-    timeout { wait until retry_ids.include?(notification.id) }
-  end
-
-  def fail_notification(notification)
-    allow(ssl_socket).to receive_messages(read: [8, 4, notification.id].pack('ccN'))
-    enable_io_select
-  end
-
-  def enable_io_select
-    called = false
-    allow(io_double).to receive(:select) do
-      if called
-        nil
-      else
-        called = true
-      end
-    end
-  end
-
-  it 'delivers a notification successfully' do
-    notification = create_notification
-    expect do
-      Rpush.push
-      notification.reload
-    end.to change(notification, :delivered).to(true)
-  end
-
-  it 'receives feedback' do
-    app
-    tuple = "N\xE3\x84\r\x00 \x83OxfU\xEB\x9F\x84aJ\x05\xAD}\x00\xAF1\xE5\xCF\xE9:\xC3\xEA\a\x8F\x1D\xA4M*N\xB0\xCE\x17"
-    allow(ssl_socket).to receive(:read).and_return(tuple, nil)
-    Rpush.apns_feedback
-    feedback = Rpush::Apns::Feedback.all.first
-    expect(feedback).not_to be_nil
-    expect(feedback.app_id).to eq(app.id)
-    expect(feedback.device_token).to eq('834f786655eb9f84614a05ad7d00af31e5cfe93ac3ea078f1da44d2a4eb0ce17')
-  end
-
-  describe 'delivery failures' do
-    before do
-      Rpush.reflect do |on|
-        on.notification_delivered do |n|
-          delivered_ids << n.id
-        end
-
-        on.notification_id_failed do |_, n_id|
-          failed_ids << n_id
-        end
-
-        on.notification_id_will_retry do |_, n_id|
-          retry_ids << n_id
-        end
-
-        on.notification_will_retry do |n|
-          retry_ids << n.id
-        end
-      end
-
-      Rpush.embed
-    end
-
-    after do
-      Rpush.reflection_stack.clear
-      Rpush.reflection_stack.push(Rpush::ReflectionCollection.new)
-
-      timeout { Rpush.shutdown }
-    end
-
-    it 'fails to deliver a notification' do
-      notification = create_notification
-      wait_for_notification_to_deliver(notification)
-      fail_notification(notification)
-      wait_for_notification_to_fail(notification)
-    end
-
-    describe 'with a failed connection' do
-      it 'retries all notifications' do
-        allow_any_instance_of(Rpush::Daemon::TcpConnection).to receive_messages(sleep: nil)
-        expect(ssl_socket).to receive(:write).at_least(1).times.and_raise(Errno::EPIPE)
-        notifications = 2.times.map { create_notification }
-        notifications.each { |n| wait_for_notification_to_retry(n) }
-      end
-    end
-
-    describe 'with multiple notifications' do
-      let(:notification1) { create_notification }
-      let(:notification2) { create_notification }
-      let(:notification3) { create_notification }
-      let(:notification4) { create_notification }
-      let(:notifications) { [notification1, notification2, notification3, notification4] }
-
-      it 'marks the correct notification as failed' do
-        notifications.each { |n| wait_for_notification_to_deliver(n) }
-        fail_notification(notification2)
-        wait_for_notification_to_fail(notification2)
-      end
-
-      it 'does not mark prior notifications as failed' do
-        notifications.each { |n| wait_for_notification_to_deliver(n) }
-        fail_notification(notification2)
-        wait_for_notification_to_fail(notification2)
-
-        expect(failed_ids).to_not include(notification1.id)
-        notification1.reload
-        expect(notification1.delivered).to eq(true)
-      end
-
-      it 'marks notifications following the failed one as retryable' do
-        notifications.each { |n| wait_for_notification_to_deliver(n) }
-        fail_notification(notification2)
-        [notification3, notification4].each { |n| wait_for_notification_to_retry(n) }
-      end
-    end
-  end
-end

data/spec/functional/gcm_priority_spec.rb

@@ -1,40 +0,0 @@
-require 'functional_spec_helper'
-
-describe 'GCM priority' do
-  let(:app) { Rpush::Gcm::App.new }
-  let(:notification) { Rpush::Gcm::Notification.new }
-  let(:hydrated_notification) { Rpush::Gcm::Notification.find(notification.id) }
-  let(:response) { double(Net::HTTPResponse, code: 200) }
-  let(:http) { double(Net::HTTP::Persistent, request: response, shutdown: nil) }
-  let(:priority) { 'normal' }
-
-  before do
-    app.name = 'test'
-    app.auth_key = 'abc123'
-    app.save!
-
-    notification.app_id = app.id
-    notification.registration_ids = ['foo']
-    notification.data = { message: 'test' }
-    notification.priority = priority
-    notification.save!
-
-    allow(Net::HTTP::Persistent).to receive_messages(new: http)
-  end
-
-  it 'supports normal priority' do
-    expect(hydrated_notification.as_json['priority']).to eq('normal')
-  end
-
-  context 'high priority' do
-    let(:priority) { 'high' }
-
-    it 'supports high priority' do
-      expect(hydrated_notification.as_json['priority']).to eq('high')
-    end
-  end
-
-  it 'does not add an error when receiving expected priority' do
-    expect(hydrated_notification.errors.messages[:priority]).to be_empty
-  end
-end

data/spec/functional/gcm_spec.rb

@@ -1,46 +0,0 @@
-require 'functional_spec_helper'
-
-describe 'GCM' do
-  let(:app) { Rpush::Gcm::App.new }
-  let(:notification) { Rpush::Gcm::Notification.new }
-  let(:response) { double(Net::HTTPResponse, code: 200) }
-  let(:http) { double(Net::HTTP::Persistent, request: response, shutdown: nil) }
-
-  before do
-    app.name = 'test'
-    app.auth_key = 'abc123'
-    app.save!
-
-    notification.app_id = app.id
-    notification.registration_ids = ['foo']
-    notification.data = { message: 'test' }
-    notification.save!
-
-    allow(Net::HTTP::Persistent).to receive_messages(new: http)
-  end
-
-  it 'delivers a notification successfully' do
-    allow(response).to receive_messages(body: JSON.dump(results: [{ message_id: notification.registration_ids.first.to_s }]))
-
-    expect do
-      Rpush.push
-      notification.reload
-    end.to change(notification, :delivered).to(true)
-  end
-
-  it 'fails to deliver a notification successfully' do
-    allow(response).to receive_messages(body: JSON.dump(results: [{ error: 'Err' }]))
-    Rpush.push
-    notification.reload
-    expect(notification.delivered).to eq(false)
-  end
-
-  it 'retries notification that fail due to a SocketError' do
-    expect(http).to receive(:request).and_raise(SocketError.new)
-    expect(notification.deliver_after).to be_nil
-    expect do
-      Rpush.push
-      notification.reload
-    end.to change(notification, :deliver_after).to(kind_of(Time))
-  end
-end

data/spec/functional/new_app_spec.rb

@@ -1,44 +0,0 @@
-require 'functional_spec_helper'
-
-describe 'New app loading' do
-  let(:timeout) { 10 }
-  let(:app) { create_app }
-  let(:tcp_socket) { double(TCPSocket, setsockopt: nil, close: nil) }
-  let(:ssl_socket) { double(OpenSSL::SSL::SSLSocket, :sync= => nil, connect: nil, write: nil, flush: nil, read: nil, close: nil) }
-  let(:io_double) { double(select: nil) }
-
-  before do
-    stub_tcp_connection
-  end
-
-  def create_app
-    app = Rpush::Apns::App.new
-    app.certificate = TEST_CERT
-    app.name = 'test'
-    app.environment = 'sandbox'
-    app.save!
-    app
-  end
-
-  def create_notification
-    notification = Rpush::Apns::Notification.new
-    notification.app = app
-    notification.alert = 'test'
-    notification.device_token = 'a' * 108
-    notification.save!
-    notification
-  end
-
-  def stub_tcp_connection
-    allow_any_instance_of(Rpush::Daemon::TcpConnection).to receive_messages(connect_socket: [tcp_socket, ssl_socket])
-    allow_any_instance_of(Rpush::Daemon::TcpConnection).to receive_messages(setup_ssl_context: double.as_null_object)
-    stub_const('Rpush::Daemon::TcpConnection::IO', io_double)
-  end
-
-  it 'delivers a notification successfully' do
-    notification = create_notification
-    Rpush.push
-    notification.reload
-    expect(notification.delivered).to eq(true)
-  end
-end

data/spec/unit/apns_feedback_spec.rb

@@ -1,39 +0,0 @@
-require 'unit_spec_helper'
-
-describe Rpush, 'apns_feedback' do
-  let!(:apns_app) { Rpush::Apns::App.create!(apns_app_params) }
-  let(:apns_app_params) do
-    {
-      name: 'test',
-      environment: 'production',
-      certificate: TEST_CERT
-    }
-  end
-  let!(:gcm_app) { Rpush::Gcm::App.create!(name: 'MyApp', auth_key: 'abc123') }
-
-  let(:receiver) { double(check_for_feedback: nil) }
-
-  before do
-    allow(Rpush::Daemon::Apns::FeedbackReceiver).to receive(:new) { receiver }
-  end
-
-  it 'initializes the daemon' do
-    expect(Rpush::Daemon).to receive(:common_init)
-    Rpush.apns_feedback
-  end
-
-  it 'checks feedback for each app' do
-    expect(Rpush::Daemon::Apns::FeedbackReceiver).to receive(:new).with(apns_app).and_return(receiver)
-    expect(receiver).to receive(:check_for_feedback)
-    Rpush.apns_feedback
-  end
-
-  context 'feedback disabled' do
-    let(:apns_app_params) { super().merge(feedback_enabled: false) }
-
-    it 'does not initialize feedback receiver' do
-      expect(Rpush::Daemon::Apns::FeedbackReceiver).not_to receive(:new)
-      Rpush.apns_feedback
-    end
-  end
-end

data/spec/unit/client/active_record/gcm/app_spec.rb

@@ -1,6 +0,0 @@
-require 'unit_spec_helper'
-
-describe Rpush::Client::ActiveRecord::Gcm::App do
-  it_behaves_like 'Rpush::Client::Gcm::App'
-  it_behaves_like 'Rpush::Client::ActiveRecord::App'
-end if active_record?

data/spec/unit/client/active_record/gcm/notification_spec.rb

@@ -1,14 +0,0 @@
-require 'unit_spec_helper'
-
-describe Rpush::Client::ActiveRecord::Gcm::Notification do
-  it_behaves_like 'Rpush::Client::Gcm::Notification'
-  it_behaves_like 'Rpush::Client::ActiveRecord::Notification'
-
-  subject(:notification) { described_class.new }
-  let(:app) { Rpush::Gcm::App.create!(name: 'test', auth_key: 'abc') }
-
-  it 'accepts non-booleans as a truthy value' do
-    notification.dry_run = 'Not a boolean'
-    expect(notification.as_json['dry_run']).to eq true
-  end
-end if active_record?

data/spec/unit/client/redis/gcm/app_spec.rb

@@ -1,5 +0,0 @@
-require 'unit_spec_helper'
-
-describe Rpush::Client::Redis::Gcm::App do
-  it_behaves_like 'Rpush::Client::Gcm::App'
-end if redis?

data/spec/unit/client/redis/gcm/notification_spec.rb

@@ -1,5 +0,0 @@
-require 'unit_spec_helper'
-
-describe Rpush::Client::Redis::Gcm::Notification do
-  it_behaves_like 'Rpush::Client::Gcm::Notification'
-end if redis?

data/spec/unit/client/shared/gcm/app.rb

@@ -1,4 +0,0 @@
-require 'unit_spec_helper'
-
-shared_examples 'Rpush::Client::Gcm::App' do
-end

data/spec/unit/client/shared/gcm/notification.rb

@@ -1,77 +0,0 @@
-require 'unit_spec_helper'
-
-shared_examples 'Rpush::Client::Gcm::Notification' do
-  let(:app) { Rpush::Gcm::App.create!(name: 'test', auth_key: 'abc') }
-  let(:notification) { described_class.new }
-
-  it "has a 'data' payload limit of 4096 bytes" do
-    notification.data = { key: "a" * 4096 }
-    expect(notification.valid?).to be_falsey
-    expect(notification.errors[:base]).to eq ["Notification payload data cannot be larger than 4096 bytes."]
-  end
-
-  it 'limits the number of registration ids to 1000' do
-    notification.registration_ids = ['a'] * (1000 + 1)
-    expect(notification.valid?).to be_falsey
-    expect(notification.errors[:base]).to eq ["Number of registration_ids cannot be larger than 1000."]
-  end
-
-  it 'validates expiry is present if collapse_key is set' do
-    notification.collapse_key = 'test'
-    notification.expiry = nil
-    expect(notification.valid?).to be_falsey
-    expect(notification.errors[:expiry]).to eq ['must be set when using a collapse_key']
-  end
-
-  it 'includes time_to_live in the payload' do
-    notification.expiry = 100
-    expect(notification.as_json['time_to_live']).to eq 100
-  end
-
-  it 'includes content_available in the payload' do
-    notification.content_available = true
-    expect(notification.as_json['content_available']).to eq true
-  end
-
-  it 'includes mutable_content in the payload' do
-    notification.mutable_content = true
-    expect(notification.as_json['mutable_content']).to eq true
-  end
-
-  it 'sets the priority to high when set to high' do
-    notification.priority = 'high'
-    expect(notification.as_json['priority']).to eq 'high'
-  end
-
-  it 'sets the priority to normal when set to normal' do
-    notification.priority = 'normal'
-    expect(notification.as_json['priority']).to eq 'normal'
-  end
-
-  it 'validates the priority is either "normal" or "high"' do
-    notification.priority = 'invalid'
-    expect(notification.errors[:priority]).to eq ['must be one of either "normal" or "high"']
-  end
-
-  it 'excludes the priority if it is not defined' do
-    expect(notification.as_json).not_to have_key 'priority'
-  end
-
-  it 'includes the notification payload if defined' do
-    notification.notification = { key: 'any key is allowed' }
-    expect(notification.as_json).to have_key 'notification'
-  end
-
-  it 'excludes the notification payload if undefined' do
-    expect(notification.as_json).not_to have_key 'notification'
-  end
-
-  it 'includes the dry_run payload if defined' do
-    notification.dry_run = true
-    expect(notification.as_json['dry_run']).to eq true
-  end
-
-  it 'excludes the dry_run payload if undefined' do
-    expect(notification.as_json).not_to have_key 'dry_run'
-  end
-end