Sha256: 6257ccbff3dffce024e8c148060a74e69eb568e6413f27c8f0232fe71dd5606f
Contents?: true
Size: 1.27 KB
Versions: 13
Compression:
Stored size: 1.27 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) if self == ClientStub || !ClientStub.stub_all? super else ClientStub.new(*args) 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
13 entries across 13 versions & 1 rubygems