spec/support/test_repository.rb in gitlab-elasticsearch-git-0.0.15 vs spec/support/test_repository.rb in gitlab-elasticsearch-git-0.0.16

@@ -1,9 +1,19 @@
 class TestRepository
   include Elasticsearch::Git::Repository
 
+  CLIENT = Elasticsearch::Client.new(
+    host: (ENV['ELASTICSEARCH_HOST'] || 'localhost')
+  )
+
+  self.__elasticsearch__.client = CLIENT
+
   def initialize(id = 'test_1')
     @repo_id = id
+  end
+
+  def client_for_indexing
+    CLIENT
   end
 
   def repository_id
     @repo_id
   end