From 03329b0eb98b309d208839344052633028e00984 Mon Sep 17 00:00:00 2001
From: Szymon Świstun <sswistun@virtuslab.com>
Date: Wed, 18 Mar 2020 14:19:54 +0100
Subject: Fix minor bugs and add core tests

---
 core/src/main/kotlin/model/aditionalExtras.kt | 22 +++++++++++++++++-----
 1 file changed, 17 insertions(+), 5 deletions(-)

(limited to 'core/src/main')

diff --git a/core/src/main/kotlin/model/aditionalExtras.kt b/core/src/main/kotlin/model/aditionalExtras.kt
index b0755759..f7d37111 100644
--- a/core/src/main/kotlin/model/aditionalExtras.kt
+++ b/core/src/main/kotlin/model/aditionalExtras.kt
@@ -5,22 +5,34 @@ import org.jetbrains.dokka.model.properties.ExtraProperty
 import org.jetbrains.dokka.model.properties.MergeStrategy
 
 class AdditionalModifiers(val content: Set<ExtraModifiers>) : ExtraProperty<Documentable> {
-    object AdditionalKey : ExtraProperty.Key<Documentable, AdditionalModifiers> {
+    companion object : ExtraProperty.Key<Documentable, AdditionalModifiers> {
         override fun mergeStrategyFor(
             left: AdditionalModifiers,
             right: AdditionalModifiers
         ): MergeStrategy<Documentable> = MergeStrategy.Replace(AdditionalModifiers(left.content + right.content))
     }
 
-    override fun equals(other: Any?): Boolean = if (other is AdditionalModifiers) other.content == content else false
+    override fun equals(other: Any?): Boolean =
+        if (other is AdditionalModifiers) other.content == content else false
+
     override fun hashCode() = content.hashCode()
-    override val key: ExtraProperty.Key<Documentable, *> = AdditionalKey
+    override val key: ExtraProperty.Key<Documentable, *> = AdditionalModifiers
 }
 
 class Annotations(val content: List<Annotation>) : ExtraProperty<Documentable> {
-    companion object : ExtraProperty.Key<Documentable, Annotations>
+    companion object : ExtraProperty.Key<Documentable, Annotations> {
+        override fun mergeStrategyFor(left: Annotations, right: Annotations): MergeStrategy<Documentable> =
+            MergeStrategy.Replace(Annotations((left.content + right.content).distinct()))
+    }
 
     override val key: ExtraProperty.Key<Documentable, *> = Annotations
 
-    data class Annotation(val dri: DRI, val params: Map<String, String>)
+    data class Annotation(val dri: DRI, val params: Map<String, String>) {
+        override fun equals(other: Any?): Boolean = when(other) {
+            is Annotation -> dri.equals(other.dri)
+            else -> false
+        }
+
+        override fun hashCode(): Int = dri.hashCode()
+    }
 }
\ No newline at end of file
-- 
cgit