validate for null

yegor256 · Jun 19, 2017 · 5771bc8 · 5771bc8
1 parent b83c546
commit 5771bc8
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/main/java/org/cactoos/func/RepeatedFunc.java b/src/main/java/org/cactoos/func/RepeatedFunc.java
@@ -24,6 +24,7 @@
 package org.cactoos.func;
 
 import org.cactoos.Func;
+import org.cactoos.text.FormattedText;
 
 /**
  * Func that repeats its calculation a few times before
@@ -67,6 +68,14 @@ public Y apply(final X input) throws Exception {
         for (int idx = 0; idx < this.times; ++idx) {
             result = this.func.apply(input);
         }
+        if (result == null) {
+            throw new IllegalArgumentException(
+                new FormattedText(
+                    "Repeat counter is equal or less than zero: %d",
+                    this.times
+                ).asString()
+            );
+        }
         return result;
     }