require_relative 'unit_helper'

describe Watir::Wait do
  describe "#until" do
    it "waits until the block returns true" do
      Watir::Wait.until(timeout: 0.5) { @result = true }
      expect(@result).to be true
    end

    it "executes block if timeout is zero" do
      Watir::Wait.until(timeout: 0) { @result = true }
      expect(@result).to be true
    end

    it "times out" do
      expect { Watir::Wait.until(timeout: 0.5) { false } }.to raise_error(Watir::Wait::TimeoutError)
    end

    it "times out with a custom message" do
      expect {
        Watir::Wait.until(timeout: 0.5, message: "oops") { false }
      }.to raise_error(Watir::Wait::TimeoutError, "timed out after 0.5 seconds, oops")
    end

    it "uses provided interval" do
      begin
        Watir::Wait.until(timeout: 0.4, interval: 0.2) do
          @result = @result.nil? ? 1 : @result + 1
          false
        end
      rescue Watir::Wait::TimeoutError
      end
      expect(@result).to eq 2
    end

    it "uses timer for waiting" do
      timer = Watir::Wait.timer
      expect(timer).to receive(:wait).with(0.5).and_call_original
      Watir::Wait.until(timeout: 0.5) { true }
    end
  end

  describe "#while" do
    it "waits while the block returns true" do
      expect(Watir::Wait.while(timeout: 0.5) { false }).to be_nil
    end

    it "executes block if timeout is zero" do
      expect(Watir::Wait.while(timeout: 0) { false }).to be_nil
    end

    it "times out" do
      expect { Watir::Wait.while(timeout: 0.5) { true } }.to raise_error(Watir::Wait::TimeoutError)
    end

    it "times out with a custom message" do
      expect {
        Watir::Wait.while(timeout: 0.5, message: "oops") { true }
      }.to raise_error(Watir::Wait::TimeoutError, "timed out after 0.5 seconds, oops")
    end

    it "uses provided interval" do
      begin
        Watir::Wait.while(timeout: 0.4, interval: 0.2) do
          @result = @result.nil? ? 1 : @result + 1
          true
        end
      rescue Watir::Wait::TimeoutError
      end
      expect(@result).to eq 2
    end

    it "uses timer for waiting" do
      timer = Watir::Wait.timer
      expect(timer).to receive(:wait).with(0.5).and_call_original
      Watir::Wait.while(timeout: 0.5) { false }
    end
  end

  describe "#timer" do
    it "returns default timer" do
      expect(Watir::Wait.timer).to be_a(Watir::Wait::Timer)
    end
  end

  describe "#timer=" do
    after { Watir::Wait.timer = nil }

    it "changes default timer" do
      timer = Class.new
      Watir::Wait.timer = timer
      expect(Watir::Wait.timer).to eq(timer)
    end
  end
end