require File.expand_path(File.dirname(__FILE__) + '/spec_helper')

include WebMock

SAMPLE_HEADERS = { "Content-Length" => "8888" }

describe "WebMock", :shared => true do
  before(:each) do
    WebMock.disable_net_connect!
    RequestRegistry.instance.reset_webmock
  end

  describe "when web connect" do

    describe "is allowed" do
      before(:each) do
        WebMock.allow_net_connect!
      end

      it "should make a real web request if request is not stubbed" do
        setup_expectations_for_real_google_request
        http_request(:get, "http://www.google.com/").
          body.should =~ /.*Google fake response.*/
      end

      it "should return stubbed response if request was stubbed" do
        stub_http_request(:get, "www.google.com").to_return(:body => "abc")
        http_request(:get, "http://www.google.com/").body.should == "abc"
      end
    end

    describe "is not allowed" do
      before(:each) do
        WebMock.disable_net_connect!
      end

      it "should return stubbed response if request was stubbed" do
        stub_http_request(:get, "www.google.com").to_return(:body => "abc")
        http_request(:get, "http://www.google.com/").body.should == "abc"
      end

      it "should raise exception if request was not stubbed" do
        lambda {
          http_request(:get, "http://www.google.com/")
        }.should raise_error(WebMock::NetConnectNotAllowedError,
          "Real HTTP connections are disabled. Unregistered request: GET http://www.google.com/")
      end
    end

  end

  describe "when matching requests" do

    describe "on method" do

      it "should match the request by method if registered" do
        stub_http_request(:get, "www.google.com")
        http_request(:get, "http://www.google.com/").status.should == "200"
      end

      it "should not match requests if method is different" do
        stub_http_request(:get, "www.google.com")
        http_request(:get, "http://www.google.com/").status.should == "200"
        lambda {
          http_request(:post, "http://www.google.com/")
        }.should raise_error(WebMock::NetConnectNotAllowedError,
          "Real HTTP connections are disabled. Unregistered request: POST http://www.google.com/"
        )
      end

    end

    describe "on body" do

      it "should match requests if body is the same" do
        stub_http_request(:get, "www.google.com").with(:body => "abc")
        http_request(
          :get, "http://www.google.com/",
          :body => "abc").status.should == "200"
      end

      it "should match requests if body is not set in the stub" do
        stub_http_request(:get, "www.google.com")
        http_request(
          :get, "http://www.google.com/",
          :body => "abc").status.should == "200"
      end

      it "should not match requests if body is different" do
        stub_http_request(:get, "www.google.com").with(:body => "abc")

        lambda {
          http_request(:get, "http://www.google.com/", :body => "def")
        }.should raise_error(WebMock::NetConnectNotAllowedError,
          "Real HTTP connections are disabled. Unregistered request: GET http://www.google.com/ with body 'def'")
      end

    end

    describe "on headers" do

      it "should match requests if headers are the same" do
        stub_http_request(:get, "www.google.com").with(:headers => SAMPLE_HEADERS )
        http_request(
          :get, "http://www.google.com/",
          :headers => SAMPLE_HEADERS).status.should == "200"
      end

      it "should match requests if request headers are not stubbed" do
        stub_http_request(:get, "www.google.com")
        http_request(
          :get, "http://www.google.com/",
          :headers => SAMPLE_HEADERS).status.should == "200"
      end


      it "should not match requests if headers are different" do
        stub_http_request(:get, "www.google.com").with(:headers => SAMPLE_HEADERS)

        lambda {
          http_request(
            :get, "http://www.google.com/",
          :headers => { 'Content-Length' => '9999'})
        }.should raise_error(WebMock::NetConnectNotAllowedError,
          %q(Real HTTP connections are disabled. Unregistered request: GET http://www.google.com/ with headers {'Content-Length'=>'9999'}))
      end
    end

    describe "with basic authentication" do

      it "should match if credentials are the same" do
        stub_http_request(:get, "user:pass@www.google.com")
        http_request(:get, "http://user:pass@www.google.com/").status.should == "200"
      end

      it "should not match if credentials are different" do
        stub_http_request(:get, "user:pass@www.google.com")
        lambda {
          http_request(:get, "http://user:pazz@www.google.com/").status.should == "200"
        }.should raise_error(WebMock::NetConnectNotAllowedError,
          %q(Real HTTP connections are disabled. Unregistered request: GET http://user:pazz@www.google.com/ with headers {'Authorization'=>'Basic dXNlcjpwYXp6'}))
      end

      it "should not match if credentials are stubbed but not provided in the request" do
        stub_http_request(:get, "user:pass@www.google.com")
        lambda {
          http_request(:get, "http://www.google.com/").status.should == "200"
        }.should raise_error(WebMock::NetConnectNotAllowedError,
          %q(Real HTTP connections are disabled. Unregistered request: GET http://www.google.com/))
      end

      it "should not match if credentials are not stubbed but exist in the request" do
        stub_http_request(:get, "www.google.com")
        lambda {
          http_request(:get, "http://user:pazz@www.google.com/").status.should == "200"
        }.should raise_error(WebMock::NetConnectNotAllowedError,
          %q(Real HTTP connections are disabled. Unregistered request: GET http://user:pazz@www.google.com/ with headers {'Authorization'=>'Basic dXNlcjpwYXp6'}))
      end

    end

  end

  describe "raising stubbed exceptions" do

    it "should raise exception if declared in a stubbed response" do
      class MyException < StandardError; end;
        stub_http_request(:get, "www.google.com").to_raise(MyException)
        lambda {
          http_request(:get, "http://www.google.com/")
        }.should raise_error(MyException, "Exception from WebMock")
      end

    end


    describe "returning stubbed responses" do

      it "should return declared body" do
        stub_http_request(:get, "www.google.com").to_return(:body => "abc")
        http_request(:get, "http://www.google.com/").body.should == "abc"
      end

      it "should return declared headers" do
        stub_http_request(:get, "www.google.com").to_return(:headers => SAMPLE_HEADERS)
        response = http_request(:get, "http://www.google.com/")
        response.headers["Content-Length"].should == "8888"
      end

      it "should return declared status" do
        stub_http_request(:get, "www.google.com").to_return(:status => 500)
        http_request(:get, "http://www.google.com/").status.should == "500"
      end

    end


    describe "precedence of stubs" do

      it "should use the last declared matching request stub" do
        stub_http_request(:get, "www.google.com").to_return(:body => "abc")
        stub_http_request(:get, "www.google.com").to_return(:body => "def")
        http_request(:get, "http://www.google.com/").body.should == "def"
      end

      it "should not be affected by the type of url or request method" do
        stub_http_request(:get, "www.google.com").to_return(:body => "abc")
        stub_http_request(:any, /.*google.*/).to_return(:body => "def")
        http_request(:get, "http://www.google.com/").body.should == "def"
      end

    end

    describe "verification of request expectation" do

      describe "when net connect not allowed" do

        before(:each) do
          WebMock.disable_net_connect!
          stub_http_request(:any, "http://www.google.com")
          stub_http_request(:any, "https://www.google.com")
        end

        it "should pass if request was executed with the same url and method" do
          lambda {
            http_request(:get, "http://www.google.com/")
            request(:get, "http://www.google.com").should have_been_made.once
          }.should_not raise_error
        end

        it "should pass if request was not expected and not executed" do
          lambda {
            request(:get, "http://www.google.com").should_not have_been_made
          }.should_not raise_error
        end

        it "should fail if request was not expected but executed" do
          lambda {
            http_request(:get, "http://www.google.com/")
            request(:get, "http://www.google.com").should_not have_been_made
          }.should fail_with("The request GET http://www.google.com/ was expected to execute 0 times but it executed 1 time")
        end


        it "should fail if request was not executed" do
          lambda {
            request(:get, "http://www.google.com").should have_been_made
          }.should fail_with("The request GET http://www.google.com/ was expected to execute 1 time but it executed 0 times")
        end

        it "should fail if request was executed to different url" do
          lambda {
            http_request(:get, "http://www.google.com/")
            request(:get, "http://www.google.org").should have_been_made
          }.should fail_with("The request GET http://www.google.org/ was expected to execute 1 time but it executed 0 times")
        end

        it "should fail if request was executed with different method" do
          lambda {
            http_request(:post, "http://www.google.com/")
            request(:get, "http://www.google.com").should have_been_made
          }.should fail_with("The request GET http://www.google.com/ was expected to execute 1 time but it executed 0 times")
        end

        it "should_pass if request was executed with different form of url" do
          lambda {
            http_request(:get, "http://www.google.com/")
            request(:get, "www.google.com").should have_been_made
          }.should_not raise_error
        end

        it "should_pass if request was executed with different form of url without port " do
          lambda {
            http_request(:get, "http://www.google.com/")
            request(:get, "www.google.com:80").should have_been_made
          }.should_not raise_error
        end

        it "should_pass if request was executed with different form of url with port" do
          lambda {
            http_request(:get, "http://www.google.com/")
            request(:get, "www.google.com:80").should have_been_made
          }.should_not raise_error
        end

        it "should fail if request was executed with different  port" do
          lambda {
            http_request(:get, "http://www.google.com:80/")
            request(:get, "www.google.com:90").should have_been_made
          }.should fail_with("The request GET http://www.google.com:90/ was expected to execute 1 time but it executed 0 times")
        end

        it "should_pass if request was executed with different form of url with https port" do
          lambda {
            http_request(:get, "https://www.google.com/")
            request(:get, "https://www.google.com:443/").should have_been_made
          }.should_not raise_error
        end

        it "should fail if requested more times than expected" do
          lambda {
            http_request(:get, "http://www.google.com/")
            http_request(:get, "http://www.google.com/")
            request(:get, "http://www.google.com").should have_been_made
          }.should fail_with("The request GET http://www.google.com/ was expected to execute 1 time but it executed 2 times")
        end

        it "should fail if requested less times than expected" do
          lambda {
            http_request(:get, "http://www.google.com/")
            request(:get, "http://www.google.com").should have_been_made.twice
          }.should fail_with("The request GET http://www.google.com/ was expected to execute 2 times but it executed 1 time")
        end

        it "should fail if requested less times than expected when 3 times expected" do
          lambda {
            http_request(:get, "http://www.google.com/")
            request(:get, "http://www.google.com").should have_been_made.times(3)
          }.should fail_with("The request GET http://www.google.com/ was expected to execute 3 times but it executed 1 time")
        end

        it "should succeed if request was executed with the same body" do
          lambda {
            http_request(:get, "http://www.google.com/", :body => "abc")
            request(:get, "www.google.com").with(:body => "abc").should have_been_made
          }.should_not raise_error
        end

        it "should fail if request was executed with different body" do
          lambda {
            http_request(:get, "http://www.google.com/", :body => "abc")
            request(:get, "www.google.com").
            with(:body => "def").should have_been_made
          }.should fail_with("The request GET http://www.google.com/ with body 'def' was expected to execute 1 time but it executed 0 times")
        end

        it "should succeed if request was executed with the same headers" do
          lambda {
            http_request(:get, "http://www.google.com/", :headers => SAMPLE_HEADERS)
            request(:get, "www.google.com").
            with(:headers => SAMPLE_HEADERS).should have_been_made
          }.should_not raise_error
        end

        it "should fail if request was executed with different headers" do
          lambda {
            http_request(:get, "http://www.google.com/", :headers => SAMPLE_HEADERS)
            request(:get, "www.google.com").
            with(:headers => { 'Content-Length' => '9999'}).should have_been_made
          }.should fail_with("The request GET http://www.google.com/ with headers {'Content-Length'=>'9999'} was expected to execute 1 time but it executed 0 times")
        end

        it "should fail if request was executed with less headers" do
          lambda {
            http_request(:get, "http://www.google.com/", :headers => {'A' => 'a'})
            request(:get, "www.google.com").
            with(:headers => {'A' => 'a', 'B' => 'b'}).should have_been_made
          }.should fail_with("The request GET http://www.google.com/ with headers {'A'=>'a', 'B'=>'b'} was expected to execute 1 time but it executed 0 times")
        end

        it "should succeed if request was executed with more headers" do
          lambda {
            http_request(:get, "http://www.google.com/",
              :headers => {'A' => 'a', 'B' => 'b'}
            )
            request(:get, "www.google.com").
            with(:headers => {'A' => 'a'}).should have_been_made
          }.should_not raise_error
        end

        it "should succeed if request was executed with body and headers but they were not specified in expectantion" do
          lambda {
            http_request(:get, "http://www.google.com/",
              :body => "abc",
              :headers => SAMPLE_HEADERS
            )
            request(:get, "www.google.com").should have_been_made
          }.should_not raise_error
        end


        describe "with authentication" do
          before(:each) do
            stub_http_request(:any, "http://user:pass@www.google.com")
            stub_http_request(:any, "http://user:pazz@www.google.com")
          end

          it "should succeed if succeed if request was executed with expected credentials" do
            lambda {
              http_request(:get, "http://user:pass@www.google.com/")
              request(:get, "http://user:pass@www.google.com").should have_been_made.once
            }.should_not raise_error
          end

          it "should fail if request was executed with different credentials than expected" do
            lambda {
              http_request(:get, "http://user:pass@www.google.com/")
              request(:get, "http://user:pazz@www.google.com").should have_been_made.once
            }.should fail_with("The request GET http://user:pazz@www.google.com/ was expected to execute 1 time but it executed 0 times")
          end

          it "should fail if request was executed without credentials but credentials were expected" do
            lambda {
              http_request(:get, "http://www.google.com/")
              request(:get, "http://user:pass@www.google.com").should have_been_made.once
            }.should fail_with("The request GET http://user:pass@www.google.com/ was expected to execute 1 time but it executed 0 times")
          end

          it "should fail if request was executed with credentials but expected without" do
            lambda {
              http_request(:get, "http://user:pass@www.google.com/")
              request(:get, "http://www.google.com").should have_been_made.once
            }.should fail_with("The request GET http://www.google.com/ was expected to execute 1 time but it executed 0 times")
          end

        end

        describe "using webmock matcher" do

          it "should verify expected requests occured" do
            lambda {
              http_request(:get, "http://www.google.com/")
              WebMock.should have_requested(:get, "http://www.google.com").once
            }.should_not raise_error
          end

          it "should verify expected requests occured" do
            lambda {
              http_request(:get, "http://www.google.com/", :body => "abc", :headers => {'A' => 'a'})
              WebMock.should have_requested(:get, "http://www.google.com").with(:body => "abc", :headers => {'A' => 'a'}).once
            }.should_not raise_error
          end

          it "should verify that non expected requests didn't occur" do
            lambda {
              http_request(:get, "http://www.google.com/")
              WebMock.should_not have_requested(:get, "http://www.google.com")
            }.should fail_with("The request GET http://www.google.com/ was expected to execute 0 times but it executed 1 time")
          end
        end
        


        describe "using assert_requested" do

          it "should verify expected requests occured" do
            lambda {
              http_request(:get, "http://www.google.com/")
              assert_requested(:get, "http://www.google.com", :times => 1)
              assert_requested(:get, "http://www.google.com")
            }.should_not raise_error
          end

          it "should verify expected requests occured" do
            lambda {
              http_request(:get, "http://www.google.com/", :body => "abc", :headers => {'A' => 'a'})
              assert_requested(:get, "http://www.google.com", :body => "abc", :headers => {'A' => 'a'})
            }.should_not raise_error
          end

          it "should verify that non expected requests didn't occur" do
            lambda {
              http_request(:get, "http://www.google.com/")
              assert_not_requested(:get, "http://www.google.com")
            }.should fail_with("The request GET http://www.google.com/ was expected to execute 0 times but it executed 1 time")
          end
        end
      end


      describe "when net connect allowed" do
        before(:each) do
          WebMock.allow_net_connect!
        end

        it "should verify expected requests occured" do
          setup_expectations_for_real_google_request
          lambda {
            http_request(:get, "http://www.google.com/")
            request(:get, "http://www.google.com").should have_been_made
          }.should_not raise_error
        end

        it "should verify that non expected requests didn't occur" do
          lambda {
            http_request(:get, "http://www.google.com/")
            request(:get, "http://www.google.com").should_not have_been_made
          }.should fail_with("The request GET http://www.google.com/ was expected to execute 0 times but it executed 1 time")
        end
      end

    end

  end