Contents

module Clockwork
  module Test
    class Manager < Clockwork::Manager
      attr_reader :total_ticks, :max_ticks, :end_time

      def initialize(opts = {})
        super()
        @history = JobHistory.new

        @total_ticks = 0
        @max_ticks = opts[:max_ticks]
        @start_time = opts[:start_time]
        @end_time = opts[:end_time]
        config[:logger].level = Logger::ERROR
      end

      def run(opts = {})
        @max_ticks = opts[:max_ticks] if opts[:max_ticks]
        @start_time = opts[:start_time] if opts[:start_time]
        @end_time = opts[:end_time] if opts[:end_time]
        @tick_speed = opts[:tick_speed]

        if @start_time
          @time_altered = true
          Timecop.travel(@start_time)
        end

        tick_loop

        Timecop.return if @time_altered
      end

      def ran_job?(job)
        history.ran_job?(job)
      end

      def times_run(job)
        history.times_run(job)
      end

      def block_for(job)
        history.block_for(job)
      end

      private

      attr_reader :history

      def tick_loop
        while true do
          update_job_history

          tick
          increase_time

          @total_ticks += 1
          break if ticks_exceeded? || time_exceeded?
        end
      end

      def increase_time
        Timecop.travel(Time.now + @tick_speed) if @tick_speed
      end

      def register(period, job, block, options)
        event = ::Clockwork::Test::Event.new(self, period, job, block || handler, options)
        @events << event
        event
      end

      def update_job_history
        history.record(events_run_now)
      end

      def events_run_now
        events_to_run(Time.now)
      end

      def ticks_exceeded?
        max_ticks && total_ticks >= max_ticks
      end

      def time_exceeded?
        end_time && Time.now >= end_time
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
clockwork-test-0.2.0	lib/clockwork/test/manager.rb