fluentd 0.12.40 → 1.6.2

data/test/plugin/test_buffer.rb

@@ -0,0 +1,1233 @@
+require_relative '../helper'
+require 'fluent/plugin/buffer'
+require 'fluent/plugin/buffer/memory_chunk'
+require 'fluent/plugin/compressable'
+require 'fluent/plugin/buffer/chunk'
+require 'fluent/event'
+require 'flexmock/test_unit'
+
+require 'fluent/log'
+require 'fluent/plugin_id'
+
+require 'time'
+
+module FluentPluginBufferTest
+  class DummyOutputPlugin < Fluent::Plugin::Base
+    include Fluent::PluginId
+    include Fluent::PluginLoggerMixin
+  end
+  class DummyMemoryChunkError < StandardError; end
+  class DummyMemoryChunk < Fluent::Plugin::Buffer::MemoryChunk
+    attr_reader :append_count, :rollbacked, :closed, :purged
+    attr_accessor :failing
+    def initialize(metadata, compress: :text)
+      super
+      @append_count = 0
+      @rollbacked = false
+      @closed = false
+      @purged = false
+      @failing = false
+    end
+    def concat(data, size)
+      @append_count += 1
+      raise DummyMemoryChunkError if @failing
+      super
+    end
+    def rollback
+      super
+      @rollbacked = true
+    end
+    def close
+      super
+      @closed = true
+    end
+    def purge
+      super
+      @purged = true
+    end
+  end
+  class DummyPlugin < Fluent::Plugin::Buffer
+    def create_metadata(timekey=nil, tag=nil, variables=nil)
+      Fluent::Plugin::Buffer::Metadata.new(timekey, tag, variables)
+    end
+    def create_chunk(metadata, data)
+      c = FluentPluginBufferTest::DummyMemoryChunk.new(metadata)
+      c.append(data)
+      c.commit
+      c
+    end
+    def create_chunk_es(metadata, es)
+      c = FluentPluginBufferTest::DummyMemoryChunk.new(metadata)
+      c.concat(es.to_msgpack_stream, es.size)
+      c.commit
+      c
+    end
+    def resume
+      dm0 = create_metadata(Time.parse('2016-04-11 16:00:00 +0000').to_i, nil, nil)
+      dm1 = create_metadata(Time.parse('2016-04-11 16:10:00 +0000').to_i, nil, nil)
+      dm2 = create_metadata(Time.parse('2016-04-11 16:20:00 +0000').to_i, nil, nil)
+      dm3 = create_metadata(Time.parse('2016-04-11 16:30:00 +0000').to_i, nil, nil)
+      staged = {
+        dm2 => create_chunk(dm2, ["b" * 100]).staged!,
+        dm3 => create_chunk(dm3, ["c" * 100]).staged!,
+      }
+      queued = [
+        create_chunk(dm0, ["0" * 100]).enqueued!,
+        create_chunk(dm1, ["a" * 100]).enqueued!,
+        create_chunk(dm1, ["a" * 3]).enqueued!,
+      ]
+      return staged, queued
+    end
+    def generate_chunk(metadata)
+      DummyMemoryChunk.new(metadata, compress: @compress)
+    end
+  end
+end
+
+class BufferTest < Test::Unit::TestCase
+  def create_buffer(hash)
+    buffer_conf = config_element('buffer', '', hash, [])
+    owner = FluentPluginBufferTest::DummyOutputPlugin.new
+    owner.configure(config_element('ROOT', '', {}, [ buffer_conf ]))
+    p = FluentPluginBufferTest::DummyPlugin.new
+    p.owner = owner
+    p.configure(buffer_conf)
+    p
+  end
+
+  def create_metadata(timekey=nil, tag=nil, variables=nil)
+    Fluent::Plugin::Buffer::Metadata.new(timekey, tag, variables)
+  end
+
+  def create_chunk(metadata, data)
+    c = FluentPluginBufferTest::DummyMemoryChunk.new(metadata)
+    c.append(data)
+    c.commit
+    c
+  end
+
+  def create_chunk_es(metadata, es)
+    c = FluentPluginBufferTest::DummyMemoryChunk.new(metadata)
+    c.concat(es.to_msgpack_stream, es.size)
+    c.commit
+    c
+  end
+
+  setup do
+    Fluent::Test.setup
+  end
+
+  sub_test_case 'using base buffer class' do
+    setup do
+      buffer_conf = config_element('buffer', '', {}, [])
+      owner = FluentPluginBufferTest::DummyOutputPlugin.new
+      owner.configure(config_element('ROOT', '', {}, [ buffer_conf ]))
+      p = Fluent::Plugin::Buffer.new
+      p.owner = owner
+      p.configure(buffer_conf)
+      @p = p
+    end
+
+    test 'default persistency is false' do
+      assert !@p.persistent?
+    end
+
+    test 'chunk bytes limit is 8MB, and total bytes limit is 512MB' do
+      assert_equal 8*1024*1024, @p.chunk_limit_size
+      assert_equal 512*1024*1024, @p.total_limit_size
+    end
+
+    test 'chunk records limit is ignored in default' do
+      assert_nil @p.chunk_limit_records
+    end
+
+    test '#storable? checks total size of staged and enqueued(includes dequeued chunks) against total_limit_size' do
+      assert_equal 512*1024*1024, @p.total_limit_size
+      assert_equal 0, @p.stage_size
+      assert_equal 0, @p.queue_size
+      assert @p.storable?
+
+      @p.stage_size = 256 * 1024 * 1024
+      @p.queue_size = 256 * 1024 * 1024 - 1
+      assert @p.storable?
+
+      @p.queue_size = 256 * 1024 * 1024
+      assert !@p.storable?
+    end
+
+    test '#resume must be implemented by subclass' do
+      assert_raise NotImplementedError do
+        @p.resume
+      end
+    end
+
+    test '#generate_chunk must be implemented by subclass' do
+      assert_raise NotImplementedError do
+        @p.generate_chunk(Object.new)
+      end
+    end
+  end
+
+  sub_test_case 'with default configuration and dummy implementation' do
+    setup do
+      @p = create_buffer({'queued_chunks_limit_size' => 100})
+      @dm0 = create_metadata(Time.parse('2016-04-11 16:00:00 +0000').to_i, nil, nil)
+      @dm1 = create_metadata(Time.parse('2016-04-11 16:10:00 +0000').to_i, nil, nil)
+      @dm2 = create_metadata(Time.parse('2016-04-11 16:20:00 +0000').to_i, nil, nil)
+      @dm3 = create_metadata(Time.parse('2016-04-11 16:30:00 +0000').to_i, nil, nil)
+      @p.start
+    end
+
+    test '#start resumes buffer states and update queued numbers per metadata' do
+      plugin = create_buffer({})
+
+      assert_equal({}, plugin.stage)
+      assert_equal([], plugin.queue)
+      assert_equal({}, plugin.dequeued)
+      assert_equal({}, plugin.queued_num)
+      assert_equal([], plugin.metadata_list)
+
+      assert_equal 0, plugin.stage_size
+      assert_equal 0, plugin.queue_size
+      assert_equal [], plugin.timekeys
+
+      # @p is started plugin
+
+      assert_equal [@dm2,@dm3], @p.stage.keys
+      assert_equal "b" * 100, @p.stage[@dm2].read
+      assert_equal "c" * 100, @p.stage[@dm3].read
+
+      assert_equal 200, @p.stage_size
+
+      assert_equal 3, @p.queue.size
+      assert_equal "0" * 100, @p.queue[0].read
+      assert_equal "a" * 100, @p.queue[1].read
+      assert_equal "a" * 3, @p.queue[2].read
+
+      assert_equal 203, @p.queue_size
+
+      # staged, queued
+      assert_equal [@dm2,@dm3,@dm0,@dm1], @p.metadata_list
+      assert_equal 1, @p.queued_num[@dm0]
+      assert_equal 2, @p.queued_num[@dm1]
+    end
+
+    test '#close closes all chunks in dequeued, enqueued and staged' do
+      dmx = create_metadata(Time.parse('2016-04-11 15:50:00 +0000').to_i, nil, nil)
+      cx = create_chunk(dmx, ["x" * 1024])
+      @p.dequeued[cx.unique_id] = cx
+
+      staged_chunks = @p.stage.values.dup
+      queued_chunks = @p.queue.dup
+
+      @p.close
+
+      assert cx.closed
+      assert{ staged_chunks.all?{|c| c.closed } }
+      assert{ queued_chunks.all?{|c| c.closed } }
+    end
+
+    test '#terminate initializes all internal states' do
+      dmx = create_metadata(Time.parse('2016-04-11 15:50:00 +0000').to_i, nil, nil)
+      cx = create_chunk(dmx, ["x" * 1024])
+      @p.dequeued[cx.unique_id] = cx
+
+      @p.close
+
+      @p.terminate
+
+      assert_nil @p.stage
+      assert_nil @p.queue
+      assert_nil @p.dequeued
+      assert_nil @p.queued_num
+      assert_nil @p.instance_eval{ @metadata_list } # #metadata_list does #dup for @metadata_list
+      assert_equal 0, @p.stage_size
+      assert_equal 0, @p.queue_size
+      assert_equal [], @p.timekeys
+    end
+
+    test '#metadata_list returns list of metadata on stage or in queue' do
+      assert_equal [@dm2,@dm3,@dm0,@dm1], @p.metadata_list
+    end
+
+    test '#new_metadata creates metadata instance without inserting metadata_list' do
+      assert_equal [@dm2,@dm3,@dm0,@dm1], @p.metadata_list
+      _m = @p.new_metadata(timekey: Time.parse('2016-04-11 16:40:00 +0000').to_i)
+      assert_equal [@dm2,@dm3,@dm0,@dm1], @p.metadata_list
+    end
+
+    test '#add_metadata adds unknown metadata into list, or return known metadata if already exists' do
+      assert_equal [@dm2,@dm3,@dm0,@dm1], @p.metadata_list
+
+      m = @p.new_metadata(timekey: Time.parse('2016-04-11 16:40:00 +0000').to_i)
+      _mx = @p.add_metadata(m)
+      assert_equal [@dm2,@dm3,@dm0,@dm1,m], @p.metadata_list
+      assert_equal m.object_id, m.object_id
+
+      my = @p.add_metadata(@dm1)
+      assert_equal [@dm2,@dm3,@dm0,@dm1,m], @p.metadata_list
+      assert_equal @dm1, my
+      assert{ @dm1.object_id != my.object_id } # 'my' is an object created in #resume
+    end
+
+    test '#metadata is utility method to create-add-and-return metadata' do
+      assert_equal [@dm2,@dm3,@dm0,@dm1], @p.metadata_list
+
+      m1 = @p.metadata(timekey: Time.parse('2016-04-11 16:40:00 +0000').to_i)
+      assert_equal [@dm2,@dm3,@dm0,@dm1,m1], @p.metadata_list
+      m2 = @p.metadata(timekey: @dm3.timekey)
+      assert_equal [@dm2,@dm3,@dm0,@dm1,m1], @p.metadata_list
+      assert_equal @dm3, m2
+    end
+
+    test '#queued_records returns total number of size in all chunks in queue' do
+      assert_equal 3, @p.queue.size
+
+      r0 = @p.queue[0].size
+      assert_equal 1, r0
+      r1 = @p.queue[1].size
+      assert_equal 1, r1
+      r2 = @p.queue[2].size
+      assert_equal 1, r2
+
+      assert_equal (r0+r1+r2), @p.queued_records
+    end
+
+    test '#queued? returns queue has any chunks or not without arguments' do
+      assert @p.queued?
+
+      @p.queue.reject!{|_c| true }
+      assert !@p.queued?
+    end
+
+    test '#queued? returns queue has chunks for specified metadata with an argument' do
+      assert @p.queued?(@dm0)
+      assert @p.queued?(@dm1)
+      assert !@p.queued?(@dm2)
+    end
+
+    test '#enqueue_chunk enqueues a chunk on stage with specified metadata' do
+      assert_equal 2, @p.stage.size
+      assert_equal [@dm2,@dm3], @p.stage.keys
+      assert_equal 3, @p.queue.size
+      assert_nil @p.queued_num[@dm2]
+
+      assert_equal 200, @p.stage_size
+      assert_equal 203, @p.queue_size
+
+      @p.enqueue_chunk(@dm2)
+
+      assert_equal [@dm3], @p.stage.keys
+      assert_equal @dm2, @p.queue.last.metadata
+      assert_equal 1, @p.queued_num[@dm2]
+      assert_equal 100, @p.stage_size
+      assert_equal 303, @p.queue_size
+    end
+
+    test '#enqueue_chunk ignores empty chunks' do
+      assert_equal 3, @p.queue.size
+
+      m = @p.metadata(timekey: Time.parse('2016-04-11 16:40:00 +0000').to_i)
+      c = create_chunk(m, [''])
+      @p.stage[m] = c
+      assert @p.stage[m].empty?
+      assert !c.closed
+
+      @p.enqueue_chunk(m)
+
+      assert_nil @p.stage[m]
+      assert_equal 3, @p.queue.size
+      assert_nil @p.queued_num[m]
+      assert c.closed
+    end
+
+    test '#enqueue_chunk calls #enqueued! if chunk responds to it' do
+      assert_equal 3, @p.queue.size
+      m = @p.metadata(timekey: Time.parse('2016-04-11 16:40:00 +0000').to_i)
+      c = create_chunk(m, ['c' * 256])
+      callback_called = false
+      (class << c; self; end).module_eval do
+        define_method(:enqueued!){ callback_called = true }
+      end
+
+      @p.stage[m] = c
+      @p.enqueue_chunk(m)
+
+      assert_equal c, @p.queue.last
+      assert callback_called
+    end
+
+    test '#enqueue_all enqueues chunks on stage which given block returns true with' do
+      m1 = @p.metadata(timekey: Time.parse('2016-04-11 16:40:00 +0000').to_i)
+      c1 = create_chunk(m1, ['c' * 256])
+      @p.stage[m1] = c1
+      m2 = @p.metadata(timekey: Time.parse('2016-04-11 16:50:00 +0000').to_i)
+      c2 = create_chunk(m2, ['c' * 256])
+      @p.stage[m2] = c2
+
+      assert_equal [@dm2,@dm3,m1,m2], @p.stage.keys
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+
+      @p.enqueue_all{ |m, c| m.timekey < Time.parse('2016-04-11 16:41:00 +0000').to_i }
+
+      assert_equal [m2], @p.stage.keys
+      assert_equal [@dm0,@dm1,@dm1,@dm2,@dm3,m1], @p.queue.map(&:metadata)
+    end
+
+    test '#enqueue_all enqueues all chunks on stage without block' do
+      m1 = @p.metadata(timekey: Time.parse('2016-04-11 16:40:00 +0000').to_i)
+      c1 = create_chunk(m1, ['c' * 256])
+      @p.stage[m1] = c1
+      m2 = @p.metadata(timekey: Time.parse('2016-04-11 16:50:00 +0000').to_i)
+      c2 = create_chunk(m2, ['c' * 256])
+      @p.stage[m2] = c2
+
+      assert_equal [@dm2,@dm3,m1,m2], @p.stage.keys
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+
+      @p.enqueue_all
+
+      assert_equal [], @p.stage.keys
+      assert_equal [@dm0,@dm1,@dm1,@dm2,@dm3,m1,m2], @p.queue.map(&:metadata)
+    end
+
+    test '#dequeue_chunk dequeues a chunk from queue if a chunk exists' do
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal({}, @p.dequeued)
+
+      m1 = @p.dequeue_chunk
+      assert_equal @dm0, m1.metadata
+      assert_equal @dm0, @p.dequeued[m1.unique_id].metadata
+
+      m2 = @p.dequeue_chunk
+      assert_equal @dm1, m2.metadata
+      assert_equal @dm1, @p.dequeued[m2.unique_id].metadata
+
+      m3 = @p.dequeue_chunk
+      assert_equal @dm1, m3.metadata
+      assert_equal @dm1, @p.dequeued[m3.unique_id].metadata
+
+      m4 = @p.dequeue_chunk
+      assert_nil m4
+    end
+
+    test '#takeback_chunk resumes a chunk from dequeued to queued at the head of queue, and returns true' do
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal({}, @p.dequeued)
+
+      m1 = @p.dequeue_chunk
+      assert_equal @dm0, m1.metadata
+      assert_equal @dm0, @p.dequeued[m1.unique_id].metadata
+      assert_equal [@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal({m1.unique_id => m1}, @p.dequeued)
+
+      assert @p.takeback_chunk(m1.unique_id)
+
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal({}, @p.dequeued)
+    end
+
+    test '#purge_chunk removes a chunk specified by argument id from dequeued chunks' do
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal({}, @p.dequeued)
+      assert_equal [@dm2,@dm3,@dm0,@dm1], @p.metadata_list
+
+      m0 = @p.dequeue_chunk
+      m1 = @p.dequeue_chunk
+
+      assert @p.takeback_chunk(m0.unique_id)
+
+      assert_equal [@dm0,@dm1], @p.queue.map(&:metadata)
+      assert_equal({m1.unique_id => m1}, @p.dequeued)
+
+      assert !m1.purged
+
+      @p.purge_chunk(m1.unique_id)
+      assert m1.purged
+
+      assert_equal [@dm0,@dm1], @p.queue.map(&:metadata)
+      assert_equal({}, @p.dequeued)
+      assert_equal [@dm2,@dm3,@dm0,@dm1], @p.metadata_list
+    end
+
+    test '#purge_chunk removes an argument metadata from metadata_list if no chunks exist on stage or in queue' do
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal({}, @p.dequeued)
+      assert_equal [@dm2,@dm3,@dm0,@dm1], @p.metadata_list
+
+      m0 = @p.dequeue_chunk
+
+      assert_equal [@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal({m0.unique_id => m0}, @p.dequeued)
+
+      assert !m0.purged
+
+      @p.purge_chunk(m0.unique_id)
+      assert m0.purged
+
+      assert_equal [@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal({}, @p.dequeued)
+      assert_equal [@dm2,@dm3,@dm1], @p.metadata_list
+    end
+
+    test '#takeback_chunk returns false if specified chunk_id is already purged' do
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal({}, @p.dequeued)
+      assert_equal [@dm2,@dm3,@dm0,@dm1], @p.metadata_list
+
+      m0 = @p.dequeue_chunk
+
+      assert_equal [@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal({m0.unique_id => m0}, @p.dequeued)
+
+      assert !m0.purged
+
+      @p.purge_chunk(m0.unique_id)
+      assert m0.purged
+
+      assert_equal [@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal({}, @p.dequeued)
+      assert_equal [@dm2,@dm3,@dm1], @p.metadata_list
+
+      assert !@p.takeback_chunk(m0.unique_id)
+
+      assert_equal [@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal({}, @p.dequeued)
+      assert_equal [@dm2,@dm3,@dm1], @p.metadata_list
+    end
+
+    test '#clear_queue! removes all chunks in queue, but leaves staged chunks' do
+      qchunks = @p.queue.dup
+
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal 2, @p.stage.size
+      assert_equal({}, @p.dequeued)
+
+      @p.clear_queue!
+
+      assert_equal [], @p.queue
+      assert_equal 0, @p.queue_size
+      assert_equal 2, @p.stage.size
+      assert_equal({}, @p.dequeued)
+
+      assert{ qchunks.all?{ |c| c.purged } }
+    end
+
+    test '#write returns immediately if argument data is empty array' do
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+
+      m = @p.metadata(timekey: Time.parse('2016-04-11 16:40:00 +0000').to_i)
+
+      @p.write({m => []})
+
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+    end
+
+    test '#write returns immediately if argument data is empty event stream' do
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+
+      m = @p.metadata(timekey: Time.parse('2016-04-11 16:40:00 +0000').to_i)
+
+      @p.write({m => Fluent::ArrayEventStream.new([])})
+
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+    end
+
+    test '#write raises BufferOverflowError if buffer is not storable' do
+      @p.stage_size = 256 * 1024 * 1024
+      @p.queue_size = 256 * 1024 * 1024
+
+      m = @p.metadata(timekey: Time.parse('2016-04-11 16:40:00 +0000').to_i)
+
+      assert_raise Fluent::Plugin::Buffer::BufferOverflowError do
+        @p.write({m => ["x" * 256]})
+      end
+    end
+
+    test '#write stores data into an existing chunk with metadata specified' do
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+
+      dm3data = @p.stage[@dm3].read.dup
+      prev_stage_size = @p.stage_size
+
+      assert_equal 1, @p.stage[@dm3].append_count
+
+      @p.write({@dm3 => ["x" * 256, "y" * 256, "z" * 256]})
+
+      assert_equal 2, @p.stage[@dm3].append_count
+      assert_equal (dm3data + ("x" * 256) + ("y" * 256) + ("z" * 256)), @p.stage[@dm3].read
+      assert_equal (prev_stage_size + 768), @p.stage_size
+
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+    end
+
+    test '#write creates new chunk and store data into it if there are no chunks for specified metadata' do
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+
+      timekey = Time.parse('2016-04-11 16:40:00 +0000').to_i
+      assert !@p.timekeys.include?(timekey)
+
+      prev_stage_size = @p.stage_size
+      
+      m = @p.metadata(timekey: timekey)
+
+      @p.write({m => ["x" * 256, "y" * 256, "z" * 256]})
+
+      assert_equal 1, @p.stage[m].append_count
+      assert_equal ("x" * 256 + "y" * 256 + "z" * 256), @p.stage[m].read
+      assert_equal (prev_stage_size + 768), @p.stage_size
+
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3,m], @p.stage.keys
+
+      assert @p.timekeys.include?(timekey)
+    end
+
+    test '#write tries to enqueue and store data into a new chunk if existing chunk is full' do
+      assert_equal 8 * 1024 * 1024, @p.chunk_limit_size
+      assert_equal 0.95, @p.chunk_full_threshold
+
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+
+      m = @p.metadata(timekey: Time.parse('2016-04-11 16:40:00 +0000').to_i)
+
+      row = "x" * 1024 * 1024
+      small_row = "x" * 1024 * 512
+      @p.write({m => [row] * 7 + [small_row]})
+
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3,m], @p.stage.keys
+      assert_equal 1, @p.stage[m].append_count
+
+      @p.write({m => [row]})
+
+      assert_equal [@dm0,@dm1,@dm1,m], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3,m], @p.stage.keys
+      assert_equal 1, @p.stage[m].append_count
+      assert_equal 1024*1024, @p.stage[m].bytesize
+      assert_equal 3, @p.queue.last.append_count # 1 -> write (2) -> write_step_by_step (3)
+      assert @p.queue.last.rollbacked
+    end
+
+    test '#write rollbacks if commit raises errors' do
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+
+      m = @p.metadata(timekey: Time.parse('2016-04-11 16:40:00 +0000').to_i)
+
+      row = "x" * 1024
+      @p.write({m => [row] * 8})
+
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3,m], @p.stage.keys
+
+      target_chunk = @p.stage[m]
+
+      assert_equal 1, target_chunk.append_count
+      assert !target_chunk.rollbacked
+
+      (class << target_chunk; self; end).module_eval do
+        define_method(:commit){ raise "yay" }
+      end
+
+      assert_raise RuntimeError.new("yay") do
+        @p.write({m => [row]})
+      end
+
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3,m], @p.stage.keys
+
+      assert_equal 2, target_chunk.append_count
+      assert target_chunk.rollbacked
+      assert_equal row * 8, target_chunk.read
+    end
+
+    test '#write w/ format raises BufferOverflowError if buffer is not storable' do
+      @p.stage_size = 256 * 1024 * 1024
+      @p.queue_size = 256 * 1024 * 1024
+
+      m = @p.metadata(timekey: Time.parse('2016-04-11 16:40:00 +0000').to_i)
+
+      es = Fluent::ArrayEventStream.new([ [event_time('2016-04-11 16:40:01 +0000'), {"message" => "xxxxxxxxxxxxxx"} ] ])
+
+      assert_raise Fluent::Plugin::Buffer::BufferOverflowError do
+        @p.write({m => es}, format: ->(e){e.to_msgpack_stream})
+      end
+    end
+
+    test '#write w/ format stores data into an existing chunk with metadata specified' do
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+
+      dm3data = @p.stage[@dm3].read.dup
+      prev_stage_size = @p.stage_size
+
+      assert_equal 1, @p.stage[@dm3].append_count
+
+      es = Fluent::ArrayEventStream.new(
+        [
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 128}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "y" * 128}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "z" * 128}],
+        ]
+      )
+
+      @p.write({@dm3 => es}, format: ->(e){e.to_msgpack_stream})
+
+      assert_equal 2, @p.stage[@dm3].append_count
+      assert_equal (dm3data + es.to_msgpack_stream), @p.stage[@dm3].read
+      assert_equal (prev_stage_size + es.to_msgpack_stream.bytesize), @p.stage_size
+
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+    end
+
+    test '#write w/ format creates new chunk and store data into it if there are not chunks for specified metadata' do
+      assert_equal 8 * 1024 * 1024, @p.chunk_limit_size
+
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+      
+      timekey = Time.parse('2016-04-11 16:40:00 +0000').to_i
+      assert !@p.timekeys.include?(timekey)
+
+      m = @p.metadata(timekey: timekey)
+
+      es = Fluent::ArrayEventStream.new(
+        [
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:03 +0000'), {"message" => "z" * 1024 * 512}],
+        ]
+      )
+      @p.write({m => es}, format: ->(e){e.to_msgpack_stream})
+
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3,m], @p.stage.keys
+      assert_equal 1, @p.stage[m].append_count
+      
+      assert @p.timekeys.include?(timekey)
+    end
+
+    test '#write w/ format tries to enqueue and store data into a new chunk if existing chunk does not have enough space' do
+      assert_equal 8 * 1024 * 1024, @p.chunk_limit_size
+
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+
+      m = @p.metadata(timekey: Time.parse('2016-04-11 16:40:00 +0000').to_i)
+
+      es = Fluent::ArrayEventStream.new(
+        [
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:03 +0000'), {"message" => "z" * 1024 * 512}],
+        ]
+      )
+      @p.write({m => es}, format: ->(e){e.to_msgpack_stream})
+
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3,m], @p.stage.keys
+      assert_equal 1, @p.stage[m].append_count
+
+      es2 = Fluent::OneEventStream.new(event_time('2016-04-11 16:40:03 +0000'), {"message" => "z" * 1024 * 1024})
+      @p.write({m => es2}, format: ->(e){e.to_msgpack_stream})
+
+      assert_equal [@dm0,@dm1,@dm1,m], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3,m], @p.stage.keys
+      assert_equal 1, @p.stage[m].append_count
+      assert_equal es2.to_msgpack_stream.bytesize, @p.stage[m].bytesize
+      assert_equal 2, @p.queue.last.append_count # 1 -> write (2) -> rollback&enqueue
+      assert @p.queue.last.rollbacked
+    end
+
+    test '#write w/ format enqueues chunk if it is already full after adding data' do
+      assert_equal 8 * 1024 * 1024, @p.chunk_limit_size
+
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+
+      m = @p.metadata(timekey: Time.parse('2016-04-11 16:40:00 +0000').to_i)
+      es = Fluent::ArrayEventStream.new(
+        [
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * (1024 * 1024 - 25)}], # 1024 * 1024 bytes as msgpack stream
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * (1024 * 1024 - 25)}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * (1024 * 1024 - 25)}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * (1024 * 1024 - 25)}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * (1024 * 1024 - 25)}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * (1024 * 1024 - 25)}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * (1024 * 1024 - 25)}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * (1024 * 1024 - 25)}],
+        ]
+      )
+      @p.write({m => es}, format: ->(e){e.to_msgpack_stream})
+
+      assert_equal [@dm0,@dm1,@dm1,m], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+      assert_equal 1, @p.queue.last.append_count
+    end
+
+    test '#write w/ format rollbacks if commit raises errors' do
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+
+      m = @p.metadata(timekey: Time.parse('2016-04-11 16:40:00 +0000').to_i)
+
+      es = Fluent::ArrayEventStream.new(
+        [
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:01 +0000'), {"message" => "x" * 1024 * 1024}],
+          [event_time('2016-04-11 16:40:03 +0000'), {"message" => "z" * 1024 * 512}],
+        ]
+      )
+      @p.write({m => es}, format: ->(e){e.to_msgpack_stream})
+
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3,m], @p.stage.keys
+
+      target_chunk = @p.stage[m]
+
+      assert_equal 1, target_chunk.append_count
+      assert !target_chunk.rollbacked
+
+      (class << target_chunk; self; end).module_eval do
+        define_method(:commit){ raise "yay" }
+      end
+
+      es2 = Fluent::ArrayEventStream.new(
+        [
+          [event_time('2016-04-11 16:40:04 +0000'), {"message" => "z" * 1024 * 128}],
+        ]
+      )
+      assert_raise RuntimeError.new("yay") do
+        @p.write({m => es2}, format: ->(e){e.to_msgpack_stream})
+      end
+
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3,m], @p.stage.keys
+
+      assert_equal 2, target_chunk.append_count
+      assert target_chunk.rollbacked
+      assert_equal es.to_msgpack_stream, target_chunk.read
+    end
+
+    test '#write writes many metadata and data pairs at once' do
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+
+      row = "x" * 1024
+      @p.write({ @dm0 => [row, row, row], @dm1 => [row, row] })
+
+      assert_equal [@dm2,@dm3,@dm0,@dm1], @p.stage.keys
+    end
+
+    test '#write does not commit on any chunks if any append operation on chunk fails' do
+      assert_equal [@dm0,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+
+      row = "x" * 1024
+      @p.write({ @dm0 => [row, row, row], @dm1 => [row, row] })
+
+      assert_equal [@dm2,@dm3,@dm0,@dm1], @p.stage.keys
+
+      dm2_size = @p.stage[@dm2].size
+      assert !@p.stage[@dm2].rollbacked
+      dm3_size = @p.stage[@dm3].size
+      assert !@p.stage[@dm3].rollbacked
+
+      assert{ @p.stage[@dm0].size == 3 }
+      assert !@p.stage[@dm0].rollbacked
+      assert{ @p.stage[@dm1].size == 2 }
+      assert !@p.stage[@dm1].rollbacked
+
+      meta_list = [@dm0, @dm1, @dm2, @dm3].sort
+      @p.stage[meta_list.last].failing = true
+
+      assert_raise(FluentPluginBufferTest::DummyMemoryChunkError) do
+        @p.write({ @dm2 => [row], @dm3 => [row], @dm0 => [row, row, row], @dm1 => [row, row] })
+      end
+
+      assert{ @p.stage[@dm2].size == dm2_size }
+      assert @p.stage[@dm2].rollbacked
+      assert{ @p.stage[@dm3].size == dm3_size }
+      assert @p.stage[@dm3].rollbacked
+
+      assert{ @p.stage[@dm0].size == 3 }
+      assert @p.stage[@dm0].rollbacked
+      assert{ @p.stage[@dm1].size == 2 }
+      assert @p.stage[@dm1].rollbacked
+    end
+
+    test '#compress returns :text' do
+      assert_equal :text, @p.compress
+    end
+  end
+
+  sub_test_case 'standard format with configuration for test with lower chunk limit size' do
+    setup do
+      @p = create_buffer({"chunk_limit_size" => 1_280_000})
+      @format = ->(e){e.to_msgpack_stream}
+      @dm0 = dm0 = create_metadata(Time.parse('2016-04-11 16:00:00 +0000').to_i, nil, nil)
+      # 1 record is 128bytes in msgpack stream
+      @es0 = es0 = Fluent::ArrayEventStream.new([ [event_time('2016-04-11 16:00:01 +0000'), {"message" => "x" * (128 - 22)}] ] * 5000)
+      (class << @p; self; end).module_eval do
+        define_method(:resume) {
+          staged = {
+            dm0 => create_chunk_es(dm0, es0).staged!,
+          }
+          queued = []
+          return staged, queued
+        }
+      end
+      @p.start
+    end
+
+    test '#write appends event stream into staged chunk' do
+      assert_equal [@dm0], @p.stage.keys
+      assert_equal [], @p.queue.map(&:metadata)
+
+      assert_equal 1_280_000, @p.chunk_limit_size
+
+      es = Fluent::ArrayEventStream.new([ [event_time('2016-04-11 16:00:02 +0000'), {"message" => "x" * (128 - 22)}] ] * 1000)
+      @p.write({@dm0 => es}, format: @format)
+
+      assert_equal [@dm0], @p.stage.keys
+      assert_equal [], @p.queue.map(&:metadata)
+
+      assert_equal (@es0.to_msgpack_stream + es.to_msgpack_stream), @p.stage[@dm0].read
+    end
+
+    test '#write writes event stream into a new chunk with enqueueing existing chunk if event stream is larger than available space of existing chunk' do
+      assert_equal [@dm0], @p.stage.keys
+      assert_equal [], @p.queue.map(&:metadata)
+
+      assert_equal 1_280_000, @p.chunk_limit_size
+
+      es = Fluent::ArrayEventStream.new([ [event_time('2016-04-11 16:00:02 +0000'), {"message" => "x" * (128 - 22)}] ] * 8000)
+      @p.write({@dm0 => es}, format: @format)
+
+      assert_equal [@dm0], @p.stage.keys
+      assert_equal [@dm0], @p.queue.map(&:metadata)
+
+      assert_equal (es.to_msgpack_stream), @p.stage[@dm0].read
+    end
+
+    test '#write writes event stream into many chunks excluding staged chunk if event stream is larger than chunk limit size' do
+      assert_equal [@dm0], @p.stage.keys
+      assert_equal [], @p.queue.map(&:metadata)
+
+      assert_equal 1_280_000, @p.chunk_limit_size
+
+      es = Fluent::ArrayEventStream.new([ [event_time('2016-04-11 16:00:02 +0000'), {"message" => "x" * (128 - 22)}] ] * 45000)
+      @p.write({@dm0 => es}, format: @format)
+
+      assert_equal [@dm0], @p.stage.keys
+      assert_equal 5400, @p.stage[@dm0].size
+      assert_equal [@dm0,@dm0,@dm0,@dm0,@dm0], @p.queue.map(&:metadata)
+      assert_equal [5000, 9900, 9900, 9900, 9900], @p.queue.map(&:size) # splits: 45000 / 100 => 450 * ...
+      # 9900 * 4 + 5400 == 45000
+    end
+
+    test '#write raises BufferChunkOverflowError if a record is biggar than chunk limit size' do
+      assert_equal [@dm0], @p.stage.keys
+      assert_equal [], @p.queue.map(&:metadata)
+
+      assert_equal 1_280_000, @p.chunk_limit_size
+
+      es = Fluent::ArrayEventStream.new([ [event_time('2016-04-11 16:00:02 +0000'), {"message" => "x" * 1_280_000}] ])
+      assert_raise Fluent::Plugin::Buffer::BufferChunkOverflowError do
+        @p.write({@dm0 => es}, format: @format)
+      end
+    end
+  end
+
+  sub_test_case 'custom format with configuration for test with lower chunk limit size' do
+    setup do
+      @p = create_buffer({"chunk_limit_size" => 1_280_000})
+      @dm0 = dm0 = create_metadata(Time.parse('2016-04-11 16:00:00 +0000').to_i, nil, nil)
+      @row = "x" * 128
+      @data0 = data0 = [@row] * 5000
+      (class << @p; self; end).module_eval do
+        define_method(:resume) {
+          staged = {
+            dm0 => create_chunk(dm0, data0).staged!,
+          }
+          queued = []
+          return staged, queued
+        }
+      end
+      @p.start
+    end
+
+    test '#write appends event stream into staged chunk' do
+      assert_equal [@dm0], @p.stage.keys
+      assert_equal [], @p.queue.map(&:metadata)
+
+      assert_equal 1_280_000, @p.chunk_limit_size
+
+      data = [@row] * 1000
+      @p.write({@dm0 => data})
+
+      assert_equal [@dm0], @p.stage.keys
+      assert_equal [], @p.queue.map(&:metadata)
+
+      assert_equal (@row * 6000), @p.stage[@dm0].read
+    end
+
+    test '#write writes event stream into a new chunk with enqueueing existing chunk if event stream is larger than available space of existing chunk' do
+      assert_equal [@dm0], @p.stage.keys
+      assert_equal [], @p.queue.map(&:metadata)
+
+      staged_chunk_object_id = @p.stage[@dm0].object_id
+
+      assert_equal 1_280_000, @p.chunk_limit_size
+
+      data = [@row] * 8000
+      @p.write({@dm0 => data})
+
+      assert_equal [@dm0], @p.queue.map(&:metadata)
+      assert_equal [staged_chunk_object_id], @p.queue.map(&:object_id)
+      assert_equal [@dm0], @p.stage.keys
+
+      assert_equal [9800], @p.queue.map(&:size)
+      assert_equal 3200, @p.stage[@dm0].size
+      # 9800 + 3200 == 5000 + 8000
+    end
+
+    test '#write writes event stream into many chunks including staging chunk if event stream is larger than chunk limit size' do
+      assert_equal [@dm0], @p.stage.keys
+      assert_equal [], @p.queue.map(&:metadata)
+
+      staged_chunk_object_id = @p.stage[@dm0].object_id
+
+      assert_equal 1_280_000, @p.chunk_limit_size
+
+      assert_equal 5000, @p.stage[@dm0].size
+
+      data = [@row] * 45000
+      @p.write({@dm0 => data})
+
+      assert_equal staged_chunk_object_id, @p.queue.first.object_id
+
+      assert_equal [@dm0], @p.stage.keys
+      assert_equal 900, @p.stage[@dm0].size
+      assert_equal [@dm0,@dm0,@dm0,@dm0,@dm0], @p.queue.map(&:metadata)
+      assert_equal [9500, 9900, 9900, 9900, 9900], @p.queue.map(&:size) # splits: 45000 / 100 => 450 * ...
+      ##### 900 + 9500 + 9900 * 4 == 5000 + 45000
+    end
+
+    test '#write raises BufferChunkOverflowError if a record is bigger than chunk limit size' do
+      assert_equal [@dm0], @p.stage.keys
+      assert_equal [], @p.queue.map(&:metadata)
+
+      assert_equal 1_280_000, @p.chunk_limit_size
+
+      es = ["x" * 1_280_000 + "x" * 300]
+      assert_raise Fluent::Plugin::Buffer::BufferChunkOverflowError do
+        @p.write({@dm0 => es})
+      end
+    end
+  end
+
+  sub_test_case 'with configuration for test with lower limits' do
+    setup do
+      @p = create_buffer({"chunk_limit_size" => 1024, "total_limit_size" => 10240})
+      @dm0 = dm0 = create_metadata(Time.parse('2016-04-11 16:00:00 +0000').to_i, nil, nil)
+      @dm1 = dm1 = create_metadata(Time.parse('2016-04-11 16:10:00 +0000').to_i, nil, nil)
+      @dm2 = dm2 = create_metadata(Time.parse('2016-04-11 16:20:00 +0000').to_i, nil, nil)
+      @dm3 = dm3 = create_metadata(Time.parse('2016-04-11 16:30:00 +0000').to_i, nil, nil)
+      (class << @p; self; end).module_eval do
+        define_method(:resume) {
+          staged = {
+            dm2 => create_chunk(dm2, ["b" * 128] * 7).staged!,
+            dm3 => create_chunk(dm3, ["c" * 128] * 5).staged!,
+          }
+          queued = [
+            create_chunk(dm0, ["0" * 128] * 8).enqueued!,
+            create_chunk(dm0, ["0" * 128] * 8).enqueued!,
+            create_chunk(dm0, ["0" * 128] * 8).enqueued!,
+            create_chunk(dm0, ["0" * 128] * 8).enqueued!,
+            create_chunk(dm0, ["0" * 128] * 8).enqueued!,
+            create_chunk(dm1, ["a" * 128] * 8).enqueued!,
+            create_chunk(dm1, ["a" * 128] * 8).enqueued!,
+            create_chunk(dm1, ["a" * 128] * 8).enqueued!, # 8th queued chunk
+            create_chunk(dm1, ["a" * 128] * 3).enqueued!,
+          ]
+          return staged, queued
+        }
+      end
+      @p.start
+    end
+
+    test '#storable? returns false when too many data exist' do
+      assert_equal [@dm0,@dm0,@dm0,@dm0,@dm0,@dm1,@dm1,@dm1,@dm1], @p.queue.map(&:metadata)
+      assert_equal [@dm2,@dm3], @p.stage.keys
+
+      assert_equal 128*8*8+128*3, @p.queue_size
+      assert_equal 128*7+128*5, @p.stage_size
+
+      assert @p.storable?
+
+      dm3 = @p.metadata(timekey: @dm3.timekey)
+      @p.write({dm3 => ["c" * 128]})
+
+      assert_equal 10240, (@p.stage_size + @p.queue_size)
+      assert !@p.storable?
+    end
+
+    test '#chunk_size_over? returns true if chunk size is bigger than limit' do
+      m = create_metadata(Time.parse('2016-04-11 16:40:00 +0000').to_i)
+
+      c1 = create_chunk(m, ["a" * 128] * 8)
+      assert !@p.chunk_size_over?(c1)
+
+      c2 = create_chunk(m, ["a" * 128] * 9)
+      assert @p.chunk_size_over?(c2)
+
+      c3 = create_chunk(m, ["a" * 128] * 8 + ["a"])
+      assert @p.chunk_size_over?(c3)
+    end
+
+    test '#chunk_size_full? returns true if chunk size is enough big against limit' do
+      m = create_metadata(Time.parse('2016-04-11 16:40:00 +0000').to_i)
+
+      c1 = create_chunk(m, ["a" * 128] * 7)
+      assert !@p.chunk_size_full?(c1)
+
+      c2 = create_chunk(m, ["a" * 128] * 8)
+      assert @p.chunk_size_full?(c2)
+
+      assert_equal 0.95, @p.chunk_full_threshold
+      c3 = create_chunk(m, ["a" * 128] * 6 + ["a" * 64])
+      assert !@p.chunk_size_full?(c3)
+    end
+  end
+
+  sub_test_case 'with configuration includes chunk_limit_records' do
+    setup do
+      @p = create_buffer({"chunk_limit_size" => 1024, "total_limit_size" => 10240, "chunk_limit_records" => 6})
+      @dm0 = dm0 = create_metadata(Time.parse('2016-04-11 16:00:00 +0000').to_i, nil, nil)
+      @dm1 = dm1 = create_metadata(Time.parse('2016-04-11 16:10:00 +0000').to_i, nil, nil)
+      @dm2 = dm2 = create_metadata(Time.parse('2016-04-11 16:20:00 +0000').to_i, nil, nil)
+      @dm3 = dm3 = create_metadata(Time.parse('2016-04-11 16:30:00 +0000').to_i, nil, nil)
+      (class << @p; self; end).module_eval do
+        define_method(:resume) {
+          staged = {
+            dm2 => create_chunk(dm2, ["b" * 128] * 1).staged!,
+            dm3 => create_chunk(dm3, ["c" * 128] * 2).staged!,
+          }
+          queued = [
+            create_chunk(dm0, ["0" * 128] * 6).enqueued!,
+            create_chunk(dm1, ["a" * 128] * 6).enqueued!,
+            create_chunk(dm1, ["a" * 128] * 6).enqueued!,
+            create_chunk(dm1, ["a" * 128] * 3).enqueued!,
+          ]
+          return staged, queued
+        }
+      end
+      @p.start
+    end
+
+    test '#chunk_size_over? returns true if too many records exists in a chunk even if its bytes is less than limit' do
+      assert_equal 6, @p.chunk_limit_records
+
+      m = create_metadata(Time.parse('2016-04-11 16:40:00 +0000').to_i)
+
+      c1 = create_chunk(m, ["a" * 128] * 6)
+      assert_equal 6, c1.size
+      assert !@p.chunk_size_over?(c1)
+
+      c2 = create_chunk(m, ["a" * 128] * 7)
+      assert @p.chunk_size_over?(c2)
+
+      c3 = create_chunk(m, ["a" * 128] * 6 + ["a"])
+      assert @p.chunk_size_over?(c3)
+    end
+
+    test '#chunk_size_full? returns true if enough many records exists in a chunk even if its bytes is less than limit' do
+      assert_equal 6, @p.chunk_limit_records
+
+      m = create_metadata(Time.parse('2016-04-11 16:40:00 +0000').to_i)
+
+      c1 = create_chunk(m, ["a" * 128] * 5)
+      assert_equal 5, c1.size
+      assert !@p.chunk_size_full?(c1)
+
+      c2 = create_chunk(m, ["a" * 128] * 6)
+      assert @p.chunk_size_full?(c2)
+
+      c3 = create_chunk(m, ["a" * 128] * 5 + ["a"])
+      assert @p.chunk_size_full?(c3)
+    end
+  end
+
+  sub_test_case 'with configuration includes queue_limit_length' do
+    setup do
+      @p = create_buffer({"chunk_limit_size" => 1024, "total_limit_size" => 10240, "queue_limit_length" => 5})
+      @dm0 = dm0 = create_metadata(Time.parse('2016-04-11 16:00:00 +0000').to_i, nil, nil)
+      @dm1 = dm1 = create_metadata(Time.parse('2016-04-11 16:10:00 +0000').to_i, nil, nil)
+      @dm2 = dm2 = create_metadata(Time.parse('2016-04-11 16:20:00 +0000').to_i, nil, nil)
+      @dm3 = dm3 = create_metadata(Time.parse('2016-04-11 16:30:00 +0000').to_i, nil, nil)
+      (class << @p; self; end).module_eval do
+        define_method(:resume) {
+          staged = {
+            dm2 => create_chunk(dm2, ["b" * 128] * 1).staged!,
+            dm3 => create_chunk(dm3, ["c" * 128] * 2).staged!,
+          }
+          queued = [
+            create_chunk(dm0, ["0" * 128] * 6).enqueued!,
+            create_chunk(dm1, ["a" * 128] * 6).enqueued!,
+            create_chunk(dm1, ["a" * 128] * 6).enqueued!,
+            create_chunk(dm1, ["a" * 128] * 3).enqueued!,
+          ]
+          return staged, queued
+        }
+      end
+      @p.start
+    end
+
+    test '#configure will overwrite standard configuration if queue_limit_length' do
+      assert_equal 1024, @p.chunk_limit_size
+      assert_equal 5, @p.queue_limit_length
+      assert_equal (1024*5), @p.total_limit_size
+    end
+  end
+
+  sub_test_case 'when compress is gzip' do
+    setup do
+      @p = create_buffer({'compress' => 'gzip'})
+    end
+
+    test '#compress returns :gzip' do
+      assert_equal :gzip, @p.compress
+    end
+
+    test 'create decompressable chunk' do
+      chunk = @p.generate_chunk(create_metadata)
+      assert chunk.singleton_class.ancestors.include?(Fluent::Plugin::Buffer::Chunk::Decompressable)
+    end
+  end
+end