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
39 val remoteSone1 = createRemoteSone()
40 val remoteSone2 = createRemoteSone()
42 val localSone1 = createLocalSone()
43 val localSone2 = createLocalSone()
45 val createRequestUri: FreenetURI get() = InsertableClientSSK.createRandom(DummyRandomSource(), "").uri
46 val createInsertUri: FreenetURI get() = InsertableClientSSK.createRandom(DummyRandomSource(), "").insertURI
47 fun createId() = InsertableClientSSK.createRandom(DummyRandomSource(), "").uri.routingKey.asFreenetBase64
49 fun createOwnIdentity(id: String = "", nickname: String = "", requestUri: String = "", insertUri: String = "", contexts: Set<String> = setOf(), properties: Map<String, String> = mapOf()): OwnIdentity =
50 DefaultOwnIdentity(id, nickname, requestUri, insertUri).apply {
52 this.properties = properties
55 fun createIdentity(id: String = "", nickname: String = "", requestUri: String = "", contexts: Set<String> = setOf(), properties: Map<String, String> = mapOf()): Identity =
56 DefaultIdentity(id, nickname, requestUri).apply {
58 this.properties = properties
61 fun createLocalSone(id: String = createId(), identity: Identity = createOwnIdentity(id)): Sone = object : IdOnlySone(id) {
62 private val options = DefaultSoneOptions()
63 private val friends = mutableListOf<String>()
64 override fun getIdentity(): Identity = identity
65 override fun getOptions() = options
66 override fun isLocal() = true
67 override fun getFriends() = friends
68 override fun hasFriend(friendSoneId: String) = friendSoneId in friends
71 fun createRemoteSone(id: String = createId(), identity: Identity = createIdentity(id)): Sone = object : IdOnlySone(id) {
72 override fun getIdentity(): Identity = identity
75 fun createPost(text: String = "text", sone: Sone? = remoteSone1, known: Boolean = false, time: Long = 1, loaded: Boolean = true, recipient: Sone? = null, id: String = UUID.randomUUID().toString()): Post {
76 return object : Post.EmptyPost(id) {
77 override fun getRecipientId() = recipient?.id.asOptional()
78 override fun getRecipient() = recipient.asOptional()
79 override fun getSone() = sone
80 override fun getText() = text
81 override fun isKnown() = known
82 override fun getTime() = time
83 override fun isLoaded() = loaded
87 fun createPostReply(text: String = "text", post: Post? = createPost(), sone: Sone = remoteSone1, known: Boolean = false, time: Long = 1, id: String = "reply-id") = object : PostReply {
89 override fun getSone() = sone
90 override fun getPostId() = post!!.id
91 override fun getPost(): Optional<Post> = Optional.fromNullable(post)
92 override fun getTime() = time
93 override fun getText() = text
94 override fun isKnown() = known
97 fun createImage(sone: Sone): Image =
98 ImageImpl().modify().setSone(sone).update()