fluent-plugin-postgresql-csvlog 0.3.2 → 0.7.0

data/lib/fluent/plugin/polling_pg_input_plugin.rb

@@ -0,0 +1,100 @@
+# frozen_string_literal: true
+
+require 'fluent/plugin/input'
+require 'pg'
+require 'pg_query'
+
+module Fluent::Plugin
+  # PollingPostgresInputPlugin is intended to be used as an base class
+  # for input plugins that poll postgres.
+  #
+  # Child classes should implement the `on_poll` method
+  class PollingPostgresInputPlugin < Input
+    desc 'PostgreSQL host'
+    config_param :host, :string
+
+    desc 'RDBMS port (default: 5432)'
+    config_param :port, :integer, default: 5432
+
+    desc 'login user name'
+    config_param :username, :string, default: nil
+
+    desc 'postgres db'
+    config_param :dbname, :string, default: nil
+
+    desc 'login password'
+    config_param :password, :string, default: nil, secret: true
+
+    # See https://www.postgresql.org/docs/current/libpq-connect.html#LIBPQ-CONNECT-SSLMODE
+    # for options
+    desc 'postgres sslmode'
+    config_param :sslmode, :string, default: 'prefer'
+
+    desc 'tag'
+    config_param :tag, :string, default: nil
+
+    desc 'interval in second to run query'
+    config_param :interval, :time, default: 300
+
+    def start
+      @stop_flag = false
+      @thread = Thread.new(&method(:thread_main))
+    end
+
+    # Fluentd shutdown method, called to terminate and cleanup plugin
+    def shutdown
+      @stop_flag = true
+
+      # Interrupt thread and wait for it to finish
+      Thread.new { @thread.run } if @thread
+      @thread.join
+    end
+
+    # Main polling loop on thread
+    def thread_main
+      until @stop_flag
+        sleep @interval
+        break if @stop_flag
+
+        begin
+          on_poll
+        rescue StandardError => e
+          log.error 'unexpected error', error: e.message, error_class: e.class
+          log.error_backtrace
+        end
+      end
+    end
+
+    protected
+
+    # Child-classes should implement this method
+    def on_poll
+      raise 'on_poll must be implemented by descendents of PollingPostgresInputPlugin'
+    end
+
+    # Since this query is very infrequent, and it may be communicating directly
+    # with postgres without pgbouncer, don't use a persistent connection and
+    # ensure that it is properly closed
+    def with_connection(&block)
+      conn = PG.connect(
+        host: @host,
+        dbname: @dbname,
+        sslmode: @sslmode,
+        user: @username,
+        password: @password
+      )
+
+      map = PG::BasicTypeMapForResults.new(conn)
+      map.default_type_map = PG::TypeMapAllStrings.new
+
+      conn.type_map_for_results = map
+
+      begin
+        block.call(conn)
+      ensure
+        # Always close the connection
+        conn.finish
+      end
+    end
+  end
+end

data/test/helper.rb

@@ -16,4 +16,6 @@ Test::Unit::TestCase.extend(Fluent::Test::Helpers)
 require 'fluent/plugin/filter_postgresql_slowlog'
 require 'fluent/plugin/filter_postgresql_redactor'
 require 'fluent/plugin/filter_marginalia'
+require 'fluent/plugin/marginalia_extractor'
 require 'fluent/plugin/in_pg_stat_statements'
+require 'fluent/plugin/in_pg_stat_activity'

data/test/plugin/itest_in_pg_stat_activity.rb

