[#23] Add tests

Orchaldir · Apr 22, 2020 · 6411a95 · 6411a95
1 parent a3e3227
commit 6411a95
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 4 deletions.
diff --git a/src/test/kotlin/util/ecs/storage/ComponentMapTest.kt b/src/test/kotlin/util/ecs/storage/ComponentMapTest.kt
@@ -31,6 +31,10 @@ private val storage = ComponentMap(TYPE, components)
 
 class ComponentMapTest {
 
+    @Test
+    fun `Get correct type`() {
+        assertThat(storage.getType()).isSameAs(TYPE)
+    }
 
     @Nested
     inner class Has {

diff --git a/src/test/kotlin/util/log/MessageLogTest.kt b/src/test/kotlin/util/log/MessageLogTest.kt
@@ -9,26 +9,35 @@ import util.ecs.EcsState
 
 class MessageLogTest {
 
+    private val message0 = mockk<Message>()
+    private val message1 = mockk<Message>()
+    private val message2 = mockk<Message>()
+
     @Test
     fun `Test empty constructor`() {
         assertThat(MessageLog()).isEqualTo(MessageLog(emptyList()))
     }
 
     @Test
     fun `Add a message to the message log`() {
-        val message0 = mockk<Message>()
-        val message1 = mockk<Message>()
         val log = MessageLog(listOf(message0))
 
         val newLog = log.add(message1)
 
         assertThat(newLog).isEqualTo(MessageLog(listOf(message0, message1)))
     }
 
+    @Test
+    fun `Add messages to the message log`() {
+        val log = MessageLog(listOf(message0))
+
+        val newLog = log.add(listOf(message1, message2))
+
+        assertThat(newLog).isEqualTo(MessageLog(listOf(message0, message1, message2)))
+    }
+
     @Test
     fun `Add a message to the state`() {
-        val message0 = mockk<Message>()
-        val message1 = mockk<Message>()
         val log = MessageLog(listOf(message0))
         val state = EcsState(dataMap = mapOf(MessageLog::class to log))