63 lines
2.0 KiB
Kotlin
63 lines
2.0 KiB
Kotlin
package moe.kageru.kagebot.features
|
|
|
|
import io.kotlintest.matchers.string.shouldContain
|
|
import io.kotlintest.shouldBe
|
|
import io.kotlintest.specs.StringSpec
|
|
import moe.kageru.kagebot.Kagebot.process
|
|
import moe.kageru.kagebot.TestUtil
|
|
import moe.kageru.kagebot.TestUtil.TEST_ROLE
|
|
import moe.kageru.kagebot.persistence.Dao
|
|
import org.javacord.api.entity.message.embed.EmbedBuilder
|
|
|
|
class TimeoutFeatureTest : StringSpec({
|
|
TestUtil.prepareTestEnvironment()
|
|
"should remove and store roles" {
|
|
clearTimeouts()
|
|
TestUtil.mockMessage("!timeout kageru 99999999").process()
|
|
Dao.getAllTimeouts().let {
|
|
it.size shouldBe 1
|
|
val user = Dao.deleteTimeout(it.first())
|
|
user shouldBe arrayOf(123, TEST_ROLE.id)
|
|
}
|
|
clearTimeouts()
|
|
}
|
|
"should announce timeout via DM" {
|
|
val dms = mutableListOf<EmbedBuilder>()
|
|
TestUtil.prepareTestEnvironment(dmEmbeds = dms)
|
|
val time = "1235436"
|
|
TestUtil.mockMessage("!timeout kageru $time").process()
|
|
dms.size shouldBe 1
|
|
TestUtil.embedToString(dms[0]) shouldContain time
|
|
clearTimeouts()
|
|
}
|
|
"should return error for invalid input" {
|
|
val replies = mutableListOf<String>()
|
|
TestUtil.mockMessage("!timeout kageruWithoutATime", replies = replies).process()
|
|
replies.size shouldBe 1
|
|
replies[0] shouldContain "Error"
|
|
}
|
|
"should catch malformed time" {
|
|
val replies = mutableListOf<String>()
|
|
TestUtil.mockMessage("!timeout kageru this is not a time", replies = replies).process()
|
|
replies.size shouldBe 1
|
|
replies[0] shouldContain "Error"
|
|
}
|
|
"should print optional reason" {
|
|
val dms = mutableListOf<EmbedBuilder>()
|
|
TestUtil.prepareTestEnvironment(dmEmbeds = dms)
|
|
val reason = "because I don’t like you"
|
|
TestUtil.mockMessage("!timeout kageru 1 $reason").process()
|
|
dms.size shouldBe 1
|
|
TestUtil.embedToString(dms[0]) shouldContain reason
|
|
clearTimeouts()
|
|
}
|
|
}) {
|
|
companion object {
|
|
private fun clearTimeouts() {
|
|
Dao.getAllTimeouts().forEach { to ->
|
|
Dao.deleteTimeout(to)
|
|
}
|
|
}
|
|
}
|
|
}
|