File: backups_spec.rb

package info (click to toggle)
puppet-agent 7.23.0-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 19,092 kB
  • sloc: ruby: 245,074; sh: 456; makefile: 38; xml: 33
file content (130 lines) | stat: -rw-r--r-- 4,623 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
require 'spec_helper'

require 'puppet/util/backups'

describe Puppet::Util::Backups do
  include PuppetSpec::Files

  let(:bucket) { double('bucket', :name => "foo") }
  let!(:file) do
    f = Puppet::Type.type(:file).new(:name => path, :backup => 'foo')
    allow(f).to receive(:bucket).and_return(bucket)
    f
  end

  describe "when backing up a file" do
    let(:path) { make_absolute('/no/such/file') }

    it "should noop if the file does not exist" do
      expect(file).not_to receive(:bucket)
      expect(Puppet::FileSystem).to receive(:exist?).with(path).and_return(false)

      file.perform_backup
    end

    it "should succeed silently if self[:backup] is false" do
      file = Puppet::Type.type(:file).new(:name => path, :backup => false)

      expect(file).not_to receive(:bucket)
      expect(Puppet::FileSystem).not_to receive(:exist?)

      file.perform_backup
    end

    it "a bucket should be used when provided" do
      lstat_path_as(path, 'file')
      expect(bucket).to receive(:backup).with(path).and_return("mysum")
      expect(Puppet::FileSystem).to receive(:exist?).with(path).and_return(true)

      file.perform_backup
    end

    it "should propagate any exceptions encountered when backing up to a filebucket" do
      lstat_path_as(path, 'file')
      expect(bucket).to receive(:backup).and_raise(ArgumentError)
      expect(Puppet::FileSystem).to receive(:exist?).with(path).and_return(true)

      expect { file.perform_backup }.to raise_error(ArgumentError)
    end

    describe "and local backup is configured" do
      let(:ext) { 'foobkp' }
      let(:backup) { path + '.' + ext }
      let(:file) { Puppet::Type.type(:file).new(:name => path, :backup => '.'+ext) }

      it "should remove any local backup if one exists" do
        lstat_path_as(backup, 'file')
        expect(Puppet::FileSystem).to receive(:unlink).with(backup)
        allow(FileUtils).to receive(:cp_r)
        expect(Puppet::FileSystem).to receive(:exist?).with(path).and_return(true)

        file.perform_backup
      end

      it "should fail when the old backup can't be removed" do
        lstat_path_as(backup, 'file')
        expect(Puppet::FileSystem).to receive(:unlink).with(backup).and_raise(ArgumentError)
        expect(FileUtils).not_to receive(:cp_r)
        expect(Puppet::FileSystem).to receive(:exist?).with(path).and_return(true)

        expect { file.perform_backup }.to raise_error(Puppet::Error)
      end

      it "should not try to remove backups that don't exist" do
        expect(Puppet::FileSystem).to receive(:lstat).with(backup).and_raise(Errno::ENOENT)
        expect(Puppet::FileSystem).not_to receive(:unlink).with(backup)
        allow(FileUtils).to receive(:cp_r)
        expect(Puppet::FileSystem).to receive(:exist?).with(path).and_return(true)

        file.perform_backup
      end

      it "a copy should be created in the local directory" do
        expect(FileUtils).to receive(:cp_r).with(path, backup, {:preserve => true})
        allow(Puppet::FileSystem).to receive(:exist?).with(path).and_return(true)

        expect(file.perform_backup).to be_truthy
      end

      it "should propagate exceptions if no backup can be created" do
        expect(FileUtils).to receive(:cp_r).and_raise(ArgumentError)

        allow(Puppet::FileSystem).to receive(:exist?).with(path).and_return(true)
        expect { file.perform_backup }.to raise_error(Puppet::Error)
      end
    end
  end

  describe "when backing up a directory" do
    let(:path) { make_absolute('/my/dir') }
    let(:filename) { File.join(path, 'file') }

    it "a bucket should work when provided" do
      allow(File).to receive(:file?).with(filename).and_return(true)
      expect(Find).to receive(:find).with(path).and_yield(filename)

      expect(bucket).to receive(:backup).with(filename).and_return(true)

      lstat_path_as(path, 'directory')

      allow(Puppet::FileSystem).to receive(:exist?).with(path).and_return(true)
      allow(Puppet::FileSystem).to receive(:exist?).with(filename).and_return(true)

      file.perform_backup
    end

    it "should do nothing when recursing" do
      file = Puppet::Type.type(:file).new(:name => path, :backup => 'foo', :recurse => true)

      expect(bucket).not_to receive(:backup)
      allow(Puppet::FileSystem).to receive(:stat).with(path).and_return(double('stat', :ftype => 'directory'))
      expect(Find).not_to receive(:find)

      file.perform_backup
    end
  end

  def lstat_path_as(path, ftype)
    expect(Puppet::FileSystem).to receive(:lstat).with(path).and_return(double('File::Stat', :ftype => ftype))
  end
end