spec/document_spec.rb in metainspector-2.1.0 vs spec/document_spec.rb in metainspector-2.2.0

@@ -87,6 +87,32 @@
       tar_url = MetaInspector::Document.new('http://pagerankalert.com/file.tar.gz', html_content_only: true, exception_log: logger)
 
       tar_url.title
     end
   end
+
+  describe 'headers' do
+    it "should include default headers" do
+      url     = 'http://example.com/headers'
+      request = double('Request', base_uri: url)
+      expected_headers = {'User-Agent' => "MetaInspector/#{MetaInspector::VERSION} (+https://github.com/jaimeiniesta/metainspector)"}
+
+      MetaInspector::Request.any_instance.should_receive(:open)
+                                         .with(url, expected_headers)
+                                         .and_return(request)
+
+      MetaInspector::Document.new(url)
+    end
+
+    it "should include passed headers on the request" do
+      url     = 'http://example.com/headers'
+      headers = {'User-Agent' => 'Mozilla', 'Referer' => 'https://github.com/'}
+      request = double('Request', base_uri: url)
+
+      MetaInspector::Request.any_instance.should_receive(:open)
+                                         .with(url, headers)
+                                         .and_return(request)
+
+      MetaInspector::Document.new(url, headers: headers)
+    end
+  end
 end