summaryrefslogtreecommitdiff
path: root/puppet/modules/augeas/spec/unit/puppet/parser/functions/augeas_spec.rb
blob: b34fa5b583418d5e34b4cdd0863d7a5132f803c0 (plain)
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
#! /usr/bin/env ruby -S rspec
require 'spec_helper'

describe 'the augeas function' do
  let(:scope) { PuppetlabsSpec::PuppetInternals.scope }

  it "should fail if the augeas feature is not present" do
    Puppet.features.expects(:augeas?).returns(false)
    expect { scope.function_augeas([]) }.to raise_error(Puppet::ParseError, /requires the augeas feature/)
  end

  it "should exist" do
    expect(Puppet::Parser::Functions.function("augeas")).to eq("function_augeas")
  end

  context "when passing wrong arguments" do
    before :each do
      Puppet.features.stubs(:augeas?).returns(true)
    end

    it "should raise a ParseError if there are no arguments" do
      expect { scope.function_augeas([]) }.to raise_error(Puppet::ParseError, /Wrong number of arguments/)
    end

    it "should raise a ParseError if content is not a string" do
      expect { scope.function_augeas([['foo'], 'Fstab.lns', []]) }.to raise_error(Puppet::ParseError, /content must be a string/)
    end

    it "should raise a ParseError if lens is not a string" do
      expect { scope.function_augeas(['foo', ['Fstab.lns'], []]) }.to raise_error(Puppet::ParseError, /lens must be a string/)
    end

    it "should raise a ParseError if changes is not an array" do
      expect { scope.function_augeas(['foo', 'Fstab.lns', 'changes']) }.to raise_error(Puppet::ParseError, /changes must be an array/)
    end
  end

  if Puppet.features.augeas?
    context "when passing invalid input" do
      it "should fail to parse input with lens" do
        expect { scope.function_augeas(['foo', 'Fstab.lns', []]) }.to raise_error(Puppet::ParseError, /Failed to parse string with lens Fstab.lns:/)
      end
    end

    context "when passing illegal changes" do
      it "should fail to apply illegal change" do
        expect { scope.function_augeas(["\n", 'Fstab.lns', ['foo bar']]) }.to raise_error(Puppet::ParseError, /Failed to apply change to tree/)
      end
    end

    context "when generating an invalid tree" do
      it "should fail to apply changes with wrong tree" do
        expect { scope.function_augeas(["\n", 'Fstab.lns', ['set ./1/opt 3']]) }.to raise_error(Puppet::ParseError, /Failed to apply changes with lens Fstab.lns:/)
      end
    end

    context "when applying valid changes" do
      it "should remove the 3rd option" do
        result = scope.function_augeas(["proc        /proc   proc    nodev,noexec,nosuid     0       0\n", 'Fstab.lns', ['rm ./1/opt[3]']])
        expect(result.class).to eq(String)
        #result.should == "proc       /proc   proc    nodev,noexec     0       0\n"
      end

      it "should set a 4th option" do
        result = scope.function_augeas(["proc        /proc   proc    nodev,noexec,nosuid     0       0\n", 'Fstab.lns', ['ins opt after ./1/opt[last()]', 'set ./1/opt[last()] nofoo']])
        expect(result.class).to eq(String)
        #result.should == "proc       /proc   proc    nodev,noexec,nosuid,nofoo     0       0\n"
      end
    end

    context "when using old libs" do
      it "should not work with Augeas prior to 1.0.0" do
        Augeas.any_instance.expects(:get).with('/augeas/version').returns('0.10.0')
        expect { scope.function_augeas(["\n", 'Fstab.lns', []]) }.to raise_error(Puppet::ParseError, /requires Augeas 1\.0\.0/)
      end

      it "should not work with ruby-augeas prior to 0.5.0" do
        Augeas.any_instance.expects(:methods).returns([])
        expect { scope.function_augeas(["\n", 'Fstab.lns', []]) }.to raise_error(Puppet::ParseError, /requires ruby-augeas 0\.5\.0/)
      end
    end
  end
end