require 'spec_helper' 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