ClientStub and improved rspec helpers

lib/telegram/bot.rb

@@ -8,6 +8,7 @@ module Telegram
     class NotFound < Error; end
 
     autoload :Client,             'telegram/bot/client'
+    autoload :ClientStub,         'telegram/bot/client_stub'
     autoload :Middleware,         'telegram/bot/middleware'
     autoload :UpdatesController,  'telegram/bot/updates_controller'
     autoload :UpdatesPoller,      'telegram/bot/updates_poller'

lib/telegram/bot/client_stub.rb

@@ -0,0 +1,21 @@
+module Telegram
+  module Bot
+    # Stubbed client for tests. Saves all requests into #requests hash.
+    class ClientStub < Client
+      attr_reader :requests
+
+      def initialize(username = nil)
+        @username = username
+        reset
+      end
+
+      def reset
+        @requests = Hash.new { |h, k| h[k] = [] }
+      end
+
+      def request(action, body)
+        requests[action.to_sym] << body
+      end
+    end
+  end
+end

lib/telegram/bot/updates_controller/rspec_helpers.rb

@@ -1,8 +1,31 @@
 RSpec.shared_context 'telegram/bot/updates_controller' do
-  let(:instance) { described_class.new(bot, update) }
+  let(:controller_class) { described_class }
+  let(:instance) { controller_class.new(bot, update) }
   let(:update) { {payload_type => payload} }
   let(:payload_type) { 'some_type' }
   let(:payload) { double(:payload) }
-  let(:bot) { double(username: bot_name) }
+  let(:bot) { Telegram::Bot::ClientStub.new(bot_name) }
   let(:bot_name) { 'bot' }
+  let(:session) do
+    session = Telegram::Bot::UpdatesController::Session::TestSessionHash.new
+    allow_any_instance_of(controller_class).to receive(:session) { session }
+    session
+  end
+
+  def dispatch_message(text, options = {})
+    payload = build_payload :message, options.merge(text: text)
+    controller_class.dispatch bot, payload
+  end
+
+  def build_payload(type, content)
+    deep_stringify type => content
+  end
+
+  def deep_stringify(input)
+    case input
+    when Array then input.map(&method(__callee__))
+    when Hash then input.transform_keys(&:to_s).transform_values(&method(__callee__))
+    else input
+    end
+  end
 end

lib/telegram/bot/updates_controller/session.rb

@@ -59,6 +59,18 @@ module Telegram
           end
         end
 
+        class TestSessionHash < SessionHash
+          def initialize
+            @data = {}
+            @loaded = true
+            @exists = true
+          end
+
+          alias_method :destroy, :clear
+          alias_method :load!, :id
+          alias_method :commit, :id
+        end
+
         module ConfigMethods
           delegate :session_store, to: :config
 

spec/telegram/bot/updates_controller/typed_update_spec.rb

@@ -1,6 +1,5 @@
 RSpec.describe Telegram::Bot::UpdatesController::TypedUpdate do
   include_context 'telegram/bot/updates_controller'
-  let(:instance) { controller_class.new(bot, update) }
   let(:controller_class) do
     described_class = self.described_class
     Class.new(Telegram::Bot::UpdatesController) do

spec/telegram/bot/updates_controller_spec.rb

@@ -148,7 +148,6 @@ RSpec.describe Telegram::Bot::UpdatesController do
     subject { -> { instance.process_action(:action) } }
 
     context 'when callbacks are defined' do
-      let(:instance) { controller_class.new }
       let(:controller_class) do
         Class.new(described_class) do
           before_action :hook