add config test

Author: mdoi

ChangeLog

@@ -1,3 +1,8 @@
+Release 0.1.1 - 2015/02/02
+
+  * add config test
+
 Release 0.1.0 - 2014/10/22
 
   * First commit
+

Rakefile

@@ -1,2 +1,11 @@
 require "bundler/gem_tasks"
+require 'rake/testtask'
+
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.test_files = FileList['test/test_*.rb']
+  test.verbose = true
+end
+
+task :default => [:build]
 

VERSION

@@ -1 +1 @@
-0.1.0
+0.1.1

fluent-plugin-dynamodb-add.gemspec

@@ -19,6 +19,7 @@ Gem::Specification.new do |spec|
 
   spec.add_development_dependency "bundler", "~> 1.7"
   spec.add_development_dependency "rake", "~> 10.0"
+  #spec.add_development_dependency "fake_dynamo", "~> 0.2.5"
   spec.add_dependency "fluentd", "~> 0.10.0"
   spec.add_dependency "aws-sdk", ">= 1.56.0"
 end

lib/fluent/plugin/out_dynamodb_add.rb

@@ -52,6 +52,7 @@ def start
     end
 
     def emit(tag, es, chain)
+      chain.next
       es.each do |time, record|
         hash_key = create_key(record)
         next unless hash_key || record[@count_key]

test/test_out_dynamodb_add.rb

@@ -0,0 +1,84 @@
+require 'fluent/test'
+require 'fluent/plugin/out_dynamodb_add'
+
+class DynamodbAddTest < Test::Unit::TestCase
+  def setup
+    Fluent::Test.setup
+  end
+
+  CONFIG = %[
+    count_key test_count_key
+    dynamo_count_key test_dynamo_count_key
+    table_name test_table_name
+    use_iam_role false
+    aws_key_id test_aws_key_id
+    aws_sec_key test_aws_sec_key
+    endpoint https://test_endpoint
+    hash_key test_hash_key1,test_hash_key2
+    hash_key_delimiter :
+    add_hash_key_prefix 3
+    range_key test_range_key
+  ]
+
+  def create_driver(conf = CONFIG)
+    Fluent::Test::OutputTestDriver.new(Fluent::DynamodbAdd) do
+      def write(chunk)
+        chunk.read
+      end
+    end.configure(conf)
+  end
+
+  def test_configure_not_use_iam_role
+    d = create_driver
+    assert_equal 'test_count_key', d.instance.count_key
+    assert_equal 'test_dynamo_count_key', d.instance.dynamo_count_key
+    assert_equal 'test_table_name', d.instance.table_name
+    assert_equal false, d.instance.use_iam_role
+    assert_equal 'test_aws_key_id', d.instance.aws_key_id
+    assert_equal 'test_aws_sec_key', d.instance.aws_sec_key
+    assert_equal 'https://test_endpoint', d.instance.endpoint
+    assert_equal ['test_hash_key1','test_hash_key2'], d.instance.hash_key
+    assert_equal ':', d.instance.hash_key_delimiter
+    assert_equal '3', d.instance.add_hash_key_prefix
+    assert_equal 'test_range_key', d.instance.range_key
+  end
+
+  def test_configure_use_iam_role
+    conf = CONFIG.clone
+    conf.gsub!(/use_iam_role\sfalse/, "use_iam_role true")
+    conf.gsub!(/aws_key_id\stest_aws_key_id/, "")
+    conf.gsub!(/aws_sec_key\stest_aws_sec_key/, "")
+
+    d = create_driver(conf)
+    assert_equal 'test_count_key', d.instance.count_key
+    assert_equal 'test_dynamo_count_key', d.instance.dynamo_count_key
+    assert_equal 'test_table_name', d.instance.table_name
+    assert_equal true, d.instance.use_iam_role
+    assert_equal nil, d.instance.aws_key_id
+    assert_equal nil, d.instance.aws_sec_key
+    assert_equal 'https://test_endpoint', d.instance.endpoint
+    assert_equal ['test_hash_key1','test_hash_key2'], d.instance.hash_key
+    assert_equal ':', d.instance.hash_key_delimiter
+    assert_equal '3', d.instance.add_hash_key_prefix
+    assert_equal 'test_range_key', d.instance.range_key
+  end
+
+  def test_configure_not_use_iam_role_and_not_set_range_key
+    conf = CONFIG.clone
+    conf.gsub!(/range_key\stest_range_key/, "")
+
+    d = create_driver(conf)
+    assert_equal 'test_count_key', d.instance.count_key
+    assert_equal 'test_dynamo_count_key', d.instance.dynamo_count_key
+    assert_equal 'test_table_name', d.instance.table_name
+    assert_equal false, d.instance.use_iam_role
+    assert_equal 'test_aws_key_id', d.instance.aws_key_id
+    assert_equal 'test_aws_sec_key', d.instance.aws_sec_key
+    assert_equal 'https://test_endpoint', d.instance.endpoint
+    assert_equal ['test_hash_key1','test_hash_key2'], d.instance.hash_key
+    assert_equal ':', d.instance.hash_key_delimiter
+    assert_equal '3', d.instance.add_hash_key_prefix
+    assert_equal nil, d.instance.range_key
+  end
+end
+