9e0256aabf
This commit adds a new parameter called "match" to the file_line resource type, and support for this new parameter to the corresponding ruby provider. This parameter is optional; file_line should work just as before if you do not specify this parameter... so this change should be backwards-compatible. If you do specify the parameter, it is treated as a regular expression that should be used when looking through the file for a line. This allows you to do things like find a line that begins with a certain prefix (e.g., "foo=.*"), and *replace* the existing line with the line you specify in your "line" parameter. Without this capability, if you already had a line "foo=bar" in your file and your "line" parameter was set to "foo=baz", you'd end up with *both* lines in the final file. In many cases this is undesirable.
127 lines
4.1 KiB
Ruby
127 lines
4.1 KiB
Ruby
require 'puppet'
|
|
require 'tempfile'
|
|
provider_class = Puppet::Type.type(:file_line).provider(:ruby)
|
|
describe provider_class do
|
|
context "when adding" do
|
|
before :each do
|
|
# TODO: these should be ported over to use the PuppetLabs spec_helper
|
|
# file fixtures once the following pull request has been merged:
|
|
# https://github.com/puppetlabs/puppetlabs-stdlib/pull/73/files
|
|
tmp = Tempfile.new('tmp')
|
|
@tmpfile = tmp.path
|
|
tmp.close!
|
|
@resource = Puppet::Type::File_line.new(
|
|
{:name => 'foo', :path => @tmpfile, :line => 'foo'}
|
|
)
|
|
@provider = provider_class.new(@resource)
|
|
end
|
|
it 'should detect if the line exists in the file' do
|
|
File.open(@tmpfile, 'w') do |fh|
|
|
fh.write('foo')
|
|
end
|
|
@provider.exists?.should be_true
|
|
end
|
|
it 'should detect if the line does not exist in the file' do
|
|
File.open(@tmpfile, 'w') do |fh|
|
|
fh.write('foo1')
|
|
end
|
|
@provider.exists?.should be_nil
|
|
end
|
|
it 'should append to an existing file when creating' do
|
|
@provider.create
|
|
File.read(@tmpfile).chomp.should == 'foo'
|
|
end
|
|
end
|
|
|
|
context "when matching" do
|
|
before :each do
|
|
# TODO: these should be ported over to use the PuppetLabs spec_helper
|
|
# file fixtures once the following pull request has been merged:
|
|
# https://github.com/puppetlabs/puppetlabs-stdlib/pull/73/files
|
|
tmp = Tempfile.new('tmp')
|
|
@tmpfile = tmp.path
|
|
tmp.close!
|
|
@resource = Puppet::Type::File_line.new(
|
|
{
|
|
:name => 'foo',
|
|
:path => @tmpfile,
|
|
:line => 'foo = bar',
|
|
:match => '^foo\s*=.*$',
|
|
}
|
|
)
|
|
@provider = provider_class.new(@resource)
|
|
end
|
|
|
|
it 'should raise an error if more than one line matches, and should not have modified the file' do
|
|
File.open(@tmpfile, 'w') do |fh|
|
|
fh.write("foo1\nfoo=blah\nfoo2\nfoo=baz")
|
|
end
|
|
@provider.exists?.should be_nil
|
|
expect { @provider.create }.to raise_error(Puppet::Error, /More than one line.*matches/)
|
|
File.read(@tmpfile).should eql("foo1\nfoo=blah\nfoo2\nfoo=baz")
|
|
end
|
|
|
|
it 'should replace a line that matches' do
|
|
File.open(@tmpfile, 'w') do |fh|
|
|
fh.write("foo1\nfoo=blah\nfoo2")
|
|
end
|
|
@provider.exists?.should be_nil
|
|
@provider.create
|
|
File.read(@tmpfile).chomp.should eql("foo1\nfoo = bar\nfoo2")
|
|
end
|
|
it 'should add a new line if no lines match' do
|
|
File.open(@tmpfile, 'w') do |fh|
|
|
fh.write("foo1\nfoo2")
|
|
end
|
|
@provider.exists?.should be_nil
|
|
@provider.create
|
|
File.read(@tmpfile).should eql("foo1\nfoo2\nfoo = bar\n")
|
|
end
|
|
it 'should do nothing if the exact line already exists' do
|
|
File.open(@tmpfile, 'w') do |fh|
|
|
fh.write("foo1\nfoo = bar\nfoo2")
|
|
end
|
|
@provider.exists?.should be_true
|
|
@provider.create
|
|
File.read(@tmpfile).chomp.should eql("foo1\nfoo = bar\nfoo2")
|
|
end
|
|
end
|
|
|
|
context "when removing" do
|
|
before :each do
|
|
# TODO: these should be ported over to use the PuppetLabs spec_helper
|
|
# file fixtures once the following pull request has been merged:
|
|
# https://github.com/puppetlabs/puppetlabs-stdlib/pull/73/files
|
|
tmp = Tempfile.new('tmp')
|
|
@tmpfile = tmp.path
|
|
tmp.close!
|
|
@resource = Puppet::Type::File_line.new(
|
|
{:name => 'foo', :path => @tmpfile, :line => 'foo', :ensure => 'absent' }
|
|
)
|
|
@provider = provider_class.new(@resource)
|
|
end
|
|
it 'should remove the line if it exists' do
|
|
File.open(@tmpfile, 'w') do |fh|
|
|
fh.write("foo1\nfoo\nfoo2")
|
|
end
|
|
@provider.destroy
|
|
File.read(@tmpfile).should eql("foo1\nfoo2")
|
|
end
|
|
|
|
it 'should remove the line without touching the last new line' do
|
|
File.open(@tmpfile, 'w') do |fh|
|
|
fh.write("foo1\nfoo\nfoo2\n")
|
|
end
|
|
@provider.destroy
|
|
File.read(@tmpfile).should eql("foo1\nfoo2\n")
|
|
end
|
|
|
|
it 'should remove any occurence of the line' do
|
|
File.open(@tmpfile, 'w') do |fh|
|
|
fh.write("foo1\nfoo\nfoo2\nfoo\nfoo")
|
|
end
|
|
@provider.destroy
|
|
File.read(@tmpfile).should eql("foo1\nfoo2\n")
|
|
end
|
|
end
|
|
end
|