File: exec_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 (63 lines) | stat: -rw-r--r-- 2,113 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
require 'spec_helper'

require 'puppet/indirector/exec'

describe Puppet::Indirector::Exec do
  before :all do
    class Puppet::ExecTestModel
      extend Puppet::Indirector
      indirects :exec_test_model
    end

    class Puppet::ExecTestModel::Exec < Puppet::Indirector::Exec
      attr_accessor :command
    end

    Puppet::ExecTestModel.indirection.terminus_class = :exec
  end

  after(:all) do
    Puppet::ExecTestModel.indirection.delete
    Puppet.send(:remove_const, :ExecTestModel)
  end

  let(:terminus) { Puppet::ExecTestModel.indirection.terminus(:exec) }
  let(:indirection) { Puppet::ExecTestModel.indirection }
  let(:model) { Puppet::ExecTestModel }
  let(:path) { File.expand_path('/echo') }
  let(:arguments) { {:failonfail => true, :combine => false } }

  before(:each) { terminus.command = [path] }

  it "should throw an exception if the command is not an array" do
    terminus.command = path
    expect { indirection.find('foo') }.to raise_error(Puppet::DevError)
  end

  it "should throw an exception if the command is not fully qualified" do
    terminus.command = ["mycommand"]
    expect { indirection.find('foo') }.to raise_error(ArgumentError)
  end

  it "should execute the command with the object name as the only argument" do
    expect(terminus).to receive(:execute).with([path, 'foo'], arguments)
    indirection.find('foo')
  end

  it "should return the output of the script" do
    expect(terminus).to receive(:execute).with([path, 'foo'], arguments).and_return("whatever")
    expect(indirection.find('foo')).to eq("whatever")
  end

  it "should return nil when the command produces no output" do
    expect(terminus).to receive(:execute).with([path, 'foo'], arguments).and_return(nil)
    expect(indirection.find('foo')).to be_nil
  end

  it "should raise an exception if there's an execution failure" do
    expect(terminus).to receive(:execute).with([path, 'foo'], arguments).and_raise(Puppet::ExecutionFailure.new("message"))
    expect {
      indirection.find('foo')
    }.to raise_exception(Puppet::Error, 'Failed to find foo via exec: message')
  end
end