ebean/src/test/java/org/tests/basic/encrypt/TestEncrypt.java

package org.tests.basic.encrypt;

import io.ebean.BaseTestCase;
import io.ebean.DB;
import io.ebean.SqlRow;
import io.ebean.annotation.ForPlatform;
import io.ebean.annotation.Platform;
import io.ebean.config.dbplatform.DbEncrypt;
import io.ebeaninternal.api.SpiEbeanServer;
import org.ebeantest.LoggedSqlCollector;
import org.junit.Test;
import org.tests.model.basic.EBasicEncrypt;
import org.tests.model.basic.EBasicEncryptRelate;

import java.sql.Date;
import java.util.List;

import static org.assertj.core.api.Assertions.assertThat;
import static org.junit.Assert.assertEquals;

public class TestEncrypt extends BaseTestCase {


  @Test
  @ForPlatform(Platform.H2) // only run this on H2 - PGCrypto not happy on CI server
  public void testQueryBind() {

    LoggedSqlCollector.start();
    DB.find(EBasicEncrypt.class)
      .where().startsWith("description", "Rob")
      .findList();

    List<String> loggedSql = LoggedSqlCollector.stop();
    assertThat(loggedSql).hasSize(1);
    assertThat(loggedSql.get(0)).contains("; --bind(****,Rob%)");
  }

  @Test
  @ForPlatform(Platform.H2) // only run this on H2 - PGCrypto not happy on CI server
  public void testQueryJoin() {

    LoggedSqlCollector.start();

    DB.find(EBasicEncryptRelate.class)
      .where().eq("other.description", "foo")
      .findList();

    DB.find(EBasicEncryptRelate.class)
      .where().eq("other.description", "foo")
      .findCount();

    List<String> loggedSql = LoggedSqlCollector.stop();
    assertThat(loggedSql).hasSize(2);
    assertThat(loggedSql.get(0)).contains("left join e_basicenc t1 on t1.id = t0.other_id");
    assertThat(loggedSql.get(1)).contains("left join e_basicenc t1 on t1.id = t0.other_id");
  }

  @Test
  public void asDto() {
    DB.find(EBasicEncrypt.class).delete();

    EBasicEncrypt e = new EBasicEncrypt();
    e.setName("name2");
    e.setDescription("descEncrypted2");
    e.setDob(new Date(System.currentTimeMillis() - 100000));

    DB.save(e);

    final EDto dto = DB.find(EBasicEncrypt.class)
      .select("name, description")
      .asDto(EDto.class)
      .findOne();

    assertThat(dto).isNotNull();
    assertThat(dto.getName()).isEqualTo("name2");
    assertThat(dto.getDescription()).isEqualTo("descEncrypted2");
  }

  public static class EDto {
    String name;
    String description;
    public EDto(){
    }
    public String getName() {
      return name;
    }

    public void setName(String name) {
      this.name = name;
    }

    public String getDescription() {
      return description;
    }

    public void setDescription(String description) {
      this.description = description;
    }
  }

  @Test
  @ForPlatform(Platform.H2)
  public void test() {

    DB.find(EBasicEncrypt.class).delete();

    EBasicEncrypt e = new EBasicEncrypt();
    e.setName("testname");
    e.setDescription("testdesc");
    e.setDob(new Date(System.currentTimeMillis() - 100000));

    DB.save(e);

    SqlRow row = DB.sqlQuery("select * from e_basicenc where id = :id")
      .setParameter("id", e.getId())
      .findOne();

    row.getString("name");
    row.get("description");

    EBasicEncrypt e1 = DB.find(EBasicEncrypt.class, e.getId());

    e1.getDescription();

    e1.setName("testmod");
    e1.setDescription("moddesc");
    e1.setStatus(EBasicEncrypt.Status.ONE);

    DB.save(e1);

    EBasicEncrypt e2 = DB.find(EBasicEncrypt.class, e.getId());

    assertEquals("moddesc", e2.getDescription());
    assertEquals(EBasicEncrypt.Status.ONE, e2.getStatus());

    SpiEbeanServer server = (SpiEbeanServer) DB.getDefault();
    DbEncrypt dbEncrypt = server.getDatabasePlatform().getDbEncrypt();

    if (dbEncrypt == null) {
      // can not test the where clause
      System.out.println("TestEncrypt: Not testing where clause as no DbEncrypt");

    } else {

      List<EBasicEncrypt> list = DB.find(EBasicEncrypt.class).where()
        .eq("description", "moddesc").findList();

      assertEquals(1, list.size());

      list = DB.find(EBasicEncrypt.class).where().startsWith("description", "modde").findList();

      assertEquals(1, list.size());
    }
  }

}