Merge pull request #665 from MITLibraries/timx-185-fix-retrieve-index

Fix default index for V2 retrieve queries

Author: Adam Jazairi

app/models/retrieve.rb

@@ -16,7 +16,7 @@ def fetch(id, client, index = nil)
   end
 
   def default_index
-    ENV.fetch('ELASTICSEARCH_INDEX', nil)
+    Flipflop.v2? ? ENV.fetch('OPENSEARCH_INDEX', nil) : ENV.fetch('ELASTICSEARCH_INDEX', nil)
   end
 
   def to_filter(id)

test/models/retrieve_test.rb

@@ -0,0 +1,17 @@
+require 'test_helper'
+
+class RetrieveTest < ActiveSupport::TestCase
+  def setup
+    @test_strategy = Flipflop::FeatureSet.current.test!
+  end
+
+  test 'graphlv1 uses correct index for retrieve queries' do
+    @test_strategy.switch!(:v2, false)
+    assert_equal ENV.fetch('ELASTICSEARCH_INDEX'), Retrieve.new.default_index
+  end
+
+  test 'graphqlv2 uses correct index for retrieve queries' do
+    @test_strategy.switch!(:v2, true)
+    assert_equal ENV.fetch('OPENSEARCH_INDEX'), Retrieve.new.default_index
+  end
+end