require_relative '../../../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)
      r
    end
    let(:query_handler) do
      r = double('query_handler')
      allow(r).to receive(:pattern).and_return(/^ALTER TABLE/i)
      expect(r).to receive(:process).with(record, normalized_query)
      r
    end
    let(:context) { double('context') }
    subject { described_class.new(context) }

    shared_examples "a dispatcher that calls query handler with correct query" do
      it do
        expect(AlterTableQueryHandler).to receive(:new).with(context).and_return(query_handler)
        expect(subject.dispatch(record))
      end
    end

    describe '#dispatch' do
      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
  end
end