2 * Sone - Mocks.kt - Copyright © 2019–2020 David ‘Bombe’ Roden
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 3 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 package net.pterodactylus.sone.test
20 import com.google.common.base.Optional
21 import freenet.crypt.DummyRandomSource
22 import freenet.keys.FreenetURI
23 import freenet.keys.InsertableClientSSK
24 import net.pterodactylus.sone.data.Image
25 import net.pterodactylus.sone.data.Post
26 import net.pterodactylus.sone.data.PostReply
27 import net.pterodactylus.sone.data.Sone
28 import net.pterodactylus.sone.data.SoneOptions.DefaultSoneOptions
29 import net.pterodactylus.sone.data.impl.IdOnlySone
30 import net.pterodactylus.sone.data.impl.ImageImpl
31 import net.pterodactylus.sone.freenet.wot.DefaultIdentity
32 import net.pterodactylus.sone.freenet.wot.DefaultOwnIdentity
33 import net.pterodactylus.sone.freenet.wot.Identity
34 import net.pterodactylus.sone.freenet.wot.OwnIdentity
35 import net.pterodactylus.sone.utils.asFreenetBase64
36 import net.pterodactylus.sone.utils.asOptional
38 val remoteSone1 = createRemoteSone()
39 val remoteSone2 = createRemoteSone()
41 val localSone1 = createLocalSone()
42 val localSone2 = createLocalSone()
44 val createRequestUri: FreenetURI get() = InsertableClientSSK.createRandom(DummyRandomSource(), "").uri
45 val createInsertUri: FreenetURI get() = InsertableClientSSK.createRandom(DummyRandomSource(), "").insertURI
46 fun createId() = InsertableClientSSK.createRandom(DummyRandomSource(), "").uri.routingKey.asFreenetBase64
48 fun createOwnIdentity(id: String = "", nickname: String = "", requestUri: String = "", insertUri: String = "", contexts: Set<String> = setOf(), properties: Map<String, String> = mapOf()): OwnIdentity =
49 DefaultOwnIdentity(id, nickname, requestUri, insertUri).apply {
51 this.properties = properties
54 fun createIdentity(id: String = "", nickname: String = "", requestUri: String = "", contexts: Set<String> = setOf(), properties: Map<String, String> = mapOf()): Identity =
55 DefaultIdentity(id, nickname, requestUri).apply {
57 this.properties = properties
60 fun createLocalSone(id: String = createId(), identity: Identity = createOwnIdentity(id)): Sone = object : IdOnlySone(id) {
61 private val options = DefaultSoneOptions()
62 private val friends = mutableListOf<String>()
63 override fun getIdentity(): Identity = identity
64 override fun getOptions() = options
65 override fun isLocal() = true
66 override fun getFriends() = friends
67 override fun hasFriend(friendSoneId: String) = friendSoneId in friends
70 fun createRemoteSone(id: String = createId(), identity: Identity = createIdentity(id)): Sone = object : IdOnlySone(id) {
71 override fun getIdentity(): Identity = identity
74 fun createPost(text: String = "", sone: Sone? = remoteSone1, known: Boolean = false, time: Long = 1, loaded: Boolean = true, recipient: Sone? = null): Post {
75 return object : Post.EmptyPost("post-id") {
76 override fun getRecipientId() = recipient?.id.asOptional()
77 override fun getRecipient() = recipient.asOptional()
78 override fun getSone() = sone
79 override fun getText() = text
80 override fun isKnown() = known
81 override fun getTime() = time
82 override fun isLoaded() = loaded
86 fun createPostReply(text: String = "", post: Post? = createPost(), sone: Sone = remoteSone1, known: Boolean = false, time: Long = 1) = object : PostReply {
87 override val id = "reply-id"
88 override fun getSone() = sone
89 override fun getPostId() = post!!.id
90 override fun getPost(): Optional<Post> = Optional.fromNullable(post)
91 override fun getTime() = time
92 override fun getText() = text
93 override fun isKnown() = known
96 fun createImage(sone: Sone): Image =
97 ImageImpl().modify().setSone(sone).update()