package org.tests.rawsql;

import io.ebean.*;
import io.ebean.annotation.ForPlatform;
import io.ebean.annotation.Platform;
import org.junit.jupiter.api.Test;
import org.tests.model.basic.Customer;
import org.tests.model.basic.Order;
import org.tests.model.basic.ResetBasicData;

import java.sql.SQLException;
import java.util.List;

import static org.assertj.core.api.Assertions.assertThat;
import static org.junit.jupiter.api.Assertions.assertNotNull;

public class TestRawSqlNamedParams extends BaseTestCase {

  @Test
  public void test() {

    ResetBasicData.reset();

    RawSql rawSql = RawSqlBuilder
      .parse("select r.id, r.name from o_customer r where r.id > :id and r.name like :name")
      .columnMapping("r.id", "id").columnMapping("r.name", "name").create();

    Query<Customer> query = DB.find(Customer.class);
    query.setRawSql(rawSql);
    query.setParameter("name", "R%");
    query.setParameter("id", 0);
    query.where().lt("id", 1000);

    List<Customer> list = query.findList();
    assertNotNull(list);
  }

  @Test
  @ForPlatform(Platform.MYSQL)
  public void testMySqlColonEquals() throws SQLException {

    ResetBasicData.reset();

    Transaction transaction = DB.beginTransaction();

    System.out.println(transaction.connection().getMetaData().getDriverName());
    try {
      if ("MariaDB Connector/J".equals(transaction.connection().getMetaData().getDriverName())) {
        // CHECKME: Should we report/fail if we are running mysql-tests with mariadb driver
        // BTW: This happens with java 8 - the drivers in drivermanager are in different order 
        return; // MariaDB Connector/J only supports callable statements in the form "? = call function x(?)"
      }
      CallableSql callableSql = DB.createCallableSql("set @total = 0");
      DB.getDefault().execute(callableSql);

      String sql = "select id, @total := 0 + id as total_items from o_order";

      Query<Order> query1 = DB.findNative(Order.class, sql);
      List<Order> list = query1.findList();

      assertThat(list.get(0).getTotalItems()).isNotNull();

      DB.getDefault().execute(callableSql);

      RawSql rawSql = RawSqlBuilder
        .parse(sql)
        .create();

      Query<Order> query = DB.find(Order.class)
        .setRawSql(rawSql);

      List<Order> list1 = query.findList();
      assertThat(list1.get(0).getTotalItems()).isNotNull();

    } finally {
      transaction.end();
    }
  }
}