Complete query migration

src/test/kotlin/ParametersTest.kt

@@ -2,6 +2,7 @@ package solutions.bitbadger.documents
 
 import org.junit.jupiter.api.AfterEach
 import org.junit.jupiter.api.DisplayName
+import org.junit.jupiter.api.assertThrows
 import kotlin.test.Test
 import kotlin.test.assertEquals
 import kotlin.test.assertNotSame
@@ -14,7 +15,7 @@ import kotlin.test.assertSame
 class ParametersTest {
 
     /**
-     * Clear the connection string (resets Dialect)
+     * Reset the dialect
      */
     @AfterEach
     fun cleanUp() {
@@ -65,7 +66,7 @@ class ParametersTest {
         assertEquals(1, nameParams.size, "There should be one name parameter")
         assertEquals(":name", nameParams[0].name, "The parameter name is incorrect")
         assertEquals(ParameterType.STRING, nameParams[0].type, "The parameter type is incorrect")
-        assertEquals("test", nameParams[0].value, "The parameter value is incorrect")
+        assertEquals("{test}", nameParams[0].value, "The parameter value is incorrect")
     }
 
     @Test
@@ -106,4 +107,10 @@ class ParametersTest {
         assertEquals(ParameterType.STRING, nameParams[2].type, "The third parameter type is incorrect")
         assertEquals("today", nameParams[2].value, "The third parameter value is incorrect")
     }
+
+    @Test
+    @DisplayName("fieldNames fails if dialect not set")
+    fun fieldNamesFails() {
+        assertThrows<DocumentException> { Parameters.fieldNames(listOf()) }
+    }
 }

src/test/kotlin/query/PatchTest.kt

@@ -0,0 +1,105 @@
+package solutions.bitbadger.documents.query
+
+import org.junit.jupiter.api.AfterEach
+import org.junit.jupiter.api.DisplayName
+import org.junit.jupiter.api.Test
+import org.junit.jupiter.api.assertThrows
+import solutions.bitbadger.documents.Configuration
+import solutions.bitbadger.documents.Dialect
+import solutions.bitbadger.documents.DocumentException
+import solutions.bitbadger.documents.Field
+import kotlin.test.assertEquals
+
+/**
+ * Unit tests for the `Patch` object
+ */
+@DisplayName("Patch (Query)")
+class PatchTest {
+
+    /** Test table name */
+    private val tbl = "test_table"
+
+    /**
+     * Reset the dialect
+     */
+    @AfterEach
+    fun cleanUp() {
+        Configuration.dialectValue = null
+    }
+
+    @Test
+    @DisplayName("byId generates correctly (PostgreSQL)")
+    fun byIdPostgres() {
+        Configuration.dialectValue = Dialect.POSTGRESQL
+        assertEquals(
+            "UPDATE $tbl SET data = data || :data WHERE data->>'id' = :id",
+            Patch.byId<String>(tbl), "Patch query not constructed correctly"
+        )
+    }
+
+    @Test
+    @DisplayName("byId generates correctly (SQLite)")
+    fun byIdSQLite() {
+        Configuration.dialectValue = Dialect.SQLITE
+        assertEquals(
+            "UPDATE $tbl SET data = json_patch(data, json(:data)) WHERE data->>'id' = :id",
+            Patch.byId<String>(tbl), "Patch query not constructed correctly"
+        )
+    }
+
+    @Test
+    @DisplayName("byFields generates correctly (PostgreSQL)")
+    fun byFieldsPostgres() {
+        Configuration.dialectValue = Dialect.POSTGRESQL
+        assertEquals(
+            "UPDATE $tbl SET data = data || :data WHERE data->>'z' = :y",
+            Patch.byFields(tbl, listOf(Field.equal("z", "", ":y"))),
+            "Patch query not constructed correctly"
+        )
+    }
+
+    @Test
+    @DisplayName("byFields generates correctly (SQLite)")
+    fun byFieldsSQLite() {
+        Configuration.dialectValue = Dialect.SQLITE
+        assertEquals(
+            "UPDATE $tbl SET data = json_patch(data, json(:data)) WHERE data->>'z' = :y",
+            Patch.byFields(tbl, listOf(Field.equal("z", "", ":y"))),
+            "Patch query not constructed correctly"
+        )
+    }
+
+    @Test
+    @DisplayName("byContains generates correctly (PostgreSQL)")
+    fun byContainsPostgres() {
+        Configuration.dialectValue = Dialect.POSTGRESQL
+        assertEquals(
+            "UPDATE $tbl SET data = data || :data WHERE data @> :criteria", Patch.byContains(tbl),
+            "Patch query not constructed correctly"
+        )
+    }
+
+    @Test
+    @DisplayName("byContains fails (SQLite)")
+    fun byContainsSQLite() {
+        Configuration.dialectValue = Dialect.SQLITE
+        assertThrows<DocumentException> { Patch.byContains(tbl) }
+    }
+
+    @Test
+    @DisplayName("byJsonPath generates correctly (PostgreSQL)")
+    fun byJsonPathPostgres() {
+        Configuration.dialectValue = Dialect.POSTGRESQL
+        assertEquals(
+            "UPDATE $tbl SET data = data || :data WHERE jsonb_path_exists(data, :path::jsonpath)",
+            Patch.byJsonPath(tbl), "Patch query not constructed correctly"
+        )
+    }
+
+    @Test
+    @DisplayName("byJsonPath fails (SQLite)")
+    fun byJsonPathSQLite() {
+        Configuration.dialectValue = Dialect.SQLITE
+        assertThrows<DocumentException> { Patch.byJsonPath(tbl) }
+    }
+}

src/test/kotlin/query/RemoveFieldsTest.kt

@@ -0,0 +1,106 @@
+package solutions.bitbadger.documents.query
+
+import org.junit.jupiter.api.AfterEach
+import org.junit.jupiter.api.DisplayName
+import org.junit.jupiter.api.Test
+import org.junit.jupiter.api.assertThrows
+import solutions.bitbadger.documents.*
+import kotlin.test.assertEquals
+
+/**
+ * Unit tests for the `RemoveFields` object
+ */
+@DisplayName("RemoveFields (Query)")
+class RemoveFieldsTest {
+
+    /** Test table name */
+    private val tbl = "test_table"
+
+    /**
+     * Reset the dialect
+     */
+    @AfterEach
+    fun cleanUp() {
+        Configuration.dialectValue = null
+    }
+
+    @Test
+    @DisplayName("byId generates correctly (PostgreSQL)")
+    fun byIdPostgres() {
+        Configuration.dialectValue = Dialect.POSTGRESQL
+        assertEquals(
+            "UPDATE $tbl SET data = data - :name::text[] WHERE data->>'id' = :id",
+            RemoveFields.byId<String>(tbl, Parameters.fieldNames(listOf("a", "z"))),
+            "Remove Fields query not constructed correctly"
+        )
+    }
+
+    @Test
+    @DisplayName("byId generates correctly (SQLite)")
+    fun byIdSQLite() {
+        Configuration.dialectValue = Dialect.SQLITE
+        assertEquals(
+            "UPDATE $tbl SET data = json_remove(data, :name0, :name1) WHERE data->>'id' = :id",
+            RemoveFields.byId<String>(tbl, Parameters.fieldNames(listOf("a", "z"))),
+            "Remove Field query not constructed correctly"
+        )
+    }
+
+    @Test
+    @DisplayName("byFields generates correctly (PostgreSQL)")
+    fun byFieldsPostgres() {
+        Configuration.dialectValue = Dialect.POSTGRESQL
+        assertEquals(
+            "UPDATE $tbl SET data = data - :name::text[] WHERE data->>'f' > :g",
+            RemoveFields.byFields(tbl, Parameters.fieldNames(listOf("b", "c")), listOf(Field.greater("f", "", ":g"))),
+            "Remove Field query not constructed correctly"
+        )
+    }
+
+    @Test
+    @DisplayName("byFields generates correctly (SQLite)")
+    fun byFieldsSQLite() {
+        Configuration.dialectValue = Dialect.SQLITE
+        assertEquals(
+            "UPDATE $tbl SET data = json_remove(data, :name0, :name1) WHERE data->>'f' > :g",
+            RemoveFields.byFields(tbl, Parameters.fieldNames(listOf("b", "c")), listOf(Field.greater("f", "", ":g"))),
+            "Remove Field query not constructed correctly"
+        )
+    }
+
+    @Test
+    @DisplayName("byContains generates correctly (PostgreSQL)")
+    fun byContainsPostgres() {
+        Configuration.dialectValue = Dialect.POSTGRESQL
+        assertEquals(
+            "UPDATE $tbl SET data = data - :name::text[] WHERE data @> :criteria",
+            RemoveFields.byContains(tbl, Parameters.fieldNames(listOf("m", "n"))),
+            "Remove Field query not constructed correctly"
+        )
+    }
+
+    @Test
+    @DisplayName("byContains fails (SQLite)")
+    fun byContainsSQLite() {
+        Configuration.dialectValue = Dialect.SQLITE
+        assertThrows<DocumentException> { RemoveFields.byContains(tbl, listOf()) }
+    }
+
+    @Test
+    @DisplayName("byJsonPath generates correctly (PostgreSQL)")
+    fun byJsonPathPostgres() {
+        Configuration.dialectValue = Dialect.POSTGRESQL
+        assertEquals(
+            "UPDATE $tbl SET data = data - :name::text[] WHERE jsonb_path_exists(data, :path::jsonpath)",
+            RemoveFields.byJsonPath(tbl, Parameters.fieldNames(listOf("o", "p"))),
+            "Remove Field query not constructed correctly"
+        )
+    }
+
+    @Test
+    @DisplayName("byJsonPath fails (SQLite)")
+    fun byJsonPathSQLite() {
+        Configuration.dialectValue = Dialect.SQLITE
+        assertThrows<DocumentException> { RemoveFields.byJsonPath(tbl, listOf()) }
+    }
+}