Polish ProtobufMessageBuilderRandomizer

murdos · May 16, 2021 · 2bd196c · 2bd196c
1 parent b629d30
commit 2bd196c
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/main/java/io/github/murdos/easyrandom/protobuf/ProtobufMessageBuilderRandomizer.java b/src/main/java/io/github/murdos/easyrandom/protobuf/ProtobufMessageBuilderRandomizer.java
@@ -20,6 +20,9 @@
 import org.jeasy.random.EasyRandomParameters;
 import org.jeasy.random.api.Randomizer;
 
+/**
+ * Generate a random Protobuf {@link Message.Builder}.
+ */
 public class ProtobufMessageBuilderRandomizer implements Randomizer<Message.Builder> {
 
     private final ProtobufMessageRandomizer protobufMessageRandomizer;
@@ -37,6 +40,7 @@ public ProtobufMessageBuilderRandomizer(
             );
     }
 
+    @SuppressWarnings("unchecked")
     private static Class<Message> retrieveMessageClassFromBuilderClass(Class<Message.Builder> messageBuilderClass) {
         return (Class<Message>) messageBuilderClass.getEnclosingClass();
     }
@@ -45,4 +49,8 @@ private static Class<Message> retrieveMessageClassFromBuilderClass(Class<Message
     public Message.Builder getRandomValue() {
         return protobufMessageRandomizer.getRandomValue().toBuilder();
     }
+
+    public String toString() {
+        return this.getClass().getSimpleName();
+    }
 }