@@ -0,0 +1,111 @@
package ltd.evilcorp.core.db
import androidx.room.Room
import androidx.test.ext.junit.runners.AndroidJUnit4
import androidx.test.platform.app.InstrumentationRegistry
import kotlinx.coroutines.ExperimentalCoroutinesApi
import kotlinx.coroutines.asExecutor
import kotlinx.coroutines.flow.first
import kotlinx.coroutines.test.TestCoroutineDispatcher
import kotlinx.coroutines.test.TestCoroutineScope
import kotlinx.coroutines.test.runBlockingTest
import ltd.evilcorp.core.vo.ConnectionStatus
import ltd.evilcorp.core.vo.User
import ltd.evilcorp.core.vo.UserStatus
import org.junit.Assert.assertEquals
import org.junit.Assert.assertFalse
import org.junit.Assert.assertNotEquals
import org.junit.Assert.assertTrue
import org.junit.Assert.fail
import org.junit.Before
import org.junit.Test
import org.junit.runner.RunWith
@ExperimentalCoroutinesApi
@RunWith(AndroidJUnit4::class)
class UserDaoTest {
private val dispatcher = TestCoroutineDispatcher()
private val scope = TestCoroutineScope(dispatcher)
private val db =
Room.inMemoryDatabaseBuilder(InstrumentationRegistry.getInstrumentation().targetContext, Database::class.java)
.setTransactionExecutor(dispatcher.asExecutor())
.setQueryExecutor(dispatcher.asExecutor())
.allowMainThreadQueries()
.build()
private val dao = db.userDao()
private val first = User(
publicKey = "1234",
name = "name",
statusMessage = "status",
status = UserStatus.Away,
connectionStatus = ConnectionStatus.UDP,
password = "password",
)
@Before
fun clearDb() = db.clearAllTables()
@Test
fun save_and_load() = scope.runBlockingTest {
dao.save(first)
assertEquals(first, dao.load(first.publicKey).first())
}
@Test
fun update() = scope.runBlockingTest {
dao.save(first)
dao.update(first.copy(name = "new name"))
assertNotEquals(first, dao.load(first.publicKey).first())
assertEquals(first.copy(name = "new name"), dao.load(first.publicKey).first())
}
@Test
fun exists() = scope.runBlockingTest {
assertFalse(dao.exists(first.publicKey))
dao.save(first)
assertTrue(dao.exists(first.publicKey))
}
@Test
fun cant_replace_user_with_save() = scope.runBlockingTest {
dao.save(first)
try {
dao.save(first.copy(name = "new name"))
fail()
} catch (e: Exception) {
}
}
@Test
fun update_name() = scope.runBlockingTest {
dao.save(first)
dao.updateName(first.publicKey, "new name")
assertNotEquals(first, dao.load(first.publicKey).first())
assertEquals(first.copy(name = "new name"), dao.load(first.publicKey).first())
}
@Test
fun update_status_message() = scope.runBlockingTest {
dao.save(first)
dao.updateStatusMessage(first.publicKey, "new status")
assertNotEquals(first, dao.load(first.publicKey).first())
assertEquals(first.copy(statusMessage = "new status"), dao.load(first.publicKey).first())
}
@Test
fun update_connection() = scope.runBlockingTest {
dao.save(first)
dao.updateConnection(first.publicKey, ConnectionStatus.TCP)
assertNotEquals(first, dao.load(first.publicKey).first())
assertEquals(first.copy(connectionStatus = ConnectionStatus.TCP), dao.load(first.publicKey).first())
}
@Test
fun update_status() = scope.runBlockingTest {
dao.save(first)
dao.updateStatus(first.publicKey, UserStatus.Busy)
assertNotEquals(first, dao.load(first.publicKey).first())
assertEquals(first.copy(status = UserStatus.Busy), dao.load(first.publicKey).first())
}
}