lib/wmi-lite/wmi.rb in wmi-lite-1.0.2 vs lib/wmi-lite/wmi.rb in wmi-lite-1.0.5

@@ -1,8 +1,8 @@
 #
 # Author:: Adam Edwards (<[e-mail]>)
-# Copyright:: Copyright 2014 Chef Software, Inc.
+# Copyright:: Copyright 2014-2019 Chef Software, Inc.
 # License:: Apache License, Version 2.0
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -15,17 +15,17 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
 
 require 'win32ole' if RUBY_PLATFORM =~ /mswin|mingw32|windows/
-require 'wmi-lite/wmi_instance'
-require 'wmi-lite/wmi_exception'
+require_relative 'wmi_instance'
+require_relative 'wmi_exception'
 
 module WmiLite
   class Wmi
     def initialize(namespace = nil)
-      @namespace = namespace.nil? ? 'root/cimv2' : namespace
+      @namespace = namespace.nil? ? "root/cimv2" : namespace
       @connection = nil
     end
 
     def query(wql_query)
       query_with_context(wql_query)
@@ -36,11 +36,11 @@
     end
 
     def first_of(wmi_class)
       query_result = start_query("select * from #{wmi_class}", wmi_class)
       first_result = nil
-      query_result.each do | record |
+      query_result.each do |record|
         first_result = record
         break
       end
       first_result.nil? ? nil : wmi_result_to_snapshot(first_result)
     end
@@ -50,11 +50,11 @@
     def query_with_context(wql_query, diagnostic_class_name = nil)
       results = start_query(wql_query, diagnostic_class_name)
 
       result_set = []
 
-      results.each do | result |
+      results.each do |result|
         result_set.push(wmi_result_to_snapshot(result))
       end
 
       result_set
     end
@@ -78,13 +78,13 @@
       result.count
     end
 
     def connect_to_namespace
       if @connection.nil?
-        namespace = @namespace.nil? ? 'root/cimv2' : @namespace
+        namespace = @namespace.nil? ? "root/cimv2" : @namespace
         locator = WIN32OLE.new("WbemScripting.SWbemLocator")
         begin
-          @connection = locator.ConnectServer('.', namespace)
+          @connection = locator.ConnectServer(".", namespace)
         rescue WIN32OLERuntimeError => native_exception
           raise WmiException.new(native_exception, :ConnectServer, @namespace)
         end
       end
     end