extensions-contrib/sqlserver-metadata-storage/src/test/java/org/apache/druid/metadata/storage/sqlserver/CustomStatementRewriterTest.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

package org.apache.druid.metadata.storage.sqlserver;

import junit.framework.Assert;
import org.apache.druid.metadata.storage.sqlserver.SQLServerConnector.CustomStatementRewriter;
import org.junit.Before;
import org.junit.Test;
import org.skife.jdbi.v2.Binding;
import org.skife.jdbi.v2.StatementContext;
import org.skife.jdbi.v2.exceptions.UnableToCreateStatementException;
import org.skife.jdbi.v2.tweak.RewrittenStatement;

@SuppressWarnings("nls")
public class CustomStatementRewriterTest
{

  private CustomStatementRewriter customStatementRewriter;
  private Binding params;
  private StatementContext ctx;

  @Before
  public void setUp()
  {
    customStatementRewriter = new CustomStatementRewriter();

    params = null;
    ctx = null;
  }

  private String rewrite(String sql)
  {
    RewrittenStatement rewrittenStatement = customStatementRewriter.rewrite(sql, params, ctx);
    return rewrittenStatement.getSql();
  }

  @Test
  public void testExactPatternReplacement()
  {

    Assert.assertEquals("BIT NOT NULL DEFAULT (0)", rewrite("BOOLEAN NOT NULL DEFAULT FALSE"));
    Assert.assertEquals("BIT NOT NULL DEFAULT (1)", rewrite("BOOLEAN NOT NULL DEFAULT TRUE"));
    Assert.assertEquals("BIT NOT NULL DEFAULT (0)", rewrite("BOOLEAN DEFAULT FALSE"));
    Assert.assertEquals("BIT NOT NULL DEFAULT (1)", rewrite("BOOLEAN DEFAULT TRUE"));
    Assert.assertEquals("BIT", rewrite("BOOLEAN"));
    Assert.assertEquals("1", rewrite("TRUE"));
    Assert.assertEquals("0", rewrite("FALSE"));
  }

  /**
   * See https://github.com/jdbi/jdbi/blob/jdbi2/src/test/java/org/skife/jdbi/v2/TestColonStatementRewriter.java
   */
  @Test
  public void testCustomStatementRewriter()
  {

    Assert.assertEquals("select column# from table1 where id = ?",
        rewrite("select column# from table1 where id = :id"));

    Assert.assertEquals("select * from table2\n where id = ?", rewrite("select * from table2\n where id = :id"));

    try {
      rewrite("select * from table3 where id = :\u0091\u009c"); // Control codes
                                                                // -
                                                                // https://en.wikipedia.org/wiki/List_of_Unicode_characters
      Assert.fail("Expected 'UnableToCreateStatementException'");
    }
    catch (UnableToCreateStatementException e) {
      // expected
    }

  }

  /**
   *
   * @see org.apache.druid.metadata.SQLMetadataConnector#createTable(String, Iterable)
   *
   */
  @Test
  public void testSQLMetadataConnectorCreateTable()
  {
    String sqlIn = "CREATE TABLE %1$s (\n"
        + "  id VARCHAR(255) NOT NULL,\n"
        + "  dataSource VARCHAR(255) NOT NULL,\n"
        + "  created_date VARCHAR(255) NOT NULL,\n"
        + "  start VARCHAR(255) NOT NULL,\n"
        + "  `end` VARCHAR(255) NOT NULL,\n"
        + "  partitioned BOOLEAN NOT NULL,\n"
        + "  version VARCHAR(255) NOT NULL,\n"
        + "  used BOOLEAN NOT NULL,\n"
        + "  payload %2$s NOT NULL,\n"
        + "  PRIMARY KEY (id)\n"
        + ")";

    String sqlOut = "CREATE TABLE %1$s (\n" +
        "  id VARCHAR(255) NOT NULL,\n" +
        "  dataSource VARCHAR(255) NOT NULL,\n" +
        "  created_date VARCHAR(255) NOT NULL,\n" +
        "  start VARCHAR(255) NOT NULL,\n" +
        "  `end` VARCHAR(255) NOT NULL,\n" +
        "  partitioned BIT NOT NULL,\n" +
        "  version VARCHAR(255) NOT NULL,\n" +
        "  used BIT NOT NULL,\n" +
        "  payload %2$s NOT NULL,\n" +
        "  PRIMARY KEY (id)\n" +
        ")";

    Assert.assertEquals(sqlOut, rewrite(sqlIn));

  }

  /**
   *
   * @see org.apache.druid.metadata.SQLMetadataStorageActionHandler#setStatus(String,
   *      boolean, Object)
   *
   */
  @Test
  public void testSQLMetadataStorageActionHandlerSetStatus()
  {
    Assert.assertEquals("UPDATE %s SET active = ?, status_payload = ? WHERE id = ? AND active = 1",
        rewrite("UPDATE %s SET active = :active, status_payload = :status_payload WHERE id = :id AND active = TRUE"));

  }
}