Sha256: 91f6e1c3ef19d700a9895a6344b13033d33f66ed5e384a46c6bfbb4f49884080

Contents?: true

Size: 1.29 KB

Versions: 7

Compression:

Stored size: 1.29 KB

Contents

module Telegram
  module Bot
    # Stubbed client for tests. Saves all requests into #requests hash.
    class ClientStub < Client
      attr_reader :requests

      module StubbedConstructor
        def new(*args, **kwargs)
          if self == ClientStub || !ClientStub.stub_all?
            super
          else
            ClientStub.new(*args, **kwargs)
          end
        end
      end

      class << self
        # Any call to Client.new will return ClientStub instance when `enabled` is true.
        # Can be used with a block.
        def stub_all!(enabled = true)
          Client.extend(StubbedConstructor) unless Client < StubbedConstructor
          return @_stub_all = enabled unless block_given?
          begin
            old = @_stub_all
            stub_all!(enabled)
            yield
          ensure
            stub_all!(old)
          end
        end

        def stub_all?
          @_stub_all
        end
      end

      def initialize(token = nil, username = nil, **options)
        @token = token || options[:token]
        @username = username || options[:username] || token
        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

Version data entries

7 entries across 7 versions & 1 rubygems

Version Path
telegram-bot-0.16.1 lib/telegram/bot/client_stub.rb
telegram-bot-0.16.0 lib/telegram/bot/client_stub.rb
telegram-bot-0.15.7 lib/telegram/bot/client_stub.rb
telegram-bot-0.15.6 lib/telegram/bot/client_stub.rb
telegram-bot-0.15.4 lib/telegram/bot/client_stub.rb
telegram-bot-0.15.3 lib/telegram/bot/client_stub.rb
telegram-bot-0.15.2 lib/telegram/bot/client_stub.rb