AbstractHsqlsbTest.java

/**
 * Licensed 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.
 *
 * Copyright 2012-2016 the original author or authors.
 */
package org.assertj.db.database.hsqldb;

import com.ninja_squad.dbsetup.DbSetup;
import com.ninja_squad.dbsetup.destination.DriverManagerDestination;
import com.ninja_squad.dbsetup.operation.Operation;
import org.assertj.db.database.AbstractDatabaseTest;
import org.assertj.db.type.DataSourceWithLetterCase;
import org.assertj.db.type.SourceWithLetterCase;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.ContextConfiguration;

import javax.sql.DataSource;
import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.sql.Date;
import java.sql.Time;
import java.sql.Timestamp;
import java.util.Locale;

import static com.ninja_squad.dbsetup.Operations.*;

/**
 * Parent for all the tests which are specific for H2 database.
 *
 * @author Régis Pouiller
 */
@ContextConfiguration(classes = { HsqlsbConfiguration.class })
public abstract class AbstractHsqlsbTest extends AbstractDatabaseTest {

  private static byte[] BYTES;

  static {
    try {
      InputStream inputStream = AbstractDatabaseTest.class.getResourceAsStream("/h2-logo-2.png");
      ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
      byte[] bytes = new byte[1024];
      int nb = inputStream.read(bytes);
      while (nb > 0) {
        byteArrayOutputStream.write(bytes, 0, nb);
        nb = inputStream.read(bytes);
      }
      BYTES = byteArrayOutputStream.toByteArray();
    } catch (Exception e) {
      e.printStackTrace();
    }
  }

  protected DataSourceWithLetterCase dataSourceUIUIUI;
  protected DataSourceWithLetterCase dataSourceNSNSNS;

  protected final SourceWithLetterCase sourceUIUIUI = new SourceWithLetterCase("jdbc:hsqldb:mem:testHsqldb", "SA", "",
                                                                               letterCaseUI,
                                                                               letterCaseUI,
                                                                               letterCaseUI);
  protected final SourceWithLetterCase sourceNSNSNS = new SourceWithLetterCase("jdbc:hsqldb:mem:testHsqldb", "SA", "",
                                                                               letterCaseNS,
                                                                               letterCaseNS,
                                                                               letterCaseNS);

  private static final Operation DELETE_ALL = deleteAllFrom("teSt");

  private static final Operation INSERT_TEST = insertInto("teSt")
          .columns("Var1", "vAr2", "vaR3", "var4", "var5", "var6", "var7", "var8", "var9", "var10",
                   "var11", "var12", "var13", "var14", "var15", "var16", "var17", "var18", "var19", "var20",
                   "var21", "var22", "var23", "var24", "var25", "var26")
          .values(1, 2, 3.3, 4.4, "5", "6", "7", "8", "9", Date.valueOf("2007-12-23"),
                  Time.valueOf("09:01:00"), Timestamp.valueOf("2007-12-23 09:01:00"),
                  Timestamp.valueOf("2007-12-23 09:01:00"), 10, 11, true, false, 12, 13, 14,
                  15, BYTES, BYTES, BYTES, Locale.FRENCH, Locale.FRENCH)
          .build();

  private static final Operation OPERATIONS = sequenceOf(DELETE_ALL, INSERT_TEST);

  static {
    DB_SETUP = new DbSetup(new DriverManagerDestination("jdbc:hsqldb:mem:testHsqldb", "SA", ""), OPERATIONS);
  }

  @Autowired(required = true)
  protected void setDataSource(DataSource dataSource) {
    this.dataSourceUIUIUI = new DataSourceWithLetterCase(dataSource, letterCaseUI, letterCaseUI, letterCaseUI);
    this.dataSourceNSNSNS = new DataSourceWithLetterCase(dataSource, letterCaseNS, letterCaseNS, letterCaseNS);
  }

  protected void update() {
    update("update test set vAr2=20");
  }
}