@@ -0,0 +1,115 @@
+# frozen_string_literal: true
+
+require_relative '../helper'
+
+class PgStatActivityInputIntegrationTest < Test::Unit::TestCase
+  # The defaults values work with the configuration in .gitlab-ci.yml on the postgres service
+  # Override with env vars for local development
+  HOST = ENV.fetch('PG_TEST_HOST', 'postgres')
+  USERNAME = ENV.fetch('PG_TEST_USER', 'testuser')
+  PASSWORD = ENV.fetch('PG_TEST_PASSWORD', 'testpass')
+
+  def setup
+    Fluent::Test.setup
+
+    @conn = PG.connect(
+      host: HOST,
+      user: USERNAME,
+      password: PASSWORD
+    )
+  end
+
+  def teardown
+    @conn&.finish
+  end
+
+  VALID_CONFIG = %(
+    tag postgres.pg_stat_statements
+    host #{HOST}
+    username #{USERNAME}
+    password #{PASSWORD}
+    interval 1
+  )
+
+  INVALID_CONFIG = %(
+    host 'invalid_host.dne'
+    port 1234
+    username #{USERNAME}
+    password #{PASSWORD}
+    interval 1
+  )
+
+  def create_driver(config)
+    Fluent::Test::InputTestDriver.new(Fluent::Plugin::PgStatActivityInput).configure(config)
+  end
+
+  sub_test_case 'configuration' do
+    test 'connects' do
+      d = create_driver(VALID_CONFIG)
+
+      emits = []
+      # wait 50 * 0.05, "see fluentd/lib/fluent/test/base.rb:79 num_waits.times { sleep 0.05 }
+      d.run(num_waits = 50) do
+        emits = d.emits
+      end
+
+      assert_false emits.empty?
+    end
+
+    # Why do we have this test? If postgres is still starting up, we don't want to cause the
+    # the fluentd configuration to fail. We would rather retry until we get a connection
+    test 'connects for an invalid config' do
+      d = create_driver(INVALID_CONFIG)
+
+      emits = []
+      # wait 50 * 0.05, "see fluentd/lib/fluent/test/base.rb:79 num_waits.times { sleep 0.05 }
+      d.run(num_waits = 50) do
+        emits = d.emits
+      end
+
+      assert_true emits.empty?
+    end
+  end
+
+  sub_test_case 'execution' do
+    test 'connects' do
+      d = create_driver(VALID_CONFIG)
+
+      emits = []
+      # wait 50 * 0.05, "see fluentd/lib/fluent/test/base.rb:79 num_waits.times { sleep 0.05 }
+      d.run(num_waits = 50) do
+        emits = d.emits
+      end
+
+      first_with_query = emits.find do |event|
+        record = event[2]
+
+        record['usename'] == USERNAME &&
+          !record['datid'].nil? &&
+          !record['query'].nil? &&
+          record['state'] == 'active'
+      end
+
+      assert_false first_with_query.nil?
+      record = first_with_query[2]
+
+      assert_false record['datname'].nil?
+      assert_false record['pid'].nil?
+      assert_false record['usesysid'].nil?
+      assert_false record['application_name'].nil?
+      assert_false record['client_addr'].nil?
+      assert_false record['client_port'].nil?
+      assert_false record['xact_start'].nil?
+      assert_false record['xact_age_s'].nil?
+      assert_false record['xact_start'].nil?
+      assert_false record['xact_age_s'].nil?
+      assert_false record['query_start'].nil?
+      assert_false record['query_age_s'].nil?
+      assert_false record['state_change'].nil?
+      assert_false record['state_age_s'].nil?
+      assert_false record['query_length'].nil?
+      assert_false record['query'].nil?
+      assert_false record['fingerprint'].nil?
+    end
+  end
+end

data/test/plugin/itest_in_pg_stat_statements.rb

@@ -98,8 +98,7 @@ class PgStatStatementsInputIntegrationTest < Test::Unit::TestCase
       expected_record = {
         'fingerprint' => '8a6e9896bd9048a2',
         'query' => 'SELECT * FROM pg_stat_statements ORDER BY queryid LIMIT $1',
-        'query_length' => 58,
-        'queryid' => 3_239_318_621_761_098_074
+        'query_length' => 58
       }
       known_statement_event = emits.find do |event|
         record = event[2]
@@ -114,7 +113,8 @@ class PgStatStatementsInputIntegrationTest < Test::Unit::TestCase
       assert_equal 'postgres.pg_stat_statements', tag
       assert_equal expected_record['fingerprint'], record['fingerprint']
       assert_equal expected_record['query_length'], record['query_length']
-      assert_true expected_record.include? 'queryid'
+      assert_true record.include? 'queryid'
+      assert_true record['queryid'].is_a? String
     end
   end
 end

data/test/plugin/test_in_pg_stat_activity.rb

