refactor: Add query timeout test code

exposed-tests/src/test/kotlin/org/jetbrains/exposed/sql/tests/mysql/MysqlTests.kt

@@ -50,16 +50,4 @@ class MysqlTests : DatabaseTestsBase() {
             }
         }
     }
-
-    @Test
-    fun timeoutStatements() {
-        withDb(listOf(TestDB.MYSQL)) {
-            this.timeout = 3
-            assertFailsWith<ExposedSQLException> {
-                TransactionManager.current().exec(
-                    "SELECT SLEEP(5) = 0;"
-                )
-            }
-        }
-    }
 }

exposed-tests/src/test/kotlin/org/jetbrains/exposed/sql/tests/postgresql/PostgresqlTests.kt

@@ -119,18 +119,6 @@ class PostgresqlTests : DatabaseTestsBase() {
         }
     }
 
-    @Test
-    fun timeoutStatements() {
-        withDb(listOf(TestDB.POSTGRESQL)) {
-            this.timeout = 3
-            assertFailsWith<ExposedSQLException> {
-                TransactionManager.current().exec(
-                    "SELECT 1 WHERE pg_sleep(5);"
-                )
-            }
-        }
-    }
-
     private fun Transaction.withTable(statement: Transaction.() -> Unit) {
         SchemaUtils.create(table)
         try {

exposed-tests/src/test/kotlin/org/jetbrains/exposed/sql/tests/shared/QueryTimeoutTest.kt

@@ -0,0 +1,44 @@
+package org.jetbrains.exposed.sql.tests.shared
+
+import org.jetbrains.exposed.exceptions.ExposedSQLException
+import org.jetbrains.exposed.sql.tests.DatabaseTestsBase
+import org.jetbrains.exposed.sql.tests.TestDB
+import org.jetbrains.exposed.sql.transactions.TransactionManager
+import org.junit.Test
+import kotlin.test.assertFailsWith
+
+/**
+ * @author [email protected]
+ */
+class QueryTimeoutTest : DatabaseTestsBase() {
+
+    private fun generateTimeoutStatements(db: TestDB, timeout: Int): String {
+        return when (db) {
+            TestDB.MYSQL -> "SELECT SLEEP($timeout) = 0;"
+            TestDB.POSTGRESQL -> "SELECT pg_sleep($timeout);"
+            else -> throw NotImplementedError()
+        }
+    }
+
+    @Test
+    fun timeoutStatements() {
+        withDb(listOf(TestDB.MYSQL, TestDB.POSTGRESQL)) {
+            this.timeout = 3
+            assertFailsWith<ExposedSQLException> {
+                TransactionManager.current().exec(
+                    generateTimeoutStatements(it, 5)
+                )
+            }
+        }
+    }
+
+    @Test
+    fun noTimeoutWithTimeoutStatement() {
+        withDb(listOf(TestDB.MYSQL, TestDB.POSTGRESQL)) {
+            this.timeout = 3
+            TransactionManager.current().exec(
+                generateTimeoutStatements(it, 1)
+            )
+        }
+    }
+}