#940 - Allow findNative to use @Formula properties (add test for MySql := use with RawSql and findNative)

src/test/java/org/tests/rawsql/TestRawSqlNamedParams.java

@@ -1,17 +1,22 @@
 package org.tests.rawsql;
 
 import io.ebean.BaseTestCase;
+import io.ebean.CallableSql;
 import io.ebean.Ebean;
 import io.ebean.Query;
 import io.ebean.RawSql;
 import io.ebean.RawSqlBuilder;
+import io.ebean.Transaction;
 import org.tests.model.basic.Customer;
+import org.tests.model.basic.Order;
 import org.tests.model.basic.ResetBasicData;
 import org.junit.Assert;
 import org.junit.Test;
 
 import java.util.List;
 
+import static org.assertj.core.api.Assertions.assertThat;
+
 public class TestRawSqlNamedParams extends BaseTestCase {
 
   @Test
@@ -33,4 +38,43 @@ public class TestRawSqlNamedParams extends BaseTestCase {
 
     Assert.assertNotNull(list);
   }
+
+  @Test
+  public void testMySqlColonEquals() {
+
+    if (!isMySql()) {
+      return;
+    }
+
+    ResetBasicData.reset();
+
+    Transaction transaction = Ebean.beginTransaction();
+
+    try {
+      CallableSql callableSql = Ebean.createCallableSql("set @total = 0");
+      Ebean.execute(callableSql);
+
+      String sql = "select id, @total := 0 + id as total_items from o_order";
+
+      Query<Order> query1 = Ebean.findNative(Order.class, sql);
+      List<Order> list = query1.findList();
+
+      assertThat(list.get(0).getTotalItems()).isNotNull();
+
+      Ebean.execute(callableSql);
+
+      RawSql rawSql = RawSqlBuilder
+        .parse(sql)
+        .create();
+
+      Query<Order> query = Ebean.find(Order.class)
+        .setRawSql(rawSql);
+
+      List<Order> list1 = query.findList();
+      assertThat(list1.get(0).getTotalItems()).isNotNull();
+
+    } finally {
+      transaction.end();
+    }
+  }
 }