Add spec for digital ocean DMI data flow

2 files changed, 9 insertions, 1 deletions

diff --git a/lib/ohai/plugins/digital_ocean.rb b/lib/ohai/plugins/digital_ocean.rb
index 08daa600..240c4a10 100644
--- a/lib/ohai/plugins/digital_ocean.rb
+++ b/lib/ohai/plugins/digital_ocean.rb
@@ -52,7 +52,7 @@ Ohai.plugin(:DigitalOcean) do
       Ohai::Log.debug("Plugin Digitalocean: looks_like_digital_ocean? == true")
       digital_ocean Mash.new
       fetch_metadata.each do |k, v|
-        next if k == 'vendor_data' # this may have sensitive data we shouldn't store
+        next if k == "vendor_data" # this may have sensitive data we shouldn't store
         digital_ocean[k] = v
       end
     else
diff --git a/spec/unit/plugins/digital_ocean_spec.rb b/spec/unit/plugins/digital_ocean_spec.rb
index 682ea73a..ad50cbd3 100644
--- a/spec/unit/plugins/digital_ocean_spec.rb
+++ b/spec/unit/plugins/digital_ocean_spec.rb
@@ -189,4 +189,12 @@ describe Ohai::System, "plugin digital_ocean" do
 >>>>>>> Remove debug code/add proper has_do_init check. Attempt to fix failing tests
     end
   end
+
+  describe "with digital_ocean DMI data" do
+    it_should_behave_like "digital_ocean"
+
+    before(:each) do
+      plugin[:dmi] = { :bios => { :all_records => [ { :Vendor => "DigitalOcean" } ] } }
+    end
+  end
 end