Consistently use runBlockingTest

Author: michaelbull

src/test/kotlin/com/github/michaelbull/jdbc/context/CoroutineConnectionTest.kt

@@ -2,7 +2,7 @@ package com.github.michaelbull.jdbc.context
 
 import io.mockk.mockk
 import kotlinx.coroutines.ExperimentalCoroutinesApi
-import kotlinx.coroutines.runBlocking
+import kotlinx.coroutines.test.runBlockingTest
 import org.junit.jupiter.api.Assertions.assertEquals
 import org.junit.jupiter.api.Test
 import org.junit.jupiter.api.assertThrows
@@ -14,15 +14,17 @@ class CoroutineConnectionTest {
     @Test
     fun `connection throws IllegalStateException if not in context`() {
         assertThrows<IllegalStateException> {
-            runBlocking { coroutineContext.connection }
+            runBlockingTest {
+                coroutineContext.connection
+            }
         }
     }
 
     @Test
     fun `connection returns connection if in context`() {
         val expected = mockk<Connection>()
 
-        runBlocking(CoroutineConnection(expected)) {
+        runBlockingTest(CoroutineConnection(expected)) {
             val actual = coroutineContext.connection
             assertEquals(expected, actual)
         }

src/test/kotlin/com/github/michaelbull/jdbc/context/CoroutineDataSourceTest.kt

@@ -2,7 +2,7 @@ package com.github.michaelbull.jdbc.context
 
 import io.mockk.mockk
 import kotlinx.coroutines.ExperimentalCoroutinesApi
-import kotlinx.coroutines.runBlocking
+import kotlinx.coroutines.test.runBlockingTest
 import org.junit.jupiter.api.Assertions
 import org.junit.jupiter.api.Test
 import org.junit.jupiter.api.assertThrows
@@ -14,15 +14,17 @@ class CoroutineDataSourceTest {
     @Test
     fun `dataSource throws IllegalStateException if not in context`() {
         assertThrows<IllegalStateException> {
-            runBlocking { coroutineContext.dataSource }
+            runBlockingTest {
+                coroutineContext.dataSource
+            }
         }
     }
 
     @Test
     fun `dataSource returns connection if in context`() {
         val expected = mockk<DataSource>()
 
-        runBlocking(CoroutineDataSource(expected)) {
+        runBlockingTest(CoroutineDataSource(expected)) {
             val actual = coroutineContext.dataSource
             Assertions.assertEquals(expected, actual)
         }