Merge pull request #518 from mhaskel/test_fix

[puppet-modules/puppetlabs-apt.git] / spec / unit / facter / apt_reboot_required_spec.rb

diff --git a/spec/unit/facter/apt_reboot_required_spec.rb b/spec/unit/facter/apt_reboot_required_spec.rb
index ab3490a20ff723cf010c03c557cbdc9d50df0807..ed6efd075e63ca89313ca060f470bcacc8e9c5cc 100644 (file)
--- a/spec/unit/facter/apt_reboot_required_spec.rb
+++ b/spec/unit/facter/apt_reboot_required_spec.rb
@@ -6,18 +6,20 @@ describe 'apt_reboot_required fact' do
 
   describe 'if a reboot is required' do
     before {
-      Facter.fact(:osfamily).stubs(:value).returns 'Debian'
+      Facter.fact(:osfamily).expects(:value).at_least(1).returns 'Debian'
       File.stubs(:file?).returns true
+      File.expects(:file?).at_least(1).with('/var/run/reboot-required').returns true
     }
-    it { expect(Facter.fact(:apt_reboot_required).value).to eq true }
+    it { is_expected.to eq true }
   end
 
   describe 'if a reboot is not required' do
     before {
-      Facter.fact(:osfamily).stubs(:value).returns 'Debian'
-      File.stubs(:file?).returns false
+      Facter.fact(:osfamily).expects(:value).at_least(1).returns 'Debian'
+      File.stubs(:file?).returns true
+      File.expects(:file?).at_least(1).with('/var/run/reboot-required').returns false
     }
-    it { expect(Facter.fact(:apt_reboot_required).value).to eq false }
+    it { is_expected.to eq false }
   end
 
 end