1 files changed, 12 insertions, 4 deletions
diff --git a/src/main/java/io/trygvis/esper/testing/web/resource/BadgeJson.java b/src/main/java/io/trygvis/esper/testing/web/resource/BadgeJson.java
index ded97ae..90d9c93 100644
--- a/src/main/java/io/trygvis/esper/testing/web/resource/BadgeJson.java
+++ b/src/main/java/io/trygvis/esper/testing/web/resource/BadgeJson.java
@@ -1,8 +1,12 @@
 package io.trygvis.esper.testing.web.resource;
 
+import io.trygvis.esper.testing.core.badge.*;
 import org.joda.time.*;
 
+import java.util.*;
+
 public class BadgeJson {
+    public final UUID uuid;
     public final DateTime createdDate;
     public final String name;
     public final int level;
@@ -13,7 +17,8 @@ public class BadgeJson {
     /**
      * For completed badges.
      */
-    public BadgeJson(DateTime createdDate, String name, int level) {
+    public BadgeJson(UUID uuid, DateTime createdDate, String name, int level) {
+        this.uuid = uuid;
         this.createdDate = createdDate;
         this.name = name;
         this.level = level;
@@ -24,8 +29,9 @@ public class BadgeJson {
     /**
      * For badges in progress.
      */
-    public BadgeJson(DateTime createdDate, String name, int level, int progress, int goal) {
-        this.createdDate = createdDate;
+    public BadgeJson(String name, int level, int progress, int goal) {
+        this.uuid = null;
+        this.createdDate = null;
         this.name = name;
         this.level = level;
         this.progress = progress;
@@ -35,10 +41,12 @@ public class BadgeJson {
 
 class BadgeDetailJson {
     public final BadgeJson badge;
+    public final PersonalBadge personalBadge;
     public final PersonJson person;
 
-    BadgeDetailJson(BadgeJson badge, PersonJson person) {
+    BadgeDetailJson(BadgeJson badge, PersonalBadge personalBadge, PersonJson person) {
         this.badge = badge;
+        this.personalBadge = personalBadge;
         this.person = person;
     }
 }