require "watirspec_helper"

describe "Browser::AfterHooks" do
  describe "#add" do
    it "raises ArgumentError when not given any arguments" do
      expect { browser.after_hooks.add }.to raise_error(ArgumentError)
    end

    it "runs the given proc on each page load" do
      output = ''
      proc = Proc.new { |browser| output << browser.text }

      begin
        browser.after_hooks.add(proc)
        browser.goto(WatirSpec.url_for("non_control_elements.html"))

        expect(output).to include('Dubito, ergo cogito, ergo sum')
      ensure
        browser.after_hooks.delete(proc)
      end
    end
  end

  describe "#delete" do
    it "removes a previously added after_hook" do
      output = ''
      after_hook = lambda{ |browser| output << browser.text }

      browser.after_hooks.add(after_hook)
      browser.goto(WatirSpec.url_for("non_control_elements.html"))
      expect(output).to include('Dubito, ergo cogito, ergo sum')

      browser.after_hooks.delete(after_hook)
      browser.goto(WatirSpec.url_for("definition_lists.html"))
      expect(output).to_not include('definition_lists')
    end
  end

  describe "#run" do
    after(:each) do
      browser.window(index: 0).use
      browser.after_hooks.delete @page_after_hook
    end

    it "runs after_hooks after Browser#goto" do
      @page_after_hook = Proc.new { @yield = browser.title == "The font element" }
      browser.after_hooks.add @page_after_hook
      browser.goto WatirSpec.url_for("font.html")
      expect(@yield).to be true
    end

    it "runs after_hooks after Browser#refresh" do
      browser.goto WatirSpec.url_for("font.html")
      @page_after_hook = Proc.new do
        @yield = browser.title == "The font element"
      end
      browser.after_hooks.add @page_after_hook
      browser.refresh
      expect(@yield).to be true
    end

    it "runs after_hooks after Element#click" do
      browser.goto(WatirSpec.url_for("non_control_elements.html"))
      @page_after_hook = Proc.new do
        browser.wait_until { |b| b.title == "Forms with input elements" }
        @yield = true
      end
      browser.after_hooks.add @page_after_hook
      browser.link(index: 2).click
      expect(@yield).to be true
    end

    bug "AutomatedTester: 'known bug with execute script'", :firefox do
      it "runs after_hooks after Element#submit" do
        browser.goto(WatirSpec.url_for("forms_with_input_elements.html"))
        @page_after_hook = Proc.new { @yield = browser.div(id: 'messages').text == 'submit' }
        browser.after_hooks.add @page_after_hook
        browser.form(id: "new_user").submit
        expect(@yield).to be true
      end
    end

    not_compliant_on :safari do
      bug "Actions Endpoint Not Yet Implemented", :firefox do
        it "runs after_hooks after Element#double_click" do
          browser.goto(WatirSpec.url_for("non_control_elements.html"))
          @page_after_hook = Proc.new { @yield = browser.title == "Non-control elements" }
          browser.after_hooks.add @page_after_hook
          browser.div(id: 'html_test').double_click
          expect(@yield).to be true
        end
      end
    end

    not_compliant_on :safari do
      bug "Actions Endpoint Not Yet Implemented", :firefox do
        it "runs after_hooks after Element#right_click" do
          browser.goto(WatirSpec.url_for("right_click.html"))
          @page_after_hook = Proc.new { @yield = browser.title == "Right Click Test" }
          browser.after_hooks.add @page_after_hook
          browser.div(id: "click").right_click
          expect(@yield).to be true
        end
      end
    end

    not_compliant_on :safari do
      it "runs after_hooks after Alert#ok" do
        browser.goto(WatirSpec.url_for("alerts.html"))
        @page_after_hook = Proc.new { @yield = browser.title == "Alerts" }
        browser.after_hooks.add @page_after_hook
        browser.after_hooks.without { browser.button(id: 'alert').click }
        browser.alert.ok
        expect(@yield).to be true
      end

      bug "https://code.google.com/p/chromedriver/issues/detail?id=26", [:chrome, :macosx] do
        it "runs after_hooks after Alert#close" do
          browser.goto(WatirSpec.url_for("alerts.html"))
          @page_after_hook = Proc.new { @yield = browser.title == "Alerts" }
          browser.after_hooks.add @page_after_hook
          browser.after_hooks.without { browser.button(id: 'alert').click }
          browser.alert.close
          expect(@yield).to be true
        end
      end

      bug "https://bugzilla.mozilla.org/show_bug.cgi?id=1279211", :firefox do
        it "raises UnhandledAlertError error when running error checks with alert present" do
          url = WatirSpec.url_for("alerts.html")
          @page_after_hook = Proc.new { browser.url }
          browser.after_hooks.add @page_after_hook
          browser.goto url
          expect { browser.button(id: "alert").click }.to raise_error(Selenium::WebDriver::Error::UnhandledAlertError)

          not_compliant_on :ff_legacy do
            browser.alert.ok
          end
        end
      end

      it "does not raise error when running error checks using #after_hooks#without with alert present" do
        url = WatirSpec.url_for("alerts.html")
        @page_after_hook = Proc.new { browser.url }
        browser.after_hooks.add @page_after_hook
        browser.goto url
        expect { browser.after_hooks.without { browser.button(id: "alert").click } }.to_not raise_error
        browser.alert.ok
      end

      it "does not raise error if no error checks are defined with alert present" do
        url = WatirSpec.url_for("alerts.html")
        @page_after_hook = Proc.new { browser.url }
        browser.after_hooks.add @page_after_hook
        browser.goto url
        browser.after_hooks.delete @page_after_hook
        expect { browser.button(id: "alert").click }.to_not raise_error
        browser.alert.ok
      end
    end

    bug "https://bugzilla.mozilla.org/show_bug.cgi?id=1223277", :firefox do
      it "does not raise error when running error checks on closed window" do
        url = WatirSpec.url_for("window_switching.html")
        @page_after_hook = Proc.new { browser.url }
        browser.after_hooks.add @page_after_hook
        browser.goto url
        browser.a(id: "open").click

        window = browser.window(title: "closeable window")
        window.use
        expect { browser.a(id: "close").click }.to_not raise_error
        browser.window(index: 0).use
      end
    end
  end
end