From 50400fd70785ab2350a1eb63ee8fe9bff5704ed3 Mon Sep 17 00:00:00 2001
From: Stepan Koltsov <stepan.koltsov@jetbrains.com>
Date: Tue, 20 Dec 2011 00:28:30 +0400
Subject: [PATCH] ReadClassDataTest: trait

---
 .../org/jetbrains/jet/ReadClassDataTest.java     | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/compiler/tests/org/jetbrains/jet/ReadClassDataTest.java b/compiler/tests/org/jetbrains/jet/ReadClassDataTest.java
index b8a9351f937..8960f3ccc53 100644
--- a/compiler/tests/org/jetbrains/jet/ReadClassDataTest.java
+++ b/compiler/tests/org/jetbrains/jet/ReadClassDataTest.java
@@ -145,7 +145,8 @@ public class ReadClassDataTest extends UsefulTestCase {
     private String serializeContent(ClassDescriptor klass) {
 
         StringBuilder sb = new StringBuilder();
-        sb.append("class ");
+        serialize(klass.getKind(), sb);
+        sb.append(" ");
         
         serialize(klass, sb);
 
@@ -185,6 +186,19 @@ public class ReadClassDataTest extends UsefulTestCase {
         sb.append("}\n");
         return sb.toString();
     }
+    
+    private void serialize(ClassKind kind, StringBuilder sb) {
+        switch (kind) {
+            case CLASS:
+                sb.append("class");
+                break;
+            case TRAIT:
+                sb.append("trait");
+                break;
+            default:
+                throw new IllegalStateException();
+        }
+    }
 
 
     private void compareFunctions(@NotNull FunctionDescriptor a, @NotNull FunctionDescriptor b) {