From a76e7f6d56c55147bfe2a6527f1f00c1d9a3df69 Mon Sep 17 00:00:00 2001
From: Andrey Breslav <andrey.breslav@jetbrains.com>
Date: Mon, 25 Jun 2012 17:48:56 +0200
Subject: [PATCH] Enum class kinds supported. Assertion for no return type made
 explicit

---
 .../jet/jvm/compiler/NamespaceComparator.java      | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/compiler/tests/org/jetbrains/jet/jvm/compiler/NamespaceComparator.java b/compiler/tests/org/jetbrains/jet/jvm/compiler/NamespaceComparator.java
index 2f6fdb9e015..37016077c53 100644
--- a/compiler/tests/org/jetbrains/jet/jvm/compiler/NamespaceComparator.java
+++ b/compiler/tests/org/jetbrains/jet/jvm/compiler/NamespaceComparator.java
@@ -246,6 +246,12 @@ public class NamespaceComparator {
                 case ANNOTATION_CLASS:
                     sb.append("annotation class");
                     break;
+                case ENUM_CLASS:
+                    sb.append("enum class");
+                    break;
+                case ENUM_ENTRY:
+                    sb.append("enum entry");
+                    break;
                 default:
                     throw new IllegalStateException("unknown class kind: " + kind);
             }
@@ -295,7 +301,13 @@ public class NamespaceComparator {
             sb.append("(");
             new TypeSerializer(sb).serializeCommaSeparated(fun.getValueParameters());
             sb.append("): ");
-            new TypeSerializer(sb).serialize(fun.getReturnType());
+            JetType returnType = fun.getReturnType();
+            if (returnType == null) {
+                throw new IllegalStateException("No return type for " + fun);
+            }
+            else {
+                new TypeSerializer(sb).serialize(returnType);
+            }
         }
 
         public void serialize(ExtensionReceiver extensionReceiver) {