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