@@ -0,0 +1,223 @@
+# frozen_string_literal: true
+
+require_relative '../helper'
+require 'date'
+
+class PgStatActivityInputTest < Test::Unit::TestCase
+  def setup
+    Fluent::Test.setup
+  end
+
+  CONFIG = %(
+    tag postgres.pg_stat_activity
+    host localhost
+    port 1234
+    dbname gitlab
+    sslmode require
+    username moo
+    password secret
+    interval 600
+    fingerprint_key fingerprint
+  )
+
+  def create_driver
+    Fluent::Test::InputTestDriver.new(Fluent::Plugin::PgStatActivityInput).configure(CONFIG)
+  end
+
+  sub_test_case 'configuration' do
+    test 'basic configuration' do
+      d = create_driver
+
+      assert_equal 'postgres.pg_stat_activity', d.instance.tag
+      assert_equal 'localhost', d.instance.host
+      assert_equal 1234, d.instance.port
+      assert_equal 'gitlab', d.instance.dbname
+      assert_equal 'require', d.instance.sslmode
+      assert_equal 'moo', d.instance.username
+      assert_equal 'secret', d.instance.password
+      assert_equal 600, d.instance.interval
+      assert_equal 'fingerprint', d.instance.fingerprint_key
+    end
+  end
+
+  sub_test_case 'execution' do
+    test 'sql' do
+      d = create_driver
+      row = {
+        'datid' => 16384,
+        'datname' => 'testuser',
+        'pid' => 376,
+        'usesysid' => 10,
+        'usename' => 'testuser',
+        'application_name' => 'psql',
+        'client_addr' => '172.17.0.1',
+        'client_hostname' => nil,
+        'client_port' => 60182,
+        'xact_start' => Time.parse('2021-07-23 12:55:25 +0000'),
+        'xact_age_s' => 0.001884,
+        'query_start' => Time.parse('2021-07-23 12:55:25 +0000'),
+        'query_age_s' => 0.001894,
+        'state_change' => Time.parse('2021-07-23 12:55:25 +0000'),
+        'state_age_s' => 0.001894,
+        'state' => 'active',
+        'query' => "SELECT * FROM users WHERE user_secret = 's3cr3t'"
+      }
+
+      record = d.instance.record_for_row(row)
+
+      expected = {
+        'application_name' => 'psql',
+        'client_addr' => '172.17.0.1',
+        'client_hostname' => nil,
+        'client_port' => 60182,
+        'datid' => 16384,
+        'datname' => 'testuser',
+        'fingerprint' => '5c4a61e156c7d822',
+        'pid' => 376,
+        'query' => 'SELECT * FROM users WHERE user_secret = $1', # NOTE: secret removed
+        'query_age_s' => 0.001894,
+        'query_length' => 48,
+        'query_start' => '2021-07-23T12:55:25.000+00:00',
+        'state' => 'active',
+        'state_age_s' => 0.001894,
+        'state_change' => '2021-07-23T12:55:25.000+00:00',
+        'usename' => 'testuser',
+        'usesysid' => 10,
+        'xact_age_s' => 0.001884,
+        'xact_start' => '2021-07-23T12:55:25.000+00:00'
+      }
+
+      assert_equal expected, record
+    end
+
+    # This test mostly checks that the code is nil safe
+    test 'nil query' do
+      d = create_driver
+      record = d.instance.record_for_row({})
+
+      expected = {
+        'application_name' => nil,
+        'client_addr' => nil,
+        'client_hostname' => nil,
+        'client_port' => nil,
+        'datid' => nil,
+        'datname' => nil,
+        'pid' => nil,
+        'query' => nil,
+        'query_age_s' => nil,
+        'query_length' => nil,
+        'query_start' => nil,
+        'state' => nil,
+        'state_age_s' => nil,
+        'state_change' => nil,
+        'usename' => nil,
+        'usesysid' => nil,
+        'xact_age_s' => nil,
+        'xact_start' => nil
+      }
+
+      assert_equal expected, record
+    end
+
+    test 'unparseable sql' do
+      d = create_driver
+      row = {
+        'datid' => 16384,
+        'datname' => 'testuser',
+        'pid' => 376,
+        'usesysid' => 10,
+        'usename' => 'testuser',
+        'application_name' => 'psql',
+        'client_addr' => '172.17.0.1',
+        'client_hostname' => nil,
+        'client_port' => 60182,
+        'xact_start' => Time.parse('2021-07-23 12:55:25 +0000'),
+        'xact_age_s' => 0.001884,
+        'query_start' => Time.parse('2021-07-23 12:55:25 +0000'),
+        'query_age_s' => 0.001894,
+        'state_change' => Time.parse('2021-07-23 12:55:25 +0000'),
+        'state_age_s' => 0.001894,
+        'state' => 'active',
+        'query' => "SELECT * FROM users WHERE user_se="
+      }
+
+      record = d.instance.record_for_row(row)
+
+      expected = {
+        'application_name' => 'psql',
+        'client_addr' => '172.17.0.1',
+        'client_hostname' => nil,
+        'client_port' => 60182,
+        'datid' => 16384,
+        'datname' => 'testuser',
+        'pid' => 376,
+        'query' => nil,
+        'query_age_s' => 0.001894,
+        'query_length' => 34,
+        'query_start' => '2021-07-23T12:55:25.000+00:00',
+        'query_unparseable' => true,
+        'state' => 'active',
+        'state_age_s' => 0.001894,
+        'state_change' => '2021-07-23T12:55:25.000+00:00',
+        'usename' => 'testuser',
+        'usesysid' => 10,
+        'xact_age_s' => 0.001884,
+        'xact_start' => '2021-07-23T12:55:25.000+00:00'
+      }
+
+      assert_equal expected, record
+    end
+
+    test 'marginalia prepended' do
+      d = create_driver
+      row = {
+        'datid' => 16384,
+        'datname' => 'testuser',
+        'pid' => 376,
+        'usesysid' => 10,
+        'usename' => 'testuser',
+        'application_name' => 'psql',
+        'client_addr' => '172.17.0.1',
+        'client_hostname' => nil,
+        'client_port' => 60182,
+        'xact_start' => Time.parse('2021-07-23 12:55:25 +0000'),
+        'xact_age_s' => 0.001884,
+        'query_start' => Time.parse('2021-07-23 12:55:25 +0000'),
+        'query_age_s' => 0.001894,
+        'state_change' => Time.parse('2021-07-23 12:55:25 +0000'),
+        'state_age_s' => 0.001894,
+        'state' => 'active',
+        'query' => "/*application:web,correlation_id:01F1D2T1SC9DM82A4865ATG1CP,endpoint_id:POST /api/:version/groups/:id/-/packages/mavenpath/:file_name*/ SELECT * FROM users WHERE user_secret = 's3cr3t'"
+      }
+
+      record = d.instance.record_for_row(row)
+
+      expected = {
+        'application' => 'web',
+        'application_name' => 'psql',
+        'client_addr' => '172.17.0.1',
+        'client_hostname' => nil,
+        'client_port' => 60182,
+        'correlation_id' => '01F1D2T1SC9DM82A4865ATG1CP',
+        'datid' => 16384,
+        'datname' => 'testuser',
+        'endpoint_id' => 'POST /api/:version/groups/:id/-/packages/mavenpath/:file_name',
+        'fingerprint' => '5c4a61e156c7d822',
+        'pid' => 376,
+        'query' => 'SELECT * FROM users WHERE user_secret = $1', # Secret removed
+        'query_age_s' => 0.001894,
+        'query_length' => 48,
+        'query_start' => '2021-07-23T12:55:25.000+00:00',
+        'state' => 'active',
+        'state_age_s' => 0.001894,
+        'state_change' => '2021-07-23T12:55:25.000+00:00',
+        'usename' => 'testuser',
+        'usesysid' => 10,
+        'xact_age_s' => 0.001884,
+        'xact_start' => '2021-07-23T12:55:25.000+00:00'
+      }
+
+      assert_equal expected, record
+    end
+  end
+end

