Fix possible infinity loop in ag/417350

ag/417350 is a cherry pick of the merged CL ag/416106 which
is why this is in klp-dev. I will cherry pick this into klp-docs
after it has been merged.

Change-Id: Ifcf1fc901acd73ad9ccfef789b73cd5a62a6302c

1 files changed, 3 insertions, 0 deletions

diff --git a/common/src/java/com/example/android/common/dummydata/Cheeses.java b/common/src/java/com/example/android/common/dummydata/Cheeses.java
index 220d66a7..783735ce 100644
--- a/common/src/java/com/example/android/common/dummydata/Cheeses.java
+++ b/common/src/java/com/example/android/common/dummydata/Cheeses.java
@@ -174,6 +174,9 @@ public class Cheeses {
         Random random = new Random();
         HashSet<String> items = new HashSet<String>();
 
+        // Make sure that don't infinity loop
+        count = Math.min(count, CHEESES.length);
+
         while (items.size() < count) {
             items.add(CHEESES[random.nextInt(CHEESES.length)]);
         }