diff --git a/spec/mechanize_spec.cr b/spec/mechanize_spec.cr index a32718f..f573f39 100644 --- a/spec/mechanize_spec.cr +++ b/spec/mechanize_spec.cr @@ -1,6 +1,7 @@ require "./spec_helper" require "webmock" -WebMock.stub(:any, "example.com") +WebMock.stub(:get, "example.com") +WebMock.stub(:get, "http://example.com/?foo=bar&foo1=bar2") describe Mechanize do it "simple GET" do @@ -12,7 +13,6 @@ describe Mechanize do end it "GET with query parameter" do - WebMock.stub(:get, "http://example.com/?foo=bar&foo1=bar2") agent = Mechanize.new params = {"foo" => "bar", "foo1" => "bar2"} uri = "http://example.com/" @@ -20,4 +20,22 @@ describe Mechanize do page.code.should eq 200 page.uri.to_s.should eq "http://example.com/?foo=bar&foo1=bar2" end + + it "GET with query parameter as URL string" do + agent = Mechanize.new + uri = "http://example.com/?foo=bar&foo1=bar2" + page = agent.get(uri) + page.code.should eq 200 + page.uri.to_s.should eq uri + end + + it "set custom request headers" do + agent = Mechanize.new + uri = "http://example.com/" + headers = HTTP::Headers{"Foo" => "Bar"} + agent.request_headers.empty?.should eq true + page = agent.get(uri, headers: headers) + agent.request_headers.empty?.should eq false + agent.request_headers["Foo"].should eq "Bar" + end end