Add tests for all raw DSL methods

Author: jeffgbutler

src/main/kotlin/org/mybatis/dynamic/sql/util/kotlin/InsertDSLExtensions.kt

@@ -1,3 +1,18 @@
+/**
+ *    Copyright 2016-2019 the original author or authors.
+ *
+ *    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.
+ */
 package org.mybatis.dynamic.sql.util.kotlin
 
 import org.mybatis.dynamic.sql.insert.InsertDSL

src/test/kotlin/examples/kotlin/mybatis3/general/GeneralKotlinTest.kt

@@ -36,11 +36,11 @@ import org.assertj.core.api.Assertions.assertThat
 import org.junit.jupiter.api.Test
 import org.mybatis.dynamic.sql.SqlBuilder.*
 import org.mybatis.dynamic.sql.util.kotlin.*
-import org.mybatis.dynamic.sql.util.kotlin.mybatis3.deleteFrom
+import org.mybatis.dynamic.sql.util.kotlin.mybatis3.*
 import org.mybatis.dynamic.sql.util.kotlin.mybatis3.from
-import org.mybatis.dynamic.sql.util.kotlin.mybatis3.update
 import java.io.InputStreamReader
 import java.sql.DriverManager
+import java.util.*
 
 class GeneralKotlinTest {
     private fun newSession(): SqlSession {
@@ -61,6 +61,41 @@ class GeneralKotlinTest {
         return SqlSessionFactoryBuilder().build(config).openSession()
     }
 
+    @Test
+    fun testRawCount() {
+        newSession().use { session ->
+            val mapper = session.getMapper(PersonMapper::class.java)
+
+            val countStatement = count(Person) {
+                where(id, isLessThan(4))
+            }
+
+            assertThat(countStatement.selectStatement).isEqualTo("select count(*) from Person" +
+                " where id < #{parameters.p1,jdbcType=INTEGER}")
+
+            val rows = mapper.count(countStatement)
+
+            assertThat(rows).isEqualTo(3)
+        }
+    }
+
+    @Test
+    fun testRawCountAllRows() {
+        newSession().use { session ->
+            val mapper = session.getMapper(PersonMapper::class.java)
+
+            val countStatement = count(Person) {
+                allRows()
+            }
+
+            assertThat(countStatement.selectStatement).isEqualTo("select count(*) from Person")
+
+            val rows = mapper.count(countStatement)
+
+            assertThat(rows).isEqualTo(6)
+        }
+    }
+
     @Test
     fun testRawDelete1() {
         newSession().use { session ->
@@ -192,6 +227,80 @@ class GeneralKotlinTest {
         }
     }
 
+    @Test
+    fun testInsert() {
+        newSession().use { session ->
+            val mapper = session.getMapper(PersonMapper::class.java)
+
+            val record = PersonRecord(100, "Joe", LastName("Jones"), Date(), true, "Developer", 1)
+
+            val insertStatement = insert(record, Person) {
+                map(id).toProperty("id")
+                map(firstName).toProperty("firstName")
+                map(lastName).toProperty("lastName")
+                map(birthDate).toProperty("birthDate")
+                map(employed).toProperty("employed")
+                map(occupation).toProperty("occupation")
+                map(addressId).toProperty("addressId")
+            }
+
+            val expected = "insert into Person (id, first_name, last_name, birth_date, employed, occupation, address_id)" +
+                " values" +
+                " (#{record.id,jdbcType=INTEGER}, #{record.firstName,jdbcType=VARCHAR}," +
+                " #{record.lastName,jdbcType=VARCHAR,typeHandler=examples.kotlin.mybatis3.canonical.LastNameTypeHandler}," +
+                " #{record.birthDate,jdbcType=DATE}, #{record.employed,jdbcType=VARCHAR,typeHandler=examples.kotlin.mybatis3.canonical.YesNoTypeHandler}," +
+                " #{record.occupation,jdbcType=VARCHAR}, #{record.addressId,jdbcType=INTEGER})"
+
+            assertThat(insertStatement.insertStatement).isEqualTo(expected)
+
+            val rows = mapper.insert(insertStatement)
+            assertThat(rows).isEqualTo(1)
+        }
+    }
+
+    @Test
+    fun testInsertMultiple() {
+        newSession().use { session ->
+            val mapper = session.getMapper(PersonMapper::class.java)
+
+            val record1 = PersonRecord(100, "Joe", LastName("Jones"), Date(), true, "Developer", 1)
+            val record2 = PersonRecord(101, "Sarah", LastName("Smith"), Date(), true, "Architect", 2)
+
+            val insertStatement = insertMultiple(listOf(record1, record2), Person) {
+                map(id).toProperty("id")
+                map(firstName).toProperty("firstName")
+                map(lastName).toProperty("lastName")
+                map(birthDate).toProperty("birthDate")
+                map(employed).toProperty("employed")
+                map(occupation).toProperty("occupation")
+                map(addressId).toProperty("addressId")
+            }
+
+            val expected = "insert into Person (id, first_name, last_name, birth_date, employed, occupation, address_id)" +
+                " values" +
+                " (#{records[0].id,jdbcType=INTEGER}," +
+                " #{records[0].firstName,jdbcType=VARCHAR}," +
+                " #{records[0].lastName,jdbcType=VARCHAR,typeHandler=examples.kotlin.mybatis3.canonical.LastNameTypeHandler}," +
+                " #{records[0].birthDate,jdbcType=DATE}," +
+                " #{records[0].employed,jdbcType=VARCHAR,typeHandler=examples.kotlin.mybatis3.canonical.YesNoTypeHandler}," +
+                " #{records[0].occupation,jdbcType=VARCHAR}," +
+                " #{records[0].addressId,jdbcType=INTEGER})" +
+                ", (#{records[1].id,jdbcType=INTEGER}," +
+                " #{records[1].firstName,jdbcType=VARCHAR}," +
+                " #{records[1].lastName,jdbcType=VARCHAR,typeHandler=examples.kotlin.mybatis3.canonical.LastNameTypeHandler}," +
+                " #{records[1].birthDate,jdbcType=DATE}," +
+                " #{records[1].employed,jdbcType=VARCHAR,typeHandler=examples.kotlin.mybatis3.canonical.YesNoTypeHandler}," +
+                " #{records[1].occupation,jdbcType=VARCHAR}," +
+                " #{records[1].addressId,jdbcType=INTEGER})"
+
+            assertThat(insertStatement.insertStatement).isEqualTo(expected)
+
+            val rows = mapper.insertMultiple(insertStatement)
+
+            assertThat(rows).isEqualTo(2)
+        }
+    }
+
     @Test
     fun testRawSelect() {
         newSession().use { session ->