gistfile2.txt

phase_state.rb

class PhaseState
  
  def initialize(*roles)
    @state = 0
    @roles = roles
  end
  
  
  def current
    @roles[@state % @roles.size]
  end
  
  
  def next!
    @state += 1
    current
  end
  
end


require 'test/unit'
class PhaseStateTest < Test::Unit::TestCase
  
  def test_phase_state
    phase_state = PhaseState.new(:killers, :doctor, :detective)
    
    assert(phase_state.current == :killers)
    assert(phase_state.next! == :doctor)
    
    assert(phase_state.current == :doctor)
    assert(phase_state.next! == :detective)
    
    assert(phase_state.current == :detective)
    assert(phase_state.next! == :killers)
    
    assert(phase_state.current == :killers)
    assert(phase_state.next! == :doctor)
  end
  
end