require 'fluent_plugins_spec_helper'
require 'flydata/fluent-plugins/mysql/binlog_query_dispatcher'

module Mysql

  QUERY_WITH_COMMENTS = <<EOT
-- ALTER TABLE QUERY comment
ALTER TABLE test_table
ADD /* test-comment */ COLUMN value10 -- aaaaa
-- bbbbb
VARCHAR(512) DEFAULT NULL
   -- flydata
   -- hapyrus
;
EOT
  NORMALIZED_QUERY_WITH_COMMENTS = "ALTER TABLE test_table ADD COLUMN value10 VARCHAR(512) DEFAULT NULL ;"

  QUERY_WITH_LEADING_NEWLINE = <<EOT

ALTER TABLE test_table
ADD /* test-comment */ COLUMN
value10 -- aaaaa
-- bbbbb
VARCHAR(512) DEFAULT NULL
   -- flydata
   -- hapyrus

;

EOT

  CUSTOMER_QUERY = <<EOT
-- ALTER TABLE UserPaymentMethods DROP COLUMN GoogleWallet;
ALTER TABLE UserPaymentMethods ADD COLUMN GoogleWallet VARCHAR(512) DEFAULT NULL
/*!*/;
EOT

  NORMALIZED_CUSTOMER_QUERY = "ALTER TABLE UserPaymentMethods ADD COLUMN GoogleWallet VARCHAR(512) DEFAULT NULL ;"

  QUERY_WITH_SPECIAL_CHARS_IN_DEFAULT = <<EOT
alter table test_table add (value1 varchar(26) not null default 'flydata -- special', value2 int auto_increment); -- real comment
EOT
  NORMALIZED_QUERY_WITH_SPECIAL_CHARS = "alter table test_table add (value1 varchar(26) not null default 'flydata -- special', value2 int auto_increment);"

  describe FlydataBinlogQueryDispatcher do
    let(:record) do
      r = double('record')
      allow(r).to receive(:[]).with("query").and_return(query)
      expect(r).to receive(:merge).with({"normalized_query" => normalized_query}).and_return(r)
      r
    end
    let(:alter_query_handler) do
      r = double('alter_query_handler')
      allow(r).to receive(:pattern).and_return(Mysql::AlterTableQueryHandler::PATTERN)
      r
    end
    let(:truncate_query_handler) do
      r = double('truncate_query_handler')
      allow(r).to receive(:pattern).and_return(Mysql::TruncateTableQueryHandler::PATTERN)
      r
    end
    let(:context) { double('context') }
    let(:subject_object) { described_class.new(context) }
    subject { subject_object }

    shared_examples "a dispatcher that calls query handler with correct query" do
      it do
        expect(correct_query_handler).to receive(:process).with(record)
        expect(subject.dispatch(record))
      end
    end

    before do
      expect(AlterTableQueryHandler).to receive(:new).with(context).and_return(alter_query_handler)
      expect(TruncateTableQueryHandler).to receive(:new).with(context).and_return(truncate_query_handler)
    end

    describe '#dispatch' do
      context "alter table queries" do
        let(:correct_query_handler) { alter_query_handler }
        context "query with comments" do
          let(:query) { QUERY_WITH_COMMENTS }
          let(:normalized_query) { NORMALIZED_QUERY_WITH_COMMENTS }

          it_behaves_like "a dispatcher that calls query handler with correct query"
        end
        context "customer query" do
          let(:query) { CUSTOMER_QUERY }
          let(:normalized_query) { NORMALIZED_CUSTOMER_QUERY }

          it_behaves_like "a dispatcher that calls query handler with correct query"
        end
        context "query with leading newline" do
          let(:query) { QUERY_WITH_LEADING_NEWLINE }
          let(:normalized_query) { NORMALIZED_QUERY_WITH_COMMENTS }

          it_behaves_like "a dispatcher that calls query handler with correct query"
        end
        context "query with special characters in DEFAULT" do
          let(:query) { QUERY_WITH_SPECIAL_CHARS_IN_DEFAULT }
          let(:normalized_query) { NORMALIZED_QUERY_WITH_SPECIAL_CHARS }

          it_behaves_like "a dispatcher that calls query handler with correct query"
        end
        context "query with preceding commands" do
          let(:query) { <<EOS }