data/test/plugin/test_in_pg_stat_statements.rb

@@ -42,13 +42,24 @@ class PgStatStatementsInputTest < Test::Unit::TestCase
   sub_test_case 'execution' do
     test 'sql' do
       d = create_driver
-      record = d.instance.record_for_row({ 'queryid' => '1234', 'query' => 'SELECT * FROM users WHERE user_id = ?' })
+      row = {
+        'queryid' => '1234',
+        'query' => 'SELECT * FROM users WHERE user_id = ?',
+        'calls' => 22,
+        'rows' => 333,
+        'total_time' => 44.44
+      }
+
+      record = d.instance.record_for_row(row)
 
       expected = {
         'fingerprint' => 'c071dee80d466e7d',
         'query' => 'SELECT * FROM users WHERE user_id = ?',
         'query_length' => 37,
-        'queryid' => '1234'
+        'queryid' => '1234',
+        'calls' => 22,
+        'rows' => 333,
+        'total_time_ms' => 44.44
       }
 
       assert_equal expected, record
@@ -56,9 +67,22 @@ class PgStatStatementsInputTest < Test::Unit::TestCase
 
     test 'nil query' do
       d = create_driver
-      record = d.instance.record_for_row({ 'queryid' => '1234', 'query' => nil })
+      row = {
+        'queryid' => '1234',
+        'query' => nil,
+        'calls' => nil,
+        'rows' => nil,
+        'total_time' => nil
+      }
+      record = d.instance.record_for_row(row)
 
