diff options
author | Adam Edwards <adamed@opscode.com> | 2014-05-15 13:20:40 -0700 |
---|---|---|
committer | Adam Edwards <adamed@opscode.com> | 2014-05-15 13:20:40 -0700 |
commit | 28c2d33c159644c6464f411697518c4db8e267fd (patch) | |
tree | 0e72eadb96403d152b0e6556314659148157578c /lib | |
parent | ac87ed20c4a87e49b14aa7b711cd043243a0317a (diff) | |
download | wmi-lite-28c2d33c159644c6464f411697518c4db8e267fd.tar.gz |
Functional tests, raise exceptions on query failures
Diffstat (limited to 'lib')
-rw-r--r-- | lib/wmi-lite/wmi.rb | 11 | ||||
-rw-r--r-- | lib/wmi-lite/wmi_instance.rb | 2 |
2 files changed, 11 insertions, 2 deletions
diff --git a/lib/wmi-lite/wmi.rb b/lib/wmi-lite/wmi.rb index 61fd0c3..1acc9bd 100644 --- a/lib/wmi-lite/wmi.rb +++ b/lib/wmi-lite/wmi.rb @@ -54,7 +54,16 @@ module WmiLite private def start_query(wql_query) - @connection.ExecQuery(wql_query) + result = @connection.ExecQuery(wql_query) + raise_if_failed(result) + result + end + + def raise_if_failed(result) + # Attempting to access the count property of the underlying + # COM (OLE) object will trigger an exception if the query + # was unsuccessful. + result.count end def new_connection(namespace) diff --git a/lib/wmi-lite/wmi_instance.rb b/lib/wmi-lite/wmi_instance.rb index a6f7256..ff0e9c6 100644 --- a/lib/wmi-lite/wmi_instance.rb +++ b/lib/wmi-lite/wmi_instance.rb @@ -27,7 +27,7 @@ module WmiLite end def [](key) - @property_map[key] + @property_map[key.downcase] end private |