// EXEC: cjc %import-path %L %l %f
// EXEC: ./main
import mysqlclient_ffi.*
import std.unittest.*
import std.unittest.testmacro.*
import std.database.sql.*
import std.time.*
import std.io.*

main(): Int64 {
    let mysqlCharSetApiTest: MysqlCharSetApiTest = MysqlCharSetApiTest()

    mysqlCharSetApiTest.mysqlCharSetApiTest001()
    mysqlCharSetApiTest.mysqlCharSetApiTest002()

    return 0
}

@Test
public class MysqlCharSetApiTest {
    @TestCase
    func mysqlCharSetApiTest001(): Unit {
        // 初始化数据库驱动
        let mysqlDriver: MysqlDriver = MysqlDriver("mysql")

        // 通过connectionString和选项打开数据源
        let mysqlDatasource: MysqlDatasource = mysqlDriver.open(
            "HOST=127.0.0.1;USER=root;PASSWD=123;DB=mysql;PORT=3306;UNIX_SOCKET=;CLIENT_FLAG=0",
            Array<(String, String)>()
        )

        // 返回一个可用的链接
        let mysqlConnection: MysqlConnection = mysqlDatasource.connect()

        let mysqlCharSetApi: MysqlCharSetApi = MysqlCharSetApi(mysqlConnection)
        @Assert("utf8mb4", mysqlCharSetApi.mysqlCharacterSetName())

        var ret: Int32 = mysqlCharSetApi.mysqlSetCharacterSet("utf8")
        @Assert(true, ret == 0)
        @Assert("utf8mb3", mysqlCharSetApi.mysqlCharacterSetName())

        ret = mysqlCharSetApi.mysqlSetCharacterSet("ascii")
        @Assert(true, ret == 0)
        @Assert("ascii", mysqlCharSetApi.mysqlCharacterSetName())

        ret = mysqlCharSetApi.mysqlSetCharacterSet("gbk")
        @Assert(true, ret == 0)
        @Assert("gbk", mysqlCharSetApi.mysqlCharacterSetName())

        ret = mysqlCharSetApi.mysqlSetCharacterSet("utf8mb4")
        @Assert(true, ret == 0)
        @Assert("utf8mb4", mysqlCharSetApi.mysqlCharacterSetName())

        mysqlConnection.close()
    }

    @TestCase
    func mysqlCharSetApiTest002(): Unit {
        // 初始化数据库驱动
        let mysqlDriver: MysqlDriver = MysqlDriver("mysql")

        // 通过connectionString和选项打开数据源
        let mysqlDatasource: MysqlDatasource = mysqlDriver.open(
            "HOST=127.0.0.1;USER=root;PASSWD=123;DB=mysql;PORT=3306;UNIX_SOCKET=;CLIENT_FLAG=0",
            Array<(String, String)>()
        )

        // 返回一个可用的链接
        let mysqlConnection: MysqlConnection = mysqlDatasource.connect()

        let mysqlCharSetApi: MysqlCharSetApi = MysqlCharSetApi(mysqlConnection)

        var ret: Int32 = mysqlCharSetApi.mysqlSetCharacterSet("1")
        @Assert(false, ret == 0)
        @Assert("utf8mb4", mysqlCharSetApi.mysqlCharacterSetName())

        ret = mysqlCharSetApi.mysqlSetCharacterSet("____")
        @Assert(false, ret == 0)
        @Assert("utf8mb4", mysqlCharSetApi.mysqlCharacterSetName())

        ret = mysqlCharSetApi.mysqlSetCharacterSet("acfasfdsfds")
        @Assert(false, ret == 0)
        @Assert("utf8mb4", mysqlCharSetApi.mysqlCharacterSetName())

        ret = mysqlCharSetApi.mysqlSetCharacterSet("fsa4290312>>")
        @Assert(false, ret == 0)
        @Assert("utf8mb4", mysqlCharSetApi.mysqlCharacterSetName())

        ret = mysqlCharSetApi.mysqlSetCharacterSet("123")
        @Assert(false, ret == 0)
        @Assert("utf8mb4", mysqlCharSetApi.mysqlCharacterSetName())

        ret = mysqlCharSetApi.mysqlSetCharacterSet(">><<<^*#@~!()")
        @Assert(false, ret == 0)
        @Assert("utf8mb4", mysqlCharSetApi.mysqlCharacterSetName())

        ret = mysqlCharSetApi.mysqlSetCharacterSet("")
        @Assert(false, ret == 0)
        @Assert("utf8mb4", mysqlCharSetApi.mysqlCharacterSetName())

        mysqlConnection.close()
    }
}