2 files changed, 63 insertions, 3 deletions
diff --git a/core/src/test/kotlin/model/ContentSourceSetTest.kt b/core/src/test/kotlin/model/ContentSourceSetTest.kt
index b96180a2..74f7854b 100644
--- a/core/src/test/kotlin/model/ContentSourceSetTest.kt
+++ b/core/src/test/kotlin/model/ContentSourceSetTest.kt
@@ -19,17 +19,17 @@ class ContentSourceSetTest {
         )
 
         assertFalse(
-            DokkaSourceSetID("m3", "s3") in contentSourceSet,
+            DokkaSourceSetID("m3", "s3") in contentSourceSet.sourceSetIDs,
             "Expected source set id not being contained in content source set"
         )
 
         assertTrue(
-            DokkaSourceSetID("m1", "s1") in contentSourceSet,
+            DokkaSourceSetID("m1", "s1") in contentSourceSet.sourceSetIDs,
             "Expected source set id being contained in content source set"
         )
 
         assertTrue(
-            DokkaSourceSetID("m1+m2", "s1+s2") in contentSourceSet,
+            DokkaSourceSetID("m1+m2", "s1+s2") in contentSourceSet.sourceSetIDs,
             "Expected merged source set being contained in content source set"
         )
     }
diff --git a/core/src/test/kotlin/utilities/SelfRepresentingSingletonSetTest.kt b/core/src/test/kotlin/utilities/SelfRepresentingSingletonSetTest.kt
new file mode 100644
index 00000000..bef43565
--- /dev/null
+++ b/core/src/test/kotlin/utilities/SelfRepresentingSingletonSetTest.kt
@@ -0,0 +1,60 @@
+package utilities
+
+import org.jetbrains.dokka.utilities.SelfRepresentingSingletonSet
+import kotlin.test.Test
+import kotlin.test.assertEquals
+import kotlin.test.assertFalse
+import kotlin.test.assertTrue
+
+class SelfRepresentingSingletonSetTest {
+
+    data class TestModel(val index: Int = 0) : SelfRepresentingSingletonSet<TestModel>
+
+    @Test
+    fun size() {
+        assertEquals(1, TestModel().size)
+    }
+
+    @Test
+    fun contains() {
+        val m0 = TestModel(0)
+        val m1 = TestModel(1)
+
+        assertFalse(m1 in m0)
+        assertFalse(m0 in m1)
+        assertTrue(m0 in m0)
+        assertTrue(m1 in m1)
+        assertTrue(TestModel(0) in m0)
+    }
+
+    @Test
+    fun `containsAll is compliant to setOf`() {
+        val setOf = setOf(TestModel())
+        val testModel = TestModel()
+
+        assertEquals(
+            setOf.containsAll(emptyList()), testModel.containsAll(emptyList())
+        )
+
+        assertEquals(
+            setOf.containsAll(listOf(TestModel())), testModel.containsAll(listOf(TestModel()))
+        )
+
+        assertEquals(
+            setOf.containsAll(listOf(TestModel(0), TestModel(1))),
+            testModel.containsAll(listOf(TestModel(0), TestModel(1)))
+        )
+    }
+
+    @Test
+    fun isEmpty() {
+        assertFalse(TestModel().isEmpty())
+    }
+
+    @Test
+    fun iterator() {
+        assertEquals(
+            listOf(TestModel()), TestModel(0).iterator().asSequence().toList()
+        )
+    }
+}