Merge pull request #4 from pondohva/package_providers

enable gem and pip providers for package auto_spec
raphink · Sep 9, 2015 · 403daee · 403daee
2 parents fd83d15 + 04fb6b9
commit 403daee
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/puppetx/policy/auto_spec/package.rb b/lib/puppetx/policy/auto_spec/package.rb
@@ -1,4 +1,6 @@
 Puppetx::Policy::AutoSpec.newspec 'Package' do |p|
+  providers = ['gem', 'pip']
   should = (p[:ensure] == 'absent' or p[:ensure] == 'purged') ? 'should_not' : 'should'
-  "  describe package('#{p[:name]}') do\n    it { #{should} be_installed }\n  end\n"
+  provider = providers.include? p[:provider] ? ".by('#{p[:provider]}')" : ''
+  "  describe package('#{p[:name]}') do\n    it { #{should} be_installed#{provider} }\n  end\n"
 end