use `mydatabase`/*!*/;
ALTER TABLE `apps` ADD `created_by_existing_admin` tinyint(1) DEFAULT 0 /*application:MyApp,line:/usr/local/lib/ruby/2.0.0/benchmark.rb:281:in `measure'*/
/*!*/;
SET TIMESTAMP=1415925954/*!*/;
EOS
          let(:normalized_query) { <<EOS.gsub(/\n$/, '') }
ALTER TABLE `apps` ADD `created_by_existing_admin` tinyint(1) DEFAULT 0 ;
EOS

          it_behaves_like "a dispatcher that calls query handler with correct query"
        end
        context "query with a comment with no space" do
          let(:query) { <<EOS.gsub(/\n$/, '') }
alter table test add/* adding a column */test varchar(4);
EOS
          let(:normalized_query) { <<EOS.gsub(/\n$/, '') }
alter table test add test varchar(4);
EOS

          it_behaves_like "a dispatcher that calls query handler with correct query"
        end
        context "query with strings including comment quotes" do
          let(:query) { <<EOS.gsub(/\n$/, '') }
alter table test add test varchar(256) DEFAULT ' /* ' COMMENT ' */ -- a part of COMMENT string';
EOS
          let(:normalized_query) { <<EOS.gsub(/\n$/, '') }
alter table test add test varchar(256) DEFAULT ' /* ' COMMENT ' */ -- a part of COMMENT string';
EOS

          it_behaves_like "a dispatcher that calls query handler with correct query"
        end
        context "query with comments including string quotes and ending without a line feed" do
          let(:query) { <<EOS.gsub(/\n$/, '') }
alter table /* /* ' */ test add test varchar(256) -- this is a comment';
EOS
          let(:normalized_query) { <<EOS.gsub(/\n$/, '') }
alter table test add test varchar(256) ;
EOS

          it_behaves_like "a dispatcher that calls query handler with correct query"
        end
        context "query with DEFAULT using double-quote" do
          let(:query) { <<EOS.gsub(/\n$/, '') }
alter table test add test varchar(256) DEFAULT "This # is default -- with \\\" in it";
EOS
          let(:normalized_query) { <<EOS.gsub(/\n$/, '') }
alter table test add test varchar(256) DEFAULT "This # is default -- with \\\" in it";
EOS

          it_behaves_like "a dispatcher that calls query handler with correct query"
        end
        context "query with # style comment" do
          let(:query) { <<EOS.gsub(/\n$/, '') }
alter table test add test varchar(256) # You never know this works in MySQL '"
  DEFAULT "This is default with \\\" in it"
EOS
          let(:normalized_query) { <<EOS.gsub(/\n$/, '') }
alter table test add test varchar(256) DEFAULT "This is default with \\\" in it";
EOS

          it_behaves_like "a dispatcher that calls query handler with correct query"
        end
        context "query with single quote string with escape" do
          let(:query) { <<EOS.gsub(/\n$/, '') }
alter table test add test varchar(256) COMMENT 'Stave\\'s # -- diner'
EOS
          let(:normalized_query) { <<EOS.gsub(/\n$/, '') }
alter table test add test varchar(256) COMMENT 'Stave\\'s # -- diner';
EOS

          it_behaves_like "a dispatcher that calls query handler with correct query"
        end
        context "query with quoted string with UTF8 3 byte chars" do
          let(:query) { <<EOS.gsub(/\n$/, '') }
