Add processing of classes into model.

2 files changed, 44 insertions, 1 deletions

diff --git a/test/src/ClassTest.kt b/test/src/ClassTest.kt
new file mode 100644
index 00000000..914df53b
--- /dev/null
+++ b/test/src/ClassTest.kt
@@ -0,0 +1,43 @@
+package com.jetbrains.dokka.tests
+
+import org.junit.Test
+import kotlin.test.*
+import com.jetbrains.dokka.*
+
+public class ClassTest {
+    Test fun emptyClass() {
+        verifyModel("test/data/classes/emptyClass.kt") { model ->
+            val item = model.nodes.single().members.single()
+            assertEquals(DocumentationNodeKind.Class, item.kind)
+            assertEquals("Klass", item.name)
+            assertEquals("", item.doc)
+            assertTrue(item.details.none())
+            assertEquals("<init>", item.members.single().name)
+            assertTrue(item.links.none())
+        }
+    }
+
+    Test fun classWithConstructor() {
+        verifyModel("test/data/classes/classWithConstructor.kt") { model ->
+            val item = model.nodes.single().members.single()
+            assertEquals(DocumentationNodeKind.Class, item.kind)
+            assertEquals("Klass", item.name)
+            assertEquals("", item.doc)
+            assertTrue(item.details.none())
+            assertEquals("<init>", item.members.single().name)
+            assertTrue(item.links.none())
+        }
+    }
+
+    Test fun classWithFunction() {
+        verifyModel("test/data/classes/classWithFunction.kt") { model ->
+            val item = model.nodes.single().members.single()
+            assertEquals(DocumentationNodeKind.Class, item.kind)
+            assertEquals("Klass", item.name)
+            assertEquals("", item.doc)
+            assertTrue(item.details.none())
+            assertEquals("<init>", item.members.single().name)
+            assertTrue(item.links.none())
+        }
+    }
+}
\ No newline at end of file
diff --git a/test/src/TopLevelFunctionTest.kt b/test/src/FunctionTest.kt
index 342708f2..e8afe97b 100644
--- a/test/src/TopLevelFunctionTest.kt
+++ b/test/src/FunctionTest.kt
@@ -5,7 +5,7 @@ import kotlin.test.*
 import com.jetbrains.dokka.*
 
 
-public class TopLevelFunctionTest {
+public class FunctionTest {
     Test fun function() {
         verifyModel("test/data/functions/function.kt") { model ->
             val item = model.nodes.single().members.single()