-      expected = { 'query_length' => nil, 'queryid' => '1234' }
+      expected = {
+        'query_length' => nil,
+        'queryid' => '1234',
+        'calls' => nil,
+        'rows' => nil,
+        'total_time_ms' => nil
+      }
       assert_equal expected, record
     end
 
@@ -75,26 +99,48 @@ class PgStatStatementsInputTest < Test::Unit::TestCase
         )
       SQL
 
-      record = d.instance.record_for_row({ 'queryid' => '1234', 'query' => ddl_sql })
+      row = {
+        'queryid' => 1234,
+        'query' => ddl_sql,
+        'calls' => 22,
+        'rows' => 333,
+        'total_time' => 44.44
+      }
+
+      record = d.instance.record_for_row(row)
 
       expected = {
         'fingerprint' => 'fa9c9d26757c4f9b',
         'query' => ddl_sql,
         'query_length' => 287,
-        'queryid' => '1234'
+        'queryid' => '1234',
+        'calls' => 22,
+        'rows' => 333,
+        'total_time_ms' => 44.44
       }
       assert_equal expected, record
     end
 
     test 'set command' do
       d = create_driver
-      record = d.instance.record_for_row({ 'queryid' => '1234', 'query' => "SET TIME ZONE 'PST8PDT'" })
+      row = {
+        'queryid' => 1234,
+        'query' => "SET TIME ZONE 'PST8PDT'",
+        'calls' => 22,
+        'rows' => 333,
+        'total_time' => 44.44
+      }
+
+      record = d.instance.record_for_row(row)
 
       expected = {
         'fingerprint' => '23f8d6eb1d3125c3',
         'query' => 'SET TIME ZONE $1',
         'query_length' => 23,
-        'queryid' => '1234'
+        'queryid' => '1234',
+        'calls' => 22,
+        'rows' => 333,
+        'total_time_ms' => 44.44
       }
 
       assert_equal expected, record
@@ -102,9 +148,17 @@ class PgStatStatementsInputTest < Test::Unit::TestCase
 
     test 'unparseable sql' do
       d = create_driver
-      record = d.instance.record_for_row({ 'queryid' => '1234', 'query' => 'SELECT * FROM' })
+      record = d.instance.record_for_row({ 'queryid' => 1234, 'query' => 'SELECT * FROM' })
+
+      expected = {
+        'query_length' => 13,
+        'query_unparseable' => true,
+        'queryid' => '1234',
+        'calls' => nil,
+        'rows' => nil,
+        'total_time_ms' => nil
+      }
 
-      expected = { 'query_length' => 13, 'query_unparseable' => true, 'queryid' => '1234' }
       assert_equal expected, record
     end
   end

data/test/plugin/test_marginalia_extractor.rb

