1 files changed, 21 insertions, 1 deletions
diff --git a/spec/system/ntp_install_spec.rb b/spec/system/ntp_install_spec.rb
index 434cf58..39759c5 100644
--- a/spec/system/ntp_install_spec.rb
+++ b/spec/system/ntp_install_spec.rb
@@ -1,11 +1,31 @@
 require 'spec_helper_system'
 
+
 describe 'ntp::install class' do
   let(:os) {
     node.facts['osfamily']
   }
 
-  describe package('ntp') do
+  case node.facts['osfamily']
+  when 'FreeBSD'
+    packagename = 'net/ntp'
+  when 'Linux'
+    case node.facts['operatingsystem']
+    when 'ArchLinux'
+      packagename = 'ntp'
+    when 'Gentoo'
+      packagename = 'net-misc/ntp'
+    end
+  else
+    packagename = 'ntp'
+  end
+
+  puppet_apply(%{
+    class { 'ntp': }
+  })
+
+  describe package(packagename) do
     it { should be_installed }
   end
+
 end