Add optional config parameter.

1 files changed, 9 insertions, 3 deletions

diff --git a/lib/chef/search/query.rb b/lib/chef/search/query.rb
index 23763ca59f..1d0e4ee122 100644
--- a/lib/chef/search/query.rb
+++ b/lib/chef/search/query.rb
@@ -27,10 +27,16 @@ class Chef
   class Search
     class Query
 
-      attr_accessor :rest
+      attr_accessor :config, :rest
+      private :config
 
-      def initialize(url=nil)
-        @rest = Chef::REST.new(url || Chef::Config[:chef_server_url])
+      def initialize(url=nil, config:Chef::Config)
+        @config = config
+        @url = url
+      end
+
+      def rest
+        @rest ||= Chef::REST.new(@url || @config[:chef_server_url])
       end
 
       # Backwards compatability for cookbooks.