1 files changed, 159 insertions, 0 deletions
diff --git a/spec/unit/util/backups_spec.rb b/spec/unit/util/backups_spec.rb
new file mode 100755
index 000000000..fa6756e10
--- /dev/null
+++ b/spec/unit/util/backups_spec.rb
@@ -0,0 +1,159 @@
+#!/usr/bin/env ruby
+
+require File.dirname(__FILE__) + '/../../spec_helper'
+
+require 'puppet/util/backups'
+include PuppetTest
+
+describe Puppet::Util::Backups do
+    before do
+        FileTest.stubs(:exists?).returns true
+    end
+
+    describe "when backing up a file" do
+        it "should noop if the file does not exist" do
+            FileTest.expects(:exists?).returns false
+            file = Puppet::Type.type(:file).new(:name => '/no/such/file')
+            file.expects(:bucket).never
+
+            file.perform_backup
+        end
+
+        it "should succeed silently if self[:backup] is false" do
+            file = Puppet::Type.type(:file).new(:name => '/no/such/file', :backup => false)
+            file.expects(:bucket).never
+            FileTest.expects(:exists?).never
+            file.perform_backup
+        end
+
+        it "a bucket should be used when provided" do
+            path = '/my/file'
+
+            File.stubs(:stat).with(path).returns(mock('stat', :ftype => 'file'))
+
+            file = Puppet::Type.type(:file).new(:name => path, :backup => 'foo')
+            bucket = stub('bucket', 'name' => 'foo')
+            file.stubs(:bucket).returns bucket
+
+            bucket.expects(:backup).with(path).returns("mysum")
+
+            file.perform_backup
+        end
+
+        it "should propagate any exceptions encountered when backing up to a filebucket" do
+            path = '/my/file'
+
+            File.stubs(:stat).with(path).returns(mock('stat', :ftype => 'file'))
+
+            file = Puppet::Type.type(:file).new(:name => path, :backup => 'foo')
+            bucket = stub('bucket', 'name' => 'foo')
+            file.stubs(:bucket).returns bucket
+
+            bucket.expects(:backup).raises ArgumentError
+
+            lambda { file.perform_backup }.should raise_error(ArgumentError)
+        end
+
+        describe "and no filebucket is configured" do
+            it "should remove any local backup if one exists" do
+                path = '/my/file'
+                FileTest.stubs(:exists?).returns true
+
+                backup = path + ".foo"
+
+                File.expects(:lstat).with(backup).returns stub("stat", :ftype => "file")
+                File.expects(:unlink).with(backup)
+
+                FileUtils.stubs(:cp_r)
+
+                file = Puppet::Type.type(:file).new(:name => path, :backup => '.foo')
+                file.perform_backup
+            end
+
+            it "should fail when the old backup can't be removed" do
+                path = '/my/file'
+                FileTest.stubs(:exists?).returns true
+
+                backup = path + ".foo"
+
+                File.expects(:lstat).with(backup).returns stub("stat", :ftype => "file")
+                File.expects(:unlink).raises ArgumentError
+
+                FileUtils.expects(:cp_r).never
+
+                file = Puppet::Type.type(:file).new(:name => path, :backup => '.foo')
+                lambda { file.perform_backup }.should raise_error(Puppet::Error)
+            end
+
+            it "should not try to remove backups that don't exist" do
+                path = '/my/file'
+                FileTest.stubs(:exists?).returns true
+
+                backup = path + ".foo"
+
+                File.expects(:lstat).with(backup).raises(Errno::ENOENT)
+                File.expects(:unlink).never
+
+                FileUtils.stubs(:cp_r)
+
+                file = Puppet::Type.type(:file).new(:name => path, :backup => '.foo')
+                file.perform_backup
+            end
+
+            it "a copy should be created in the local directory" do
+                path = '/my/file'
+                FileTest.stubs(:exists?).with(path).returns true
+
+                FileUtils.expects(:cp_r).with(path, path + ".foo", :preserve => true)
+
+                file = Puppet::Type.type(:file).new(:name => path, :backup => '.foo')
+                file.perform_backup.should be_true
+            end
+
+            it "should propagate exceptions if no backup can be created" do
+                path = '/my/file'
+                FileTest.stubs(:exists?).with(path).returns true
+
+                FileUtils.expects(:cp_r).raises ArgumentError
+
+                file = Puppet::Type.type(:file).new(:name => path, :backup => '.foo')
+                lambda { file.perform_backup }.should raise_error(Puppet::Error)
+            end
+        end
+    end
+
+    describe "when backing up a directory" do
+        it "a bucket should work when provided" do
+            path = '/my/dir'
+
+            File.stubs(:file?).returns true
+            Find.expects(:find).with(path).yields("/my/dir/file")
+
+            bucket = stub('bucket', :name => "eh")
+            bucket.expects(:backup).with("/my/dir/file").returns true
+
+            file = Puppet::Type.type(:file).new(:name => path, :backup => 'foo')
+            file.stubs(:bucket).returns bucket
+
+            File.stubs(:stat).with(path).returns(stub('stat', :ftype => 'directory'))
+
+            file.perform_backup
+        end
+
+        it "should do nothing when recursing" do
+            path = '/my/dir'
+
+            bucket = stub('bucket', :name => "eh")
+            bucket.expects(:backup).never
+
+            file = Puppet::Type.type(:file).new(:name => path, :backup => 'foo', :recurse => true)
+            file.stubs(:bucket).returns bucket
+
+            File.stubs(:stat).with(path).returns(stub('stat', :ftype => 'directory'))
+
+            Find.expects(:find).never
+
+            file.perform_backup
+        end
+    end
+end