@@ -0,0 +1,103 @@
+# frozen_string_literal: true
+
+require_relative '../helper'
+
+class Marginalia < Test::Unit::TestCase
+  include Fluent::Plugin::MarginaliaExtractor
+
+  def test_parse(sql, record, key, strip_comment, expected)
+    record[key] = sql
+    parse_marginalia_into_record(record, key, strip_comment)
+    assert_equal(expected, record)
+  end
+
+  test 'no marginalia' do
+    sql = 'SELECT * FROM projects'
+    expected = { 'sql' => 'SELECT * FROM projects' }
+    test_parse(sql, {}, 'sql', true, expected)
+  end
+
+  test 'normal comment appended' do
+    sql = 'SELECT COUNT(*) FROM "projects" /* this is just a comment */'
+    expected = {
+      'sql' => 'SELECT COUNT(*) FROM "projects"'
+    }
+    test_parse(sql, {}, 'sql', true, expected)
+  end
+
+  test 'marginalia appended for sidekiq' do
+    sql = 'SELECT COUNT(*) FROM "projects" /*application:sidekiq,correlation_id:d67cae54c169e0cab7d73389e2934f0e,jid:52a1c8a9e4c555ea573f20f0,job_class:Geo::MetricsUpdateWorker*/'
+    expected = {
+      'application' => 'sidekiq',
+      'correlation_id' => 'd67cae54c169e0cab7d73389e2934f0e',
+      'jid' => '52a1c8a9e4c555ea573f20f0',
+      'job_class' => 'Geo::MetricsUpdateWorker',
+      'sql' => 'SELECT COUNT(*) FROM "projects"'
+    }
+    test_parse(sql, {}, 'sql', true, expected)
+  end
+
+  test 'marginalia appended for web' do
+    sql = 'SELECT COUNT(*) FROM "projects" /*application:web,correlation_id:01F1D2T1SC9DM82A4865ATG1CP,endpoint_id:POST /api/:version/groups/:id/-/packages/mavenpath/:file_name*/'
+    expected = {
+      'application' => 'web',
+      'correlation_id' => '01F1D2T1SC9DM82A4865ATG1CP',
+      'endpoint_id' => 'POST /api/:version/groups/:id/-/packages/mavenpath/:file_name',
+      'sql' => 'SELECT COUNT(*) FROM "projects"'
+    }
+    test_parse(sql, {}, 'sql', true, expected)
+  end
+
+  test 'normal comment prepended' do
+    sql = '/* this is just a comment */ SELECT COUNT(*) FROM "projects"'
+    expected = {
+      "sql" => 'SELECT COUNT(*) FROM "projects"'
+    }
+    test_parse(sql, {}, 'sql', true, expected)
+  end
+
+  test 'marginalia prepended for sidekiq' do
+    sql = '/*application:sidekiq,correlation_id:d67cae54c169e0cab7d73389e2934f0e,jid:52a1c8a9e4c555ea573f20f0,job_class:Geo::MetricsUpdateWorker*/ SELECT COUNT(*) FROM "projects"'
+    expected = {
+      'application' => 'sidekiq',
+      'correlation_id' => 'd67cae54c169e0cab7d73389e2934f0e',
+      'jid' => '52a1c8a9e4c555ea573f20f0',
+      'job_class' => 'Geo::MetricsUpdateWorker',
+      'sql' => 'SELECT COUNT(*) FROM "projects"'
+    }
+    test_parse(sql, {}, 'sql', true, expected)
+  end
+
+  test 'marginalia prepended for web' do
+    sql = '/*application:web,correlation_id:01F1D2T1SC9DM82A4865ATG1CP,endpoint_id:POST /api/:version/groups/:id/-/packages/mavenpath/:file_name*/ SELECT COUNT(*) FROM "projects"'
+    expected = {
+      'application' => 'web',
+      'correlation_id' => '01F1D2T1SC9DM82A4865ATG1CP',
+      'endpoint_id' => 'POST /api/:version/groups/:id/-/packages/mavenpath/:file_name',
+      'sql' => 'SELECT COUNT(*) FROM "projects"'
+    }
+    test_parse(sql, {}, 'sql', true, expected)
+  end
+
+  test 'marginalia prepended for web, comment_strip disabled' do
+    sql = 'SELECT COUNT(*) FROM "projects" /*application:sidekiq,correlation_id:d67cae54c169e0cab7d73389e2934f0e,jid:52a1c8a9e4c555ea573f20f0,job_class:Geo::MetricsUpdateWorker*/'
+    expected = {
+      'application' => 'sidekiq',
+      'correlation_id' => 'd67cae54c169e0cab7d73389e2934f0e',
+      'jid' => '52a1c8a9e4c555ea573f20f0',
+      'job_class' => 'Geo::MetricsUpdateWorker',
+      'sql' => sql
+    }
+    test_parse(sql, { 'sql' => sql }, 'sql', false, expected)
+  end
+
+  test 'avoid clash' do
+    sql = '/*clash_key:bbb*/ SELECT COUNT(*) FROM "projects"'
+    expected = {
+      'clash_key' => 'aaa',
+      'sql_clash_key' => 'bbb',
+      'sql' => 'SELECT COUNT(*) FROM "projects"'
+    }
+    test_parse(sql, { 'clash_key' => 'aaa' }, 'sql', true, expected)
+  end
+end