Add validation for Primo env

Author: jazairi

app/models/primo_search.rb

@@ -1,14 +1,10 @@
 # Searches Primo Search API and formats results
 #
 class PrimoSearch
-  PRIMO_API_URL = ENV.fetch('PRIMO_API_URL', nil).freeze
-  PRIMO_API_KEY = ENV.fetch('PRIMO_API_KEY', nil)
-  PRIMO_SCOPE = ENV.fetch('PRIMO_SCOPE', nil)
-  PRIMO_TAB = ENV.fetch('PRIMO_TAB', nil)
-  PRIMO_VID = ENV.fetch('PRIMO_VID', nil)
 
   def initialize
-    @primo_http = HTTP.persistent(PRIMO_API_URL)
+    validate_env
+    @primo_http = HTTP.persistent(primo_api_url)
     @results = {}
   end
 
@@ -17,7 +13,7 @@ def search(term, per_page)
     result = @primo_http.timeout(http_timeout)
                       .headers(
                         accept: 'application/json',
-                        Authorization: "apikey #{PRIMO_API_KEY}"
+                        Authorization: "apikey #{primo_api_key}"
                       )
                       .get(url)
 
@@ -28,6 +24,41 @@ def search(term, per_page)
 
   private
 
+  def validate_env
+    missing_vars = []
+    
+    missing_vars << 'PRIMO_API_URL' if primo_api_url.nil?
+    missing_vars << 'PRIMO_API_KEY' if primo_api_key.nil?
+    missing_vars << 'PRIMO_SCOPE' if primo_scope.nil?
+    missing_vars << 'PRIMO_TAB' if primo_tab.nil?
+    missing_vars << 'PRIMO_VID' if primo_vid.nil?
+    
+    return if missing_vars.empty?
+    
+    raise ArgumentError, "Required Primo environment variables are not set: #{missing_vars.join(', ')}"
+  end
+
+  # Environment variable accessors
+  def primo_api_url
+    ENV.fetch('PRIMO_API_URL', nil)
+  end
+
+  def primo_api_key
+    ENV.fetch('PRIMO_API_KEY', nil)
+  end
+
+  def primo_scope
+    ENV.fetch('PRIMO_SCOPE', nil)
+  end
+
+  def primo_tab
+    ENV.fetch('PRIMO_TAB', nil)
+  end
+
+  def primo_vid
+    ENV.fetch('PRIMO_VID', nil)
+  end
+
   # Initial search term sanitization
   def clean_term(term)
     term.strip.tr(' :,', '+').gsub(/\++/, '+')
@@ -36,19 +67,19 @@ def clean_term(term)
   # Constructs the search URL with required parameters for Primo API
   def search_url(term, per_page)
     [
-      PRIMO_API_URL,
+      primo_api_url,
       '/search?q=any,contains,',
       clean_term(term),
       '&vid=',
-      PRIMO_VID,
+      primo_vid,
       '&tab=',
-      PRIMO_TAB,
+      primo_tab,
       '&scope=',
-      PRIMO_SCOPE,
+      primo_scope,
       '&limit=',
       per_page,
       '&apikey=',
-      PRIMO_API_KEY
+      primo_api_key
     ].join
   end
 

test/models/primo_search_test.rb

@@ -1,6 +1,43 @@
 require 'test_helper'
 
 class PrimoSearchTest < ActiveSupport::TestCase
+  # Assumes env is correctly set in .env.test.
+  test 'initializes successfully with all required environment variables' do
+    assert_nothing_raised do
+      PrimoSearch.new
+    end
+  end
+
+  test 'raises error when PRIMO_API_URL is missing' do
+    ClimateControl.modify(PRIMO_API_URL: nil) do
+      error = assert_raises(ArgumentError) do
+        PrimoSearch.new
+      end
+      assert_match(/PRIMO_API_URL/, error.message)
+      assert_match(/Required Primo environment variables are not set/, error.message)
+    end
+  end
+
+  test 'raises error when PRIMO_API_KEY is missing' do
+    ClimateControl.modify(PRIMO_API_KEY: nil) do
+      error = assert_raises(ArgumentError) do
+        PrimoSearch.new
+      end
+      assert_match(/PRIMO_API_KEY/, error.message)
+    end
+  end
+
+  test 'raises error when multiple environment variables are missing' do
+    ClimateControl.modify(PRIMO_API_URL: nil, PRIMO_SCOPE: nil, PRIMO_VID: nil) do
+      error = assert_raises(ArgumentError) do
+        PrimoSearch.new
+      end
+      assert_match(/PRIMO_API_URL/, error.message)
+      assert_match(/PRIMO_SCOPE/, error.message)
+      assert_match(/PRIMO_VID/, error.message)
+    end
+  end
+
   test 'search returns results' do
     VCR.use_cassette('primo_search_success') do
       search = PrimoSearch.new