require 'spec_helper' require 'tempfile' provider_class = Puppet::Type.type(:file_line).provider(:ruby) describe provider_class do context "when adding" do let :tmpfile do tmp = Tempfile.new('tmp') path = tmp.path tmp.close! path end let :resource do Puppet::Type::File_line.new( {:name => 'foo', :path => tmpfile, :line => 'foo'} ) end let :provider do 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 describe 'using match' do 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 all lines that matches' do @resource = Puppet::Type::File_line.new( { :name => 'foo', :path => @tmpfile, :line => 'foo = bar', :match => '^foo\s*=.*$', :multiple => true } ) @provider = provider_class.new(@resource) File.open(@tmpfile, 'w') do |fh| fh.write("foo1\nfoo=blah\nfoo2\nfoo=baz") end @provider.exists?.should be_nil @provider.create File.read(@tmpfile).chomp.should eql("foo1\nfoo = bar\nfoo2\nfoo = bar") end it 'should raise an error with invalid values' do expect { @resource = Puppet::Type::File_line.new( { :name => 'foo', :path => @tmpfile, :line => 'foo = bar', :match => '^foo\s*=.*$', :multiple => 'asgadga' } ) }.to raise_error(Puppet::Error, /Invalid value "asgadga"\. Valid values are true, false\./) 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 describe 'using after' do let :resource do Puppet::Type::File_line.new( { :name => 'foo', :path => @tmpfile, :line => 'inserted = line', :after => '^foo1', } ) end let :provider do provider_class.new(resource) end context 'with one line matching the after expression' do before :each do File.open(@tmpfile, 'w') do |fh| fh.write("foo1\nfoo = blah\nfoo2\nfoo = baz") end end it 'inserts the specified line after the line matching the "after" expression' do provider.create File.read(@tmpfile).chomp.should eql("foo1\ninserted = line\nfoo = blah\nfoo2\nfoo = baz") end end context 'with two lines matching the after expression' do before :each do File.open(@tmpfile, 'w') do |fh| fh.write("foo1\nfoo = blah\nfoo2\nfoo1\nfoo = baz") end end it 'errors out stating "One or no line must match the pattern"' do expect { provider.create }.to raise_error(Puppet::Error, /One or no line must match the pattern/) end end context 'with no lines matching the after expression' do let :content do "foo3\nfoo = blah\nfoo2\nfoo = baz\n" end before :each do File.open(@tmpfile, 'w') do |fh| fh.write(content) end end it 'appends the specified line to the file' do provider.create File.read(@tmpfile).should eq(content << resource[:line] << "\n") end end 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