alter table test add test varchar(256) DEFAULT "無国籍" COMMENT 'SJIS is トラブルメーカー'
EOS
          let(:normalized_query) { <<EOS.gsub(/\n$/, '') }
alter table test add test varchar(256) DEFAULT "無国籍" COMMENT 'SJIS is トラブルメーカー';
EOS

          it_behaves_like "a dispatcher that calls query handler with correct query"
        end
        context "query with comments with UTF8 3 byte chars" do
          let(:query) { <<EOS.gsub(/\n$/, '') }
alter table test add test varchar(256) /* 無国籍 */ -- 'SJIS is トラブルメーカー'
DEFAULT NULL;
EOS
          let(:normalized_query) { <<EOS.gsub(/\n$/, '') }
alter table test add test varchar(256) DEFAULT NULL;
EOS

          it_behaves_like "a dispatcher that calls query handler with correct query"
        end
      end
      context "truncate table queries" do
        let(:correct_query_handler) { truncate_query_handler }

        shared_examples "truncate table tests" do
          context "simple truncate query" do
            let(:query) { <<EOS.gsub(/\n$/, '') }
TRUNCATE#{opt_table}  users;
EOS
            let(:normalized_query) { <<EOS.gsub(/\n$/, '') }
TRUNCATE#{opt_table} users;
EOS

            it_behaves_like "a dispatcher that calls query handler with correct query"
          end
          context "truncate query with dbname and table name and a comment in the beginning" do
            let(:query) { <<EOS.gsub(/\n$/, '') }
# I am going to truncate this table
truncate#{opt_table} flydata_sync.users;
EOS
            let(:normalized_query) { <<EOS.gsub(/\n$/, '') }
truncate#{opt_table} flydata_sync.users;
EOS

            it_behaves_like "a dispatcher that calls query handler with correct query"
          end
          context "truncate query with quoted dbname and table name with comments on the same line" do
            let(:query) { <<EOS.gsub(/\n$/, '') }
truncate#{opt_table} `flydata_sync`.`users`; /* 無国籍 */ -- 'SJIS is トラブルメーカー'
EOS
            let(:normalized_query) { <<EOS.gsub(/\n$/, '') }
truncate#{opt_table} `flydata_sync`.`users`;
EOS

            it_behaves_like "a dispatcher that calls query handler with correct query"
          end
        end

        context "with table key word" do
          let(:opt_table) { " table" }
          include_examples "truncate table tests"
        end

        context "without table key word" do
          let(:opt_table) { "" }
          include_examples "truncate table tests"
        end
      end
    end
    describe '#normalize_query' do
      context 'straightforwad query' do
        let(:query) { "ALTER TABLE test_table DROP COLUMN test_column" }
        let(:expected_normalized) { ["#{query};"] }

        it do
          expect {|b| subject_object.send(:normalize_query, query, &b) }.to yield_successive_args(*expected_normalized)
        end
      end
      context 'two queries' do
        let(:subquery1) { "ALTER TABLE test_table DROP COLUMN test_column" }
        let(:subquery2) { "ALTER TABLE test_table2 DROP COLUMN test_column2" }
        let(:query) { "#{subquery1}; #{subquery2};" }
        let(:expected_normalized) { ["#{subquery1};", "#{subquery2};"] }

        it do
          expect {|b| subject_object.send(:normalize_query, query, &b) }.to yield_successive_args(*expected_normalized)
        end
      end
      context 'a column name includes a semicolon' do
        let(:subquery1) { %Q|ALTER TABLE test_table DROP COLUMN "test_;column"| }
        let(:subquery2) { "ALTER TABLE test_table2 DROP COLUMN test_column2" }
        let(:query) { "#{subquery1}; #{subquery2};" }
        let(:expected_normalized) { ["#{subquery1};", "#{subquery2};"] }

        it do
          expect {|b| subject_object.send(:normalize_query, query, &b) }.to yield_successive_args(*expected_normalized)
        end
      end
    end
  end
end