Add a spec for passing additional DMI IDsdmi_ids

Signed-off-by: Tim Smith <tsmith@chef.io>

1 files changed, 9 insertions, 2 deletions

diff --git a/spec/unit/plugins/dmi_spec.rb b/spec/unit/plugins/dmi_spec.rb
index b7fa44d9..a8074bf0 100644
--- a/spec/unit/plugins/dmi_spec.rb
+++ b/spec/unit/plugins/dmi_spec.rb
@@ -142,8 +142,15 @@ describe Ohai::System, "plugin dmi" do
     end
   end
 
-  it "correctly ignores unwanted data" do
+  it "allows capturing additional DMI data" do
+    Ohai.config[:additional_dmi_ids] = [ 16 ]
     plugin.run
-    expect(plugin[:dmi][:base_board]).not_to have_key(:error_correction_type)
+    expect(plugin[:dmi]).to have_key(:physical_memory_array)
+  end
+
+  it "correctly ignores data in excluded DMI IDs" do
+    expect(plugin).to receive(:shell_out).with("dmidecode").and_return(mock_shell_out(0, stdout, ""))
+    plugin.run
+    expect(plugin[:dmi]).not_to have_key(:physical_memory_array)
   end
 end