using head requests for alma and wowza okcomputer

Author: jason-raitz

config/initializers/okcomputer.rb

@@ -26,13 +26,29 @@ def check
   end
 end
 
+class HeadCheck < OkComputer::HttpCheck
+  def perform_request
+    Timeout.timeout(request_timeout) do
+      options = { read_timeout: request_timeout }
+
+      if basic_auth_options.any?
+        options[:http_basic_authentication] = basic_auth_options
+      end
+
+      RestClient.head(url.to_s, options)
+    end
+  rescue => e
+    raise ConnectionFailed, e
+  end
+end
+
 # Ensure Alma API is working.
-OkComputer::Registry.register 'alma-metadata', OkComputer::HttpCheck.new(ALMA_TEST_URL)
+OkComputer::Registry.register 'alma-metadata', HeadCheck.new(ALMA_TEST_URL)
 
 # Ensure TIND API is working. This cannot use `OkComputer::HttpCheck`
 # out of the box as we can't yet inject headers into the request without
 # subclassing the whole thing.
 OkComputer::Registry.register 'tind-metadata', TindCheck.new
 
 # Ensure Wowza is working
-OkComputer::Registry.register 'wowza-streaming', OkComputer::HttpCheck.new(WOWZA_TEST_URL)
+OkComputer::Registry.register 'wowza-streaming', HeadCheck.new(WOWZA_TEST_URL)

spec/request/okcomputer_spec.rb

@@ -2,11 +2,11 @@
 
 RSpec.describe 'OKComputer', type: :request do
   before do
-    stub_sru_request('b23305522')
+    stub_sru_head_request('b23305522')
 
     hls_manifest = File.read('spec/data/playlist.m3u8')
     manifest_url = 'https://wowza.lib.berkeley.edu/Pacifica/mp3:PRA_NHPRC1_AZ1084_00_000_00.mp3/playlist.m3u8'
-    stub_request(:get, manifest_url).to_return(status: 200, body: hls_manifest)
+    stub_request(:head, manifest_url).to_return(status: 200)
 
     record_id = '(pacradio)01469'
     tind_url = BerkeleyLibrary::AV::Metadata::Source::TIND.marc_uri_for(record_id)

spec/spec_helper.rb

@@ -72,6 +72,11 @@ def stub_sru_request(record_id, body: nil)
   stub_request(:get, sru_url).to_return(status: 200, body: body || File.new(alma_sru_data_path_for(record_id)))
 end
 
+def stub_sru_head_request(record_id, body: nil)
+  sru_url = alma_sru_url_for(record_id)
+  stub_request(:head, sru_url).to_return(status: 200)
+end
+
 def alma_marc_record_for(record_id)
   marc_xml_path = alma_sru_data_path_for(record_id)
   MARC::XMLReader.new(marc_xml_path).first