Sha256: 0fb1ea818fc375f29a9b162fdafa402df9124e6d07e7238d6b7f5d95aad39a5a
Contents?: true
Size: 1.22 KB
Versions: 57
Compression:
Stored size: 1.22 KB
Contents
# encoding: utf-8
# frozen_string_literal: true
module Mail
class TestRetriever < Retriever
def self.emails
@@emails
end
def self.emails=(val)
@@emails = val
end
def initialize(values)
@@emails = []
end
def find(options = {}, &block)
options[:count] ||= :all
options[:order] ||= :asc
options[:what] ||= :first
emails_index = (0...@@emails.size).to_a
emails_index.reverse! if options[:what] == :last
emails_index = case count = options[:count]
when :all then emails_index
when Integer then emails_index[0, count]
else
raise 'Invalid count option value: ' + count.inspect
end
if options[:what] == :last && options[:order] == :asc || options[:what] == :first && options[:order] == :desc
emails_index.reverse!
end
emails_index.each { |idx| @@emails[idx].mark_for_delete = true } if options[:delete_after_find]
emails = emails_index.map { |idx| @@emails[idx] }
emails.each { |email| yield email } if block_given?
@@emails.reject!(&:is_marked_for_delete?) if options[:delete_after_find]
emails.size == 1 && options[:count] == 1 ? emails.first : emails
end
end
end
Version data entries
57 entries across 43 versions & 10 rubygems