Move createSone() method to test base class

[Sone.git] / src / test / kotlin / net / pterodactylus / sone / fcp / SoneCommandTest.kt

diff --git a/src/test/kotlin/net/pterodactylus/sone/fcp/SoneCommandTest.kt b/src/test/kotlin/net/pterodactylus/sone/fcp/SoneCommandTest.kt
index f34b19c..8344531 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/fcp/SoneCommandTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/fcp/SoneCommandTest.kt
@@ -4,6 +4,7 @@ import com.google.common.base.Optional
 import com.google.common.base.Optional.absent
 import freenet.support.SimpleFieldSet
 import net.pterodactylus.sone.core.Core
+import net.pterodactylus.sone.data.Profile
 import net.pterodactylus.sone.data.Sone
 import net.pterodactylus.sone.freenet.fcp.FcpException
 import net.pterodactylus.sone.test.mock
@@ -38,6 +39,16 @@ abstract class SoneCommandTest {
                whenever(core.getPostReply(anyString())).thenReturn(absent())
        }
 
+       protected fun createSone(id: String, name: String, firstName: String, lastName: String, time: Long) = mock<Sone>().apply {
+               whenever(this.id).thenReturn(id)
+               whenever(this.name).thenReturn(name)
+               whenever(profile).thenReturn(Profile(this).apply {
+                       this.firstName = firstName
+                       this.lastName = lastName
+               })
+               whenever(this.time).thenReturn(time)
+       }
+
        protected fun executeCommandAndExpectFcpException() {
                expectedException.expect(FcpException::class.java)
                command.execute(parameters)