# various classes used by the specs
module RSpec
  module Expectations
    module Helper
      class CollectionWithSizeMethod
        def initialize; @list = []; end
        def size; @list.size; end
        def push(item); @list.push(item); end
      end

      class CollectionWithLengthMethod
        def initialize; @list = []; end
        def length; @list.size; end
        def push(item); @list.push(item); end
      end

      class CollectionWithCountMethod
        def initialize; @list = []; end
        def count; @list.count; end
        def push(item); @list.push(item); end
      end

      class CollectionOwner
        attr_reader :items_in_collection_with_size_method, :items_in_collection_with_length_method, :items_in_collection_with_count_method

        def initialize
          @items_in_collection_with_size_method = CollectionWithSizeMethod.new
          @items_in_collection_with_length_method = CollectionWithLengthMethod.new
          @items_in_collection_with_count_method = CollectionWithCountMethod.new
        end

        def add_to_collection_with_size_method(item)
          @items_in_collection_with_size_method.push(item)
        end

      def add_to_collection_with_length_method(item)
        @items_in_collection_with_length_method.push(item)
      end

      def add_to_collection_with_count_method(item)
        @items_in_collection_with_count_method.push(item)
      end

        def items_for(arg)
          return [1, 2, 3] if arg == 'a'
          [1]
        end

        def items
          @items_in_collection_with_size_method
        end
      end

    end
  end
end