Add MSSQL test suite

Author: kiendang

scalasql/src/dialects/MsSqlDialect.scala

@@ -5,6 +5,8 @@ import scalasql.operations
 import scalasql.core.SqlStr.SqlStringSyntax
 import scalasql.operations.{ConcatOps, MathOps, TrimOps}
 
+import java.time.{Instant, LocalDateTime, OffsetDateTime}
+
 trait MsSqlDialect extends Dialect {
   protected def dialectCastParams = false
 
@@ -14,6 +16,25 @@ trait MsSqlDialect extends Dialect {
   override implicit def StringType: TypeMapper[String] = new MsSqlStringType
   class MsSqlStringType extends StringType { override def castTypeString = "VARCHAR" }
 
+  override implicit def BooleanType: TypeMapper[Boolean] = new BooleanType
+  class MsSqlBooleanType extends BooleanType { override def castTypeString = "BIT" }
+
+  override implicit def UtilDateType: TypeMapper[java.util.Date] = new MsSqlUtilDateType
+  class MsSqlUtilDateType extends UtilDateType { override def castTypeString = "DATETIME2" }
+
+  override implicit def LocalDateTimeType: TypeMapper[LocalDateTime] = new MsSqlLocalDateTimeType
+  class MsSqlLocalDateTimeType extends LocalDateTimeType {
+    override def castTypeString = "DATETIME2"
+  }
+
+  override implicit def InstantType: TypeMapper[Instant] = new MsSqlInstantType
+  class MsSqlInstantType extends InstantType { override def castTypeString = "DATETIME2" }
+
+  override implicit def OffsetDateTimeType: TypeMapper[OffsetDateTime] = new MsSqlOffsetDateTimeType
+  class MsSqlOffsetDateTimeType extends OffsetDateTimeType {
+    override def castTypeString = "DATETIMEOFFSET"
+  }
+
   override implicit def ExprStringOpsConv(v: Expr[String]): MsSqlDialect.ExprStringOps[String] =
     new MsSqlDialect.ExprStringOps(v)
 

scalasql/test/resources/mssql-customer-schema.sql

@@ -0,0 +1,100 @@
+IF EXISTS (SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_NAME = N'shipping_info')
+    ALTER TABLE shipping_info DROP CONSTRAINT IF EXISTS fk_shipping_info_buyer;
+IF EXISTS (SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_NAME = N'purchase')
+    ALTER TABLE purchase DROP CONSTRAINT IF EXISTS fk_purchase_shipping_info;
+IF EXISTS (SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_NAME = N'purchase')
+    ALTER TABLE purchase DROP CONSTRAINT IF EXISTS fk_purchase_product;
+DROP TABLE IF EXISTS buyer;
+DROP TABLE IF EXISTS product;
+DROP TABLE IF EXISTS shipping_info;
+DROP TABLE IF EXISTS purchase;
+DROP TABLE IF EXISTS data_types;
+DROP TABLE IF EXISTS non_round_trip_types;
+DROP TABLE IF EXISTS opt_cols;
+DROP TABLE IF EXISTS nested;
+DROP TABLE IF EXISTS enclosing;
+DROP TABLE IF EXISTS invoice;
+-- DROP SCHEMA IF EXISTS otherschema;
+
+CREATE TABLE buyer (
+    id INT PRIMARY KEY IDENTITY(1, 1),
+    name VARCHAR(256),
+    date_of_birth DATE
+);
+
+CREATE TABLE product (
+    id INT PRIMARY KEY IDENTITY(1, 1),
+    kebab_case_name VARCHAR(256),
+    name VARCHAR(256),
+    price DECIMAL(20, 2)
+);
+
+CREATE TABLE shipping_info (
+    id INT PRIMARY KEY IDENTITY(1, 1),
+    buyer_id INT,
+    shipping_date DATE,
+    CONSTRAINT fk_shipping_info_buyer
+        FOREIGN KEY(buyer_id) REFERENCES buyer(id)
+);
+
+CREATE TABLE purchase (
+    id INT PRIMARY KEY IDENTITY(1, 1),
+    shipping_info_id INT,
+    product_id INT,
+    count INT,
+    total DECIMAL(20, 2),
+    CONSTRAINT fk_purchase_shipping_info
+        FOREIGN KEY(shipping_info_id) REFERENCES shipping_info(id),
+    CONSTRAINT fk_purchase_product
+        FOREIGN KEY(product_id) REFERENCES product(id)
+);
+
+CREATE TABLE data_types (
+    my_tiny_int TINYINT,
+    my_small_int SMALLINT,
+    my_int INT,
+    my_big_int BIGINT,
+    my_double FLOAT(53),
+    my_boolean BIT,
+    my_local_date DATE,
+    my_local_time TIME,
+    my_local_date_time DATETIME2,
+    my_util_date DATETIMEOFFSET,
+    my_instant DATETIMEOFFSET,
+    my_var_binary VARBINARY,
+    my_uuid UNIQUEIDENTIFIER,
+    my_enum VARCHAR(256)
+--     my_offset_time TIME WITH TIME ZONE,
+
+);
+
+CREATE TABLE non_round_trip_types(
+    my_zoned_date_time DATETIMEOFFSET,
+    my_offset_date_time DATETIMEOFFSET
+);
+
+CREATE TABLE opt_cols(
+    my_int INT,
+    my_int2 INT
+);
+
+CREATE TABLE nested(
+    foo_id INT,
+    my_boolean BIT
+);
+
+CREATE TABLE enclosing(
+    bar_id INT,
+    my_string VARCHAR(256),
+    foo_id INT,
+    my_boolean BIT
+);
+
+
+-- CREATE SCHEMA otherschema;
+
+-- CREATE TABLE otherschema.invoice(
+--     id PRIMARY KEY IDENTITY(1, 1),
+--     total DECIMAL(20, 2),
+--     vendor_name VARCHAR(256)
+-- );

scalasql/test/src/ConcreteTestSuites.scala

@@ -35,7 +35,8 @@ import scalasql.dialects.{
   MySqlDialectTests,
   PostgresDialectTests,
   SqliteDialectTests,
-  H2DialectTests
+  H2DialectTests,
+  MsSqlDialectTests
 }
 
 package postgres {
@@ -267,3 +268,48 @@ package h2 {
 
   object H2DialectTests extends H2DialectTests
 }
+
+package mssql {
+
+  import utils.MsSqlSuite
+
+  object DbApiTests extends DbApiTests with MsSqlSuite
+  object TransactionTests extends TransactionTests with MsSqlSuite
+
+  object SelectTests extends SelectTests with MsSqlSuite
+  object JoinTests extends JoinTests with MsSqlSuite
+  object FlatJoinTests extends FlatJoinTests with MsSqlSuite
+  object InsertTests extends InsertTests with MsSqlSuite
+  object UpdateTests extends UpdateTests with MsSqlSuite
+  object DeleteTests extends DeleteTests with MsSqlSuite
+  object CompoundSelectTests extends CompoundSelectTests with MsSqlSuite
+  object UpdateJoinTests extends UpdateJoinTests with MsSqlSuite
+  object UpdateSubQueryTests extends UpdateSubQueryTests with MsSqlSuite
+  // object ReturningTests extends ReturningTests with MsSqlSuite
+  // object OnConflictTests extends OnConflictTests with MsSqlSuite
+  object ValuesTests extends ValuesTests with MsSqlSuite
+  // object LateralJoinTests extends LateralJoinTests with MsSqlSuite
+  object WindowFunctionTests extends WindowFunctionTests with MsSqlSuite
+  object GetGeneratedKeysTests extends GetGeneratedKeysTests with MsSqlSuite
+  object SchemaTests extends SchemaTests with MsSqlSuite
+
+  object SubQueryTests extends SubQueryTests with MsSqlSuite
+  object WithCteTests extends WithCteTests with MsSqlSuite
+
+  object DbApiOpsTests extends DbApiOpsTests with MsSqlSuite
+  object ExprOpsTests extends ExprOpsTests with MsSqlSuite
+  object ExprBooleanOpsTests extends ExprBooleanOpsTests with MsSqlSuite
+  object ExprNumericOpsTests extends ExprNumericOpsTests with MsSqlSuite
+  object ExprSeqNumericOpsTests extends ExprAggNumericOpsTests with MsSqlSuite
+  object ExprSeqOpsTests extends ExprAggOpsTests with MsSqlSuite
+  object ExprStringOpsTests extends ExprStringOpsTests with MsSqlSuite
+  object ExprBlobOpsTests extends ExprBlobOpsTests with MsSqlSuite
+  object ExprMathOpsTests extends ExprMathOpsTests with MsSqlSuite
+
+  object DataTypesTests extends datatypes.DataTypesTests with MsSqlSuite
+
+  object OptionalTests extends datatypes.OptionalTests with MsSqlSuite
+
+  object MsSqlDialectTests extends MsSqlDialectTests
+
+}

scalasql/test/src/dialects/MsSqlDialectTests.scala

@@ -0,0 +1,11 @@
+package scalasql.dialects
+
+import scalasql._
+import utest._
+import utils.MsSqlSuite
+
+trait MsSqlDialectTests extends MsSqlSuite {
+  def description = "Operations specific to working with Microsoft SQL Databases"
+
+  def tests = Tests {}
+}

scalasql/test/src/utils/ScalaSqlSuite.scala

@@ -83,3 +83,16 @@ trait MySqlSuite extends ScalaSqlSuite with MySqlDialect {
 
   checker.reset()
 }
+
+trait MsSqlSuite extends ScalaSqlSuite with MsSqlDialect {
+  val checker = new TestChecker(
+    scalasql.example.MsSqlExample.mssqlClient,
+    "mssql-customer-schema.sql",
+    "customer-data.sql",
+    getClass.getName,
+    suiteLine.value,
+    description
+  )
+
+  checker.reset()
+}