File: yield_parameters.rb

package info (click to toggle)
ruby-mocha 3.1.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 2,656 kB
  • sloc: ruby: 12,304; javascript: 499; makefile: 14
file content (23 lines) | stat: -rw-r--r-- 453 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
 
# frozen_string_literal: true

module Mocha
  class YieldParameters
    def initialize
      @parameter_groups = []
    end

    def next_invocation
      case @parameter_groups.length
      when 0 then []
      when 1 then @parameter_groups.first
      else @parameter_groups.shift
      end
    end

    def add(*parameter_groups)
      @parameter_groups << parameter_groups.map do |pg|
        pg.is_a?(Array) ? pg : [pg]
      end
    end
  end
end