require 'fluent_plugins_spec_helper'
require 'flydata/fluent-plugins/mysql/ddl_query_handler'
require 'flydata/fluent-plugins/mysql/shared_query_handler_context'

module Mysql
  describe DdlQueryHandler do
    include_context "query handler context"

    describe "#table_info" do
      context "alter table query with no db name" do
        let(:normalized_query) { "ALTER TABLE `apps` ADD `created_by_existing_admin` tinyint(1) DEFAULT 0;" }
        it do
          expect(subject.table_info(record)).to eq({db_name: "testdb", table_name: "apps"})
        end
      end
      context "alter table query with db name" do
        let(:normalized_query) { "alter table testdb1.testtable1 add testcol varchar(256) DEFAULT ' /* ' COMMENT ' */ -- a part of COMMENT string' ;" }
        it do
          expect(subject.table_info(record)).to eq({db_name: "testdb1", table_name: "testtable1"})
        end
      end
      context "alter table query with quoted db name" do
        let(:normalized_query) { <<EOS.gsub(/\n$/, '') }
alter table `testdb2`.`testtable2` add test varchar(256) DEFAULT "This # is default -- with \\\" in it";
EOS
        it do
          expect(subject.table_info(record)).to eq({db_name: "testdb2", table_name: "testtable2"})
        end
      end
      shared_examples "truncate table queries" do
        context "truncate table query with no db name" do
          let(:normalized_query) { "TRUNCATE#{opt_table} apps ;" }
          it do
            expect(subject.table_info(record)).to eq({db_name: "testdb", table_name: "apps"})
          end
        end
        context "truncate table query with db name" do
          let(:normalized_query) { "truncate#{opt_table} `testdb1`.`apps1`;" }
          it do
            expect(subject.table_info(record)).to eq({db_name: "testdb1", table_name: "apps1"})
          end
        end
        context "truncate table query with quoted db name" do
          let(:normalized_query) { "truncate#{opt_table} testdb2.apps2;" }
          it do
            expect(subject.table_info(record)).to eq({db_name: "testdb2", table_name: "apps2"})
          end
        end
      end
      context "truncate table query with table key word" do
        let(:opt_table) { " TABLE" }
        include_examples "truncate table queries"
      end
      context "truncate table query without table key word" do
        let(:opt_table) { "" }
        include_examples "truncate table queries"
      end
      context "query does not match DDL_TABLE_QUERY" do #This should not happen, ideally
        let(:normalized_query) { "CREATE TRIGGER mytrigger BEFORE INSERT ON TABLE_1 FOR EACH ROW SET NEW.MY_DATETIME_COLUMN = NOW();" }
        it do
          expect(subject.table_info(record)).to eq({db_name: "testdb", table_name: nil})
        end
      end
    end
  end
end