test: add limit parameter

Author: xchopin

src/test/java/org/apereo/openlrw/risk/RiskControllerTest.java

@@ -70,8 +70,8 @@ public void init() throws OrgNotFoundException, LineItemNotFoundException {
   public void testGetResultForUserResultNotFoundException() throws Exception {
     String userSourcedId = "jo zimmerman";
     String classSourcedId = "dead mow cinco";
-    when(riskService.getRisksForUserAndClass(TestData.TENANT_1, "*", classSourcedId, userSourcedId, "")).thenThrow(Exception.class);
-    riskController.getClassUser(jwtToken, classSourcedId, userSourcedId, "");
+    when(riskService.getRisksForUserAndClass(TestData.TENANT_1, "*", classSourcedId, userSourcedId, "", 0)).thenThrow(Exception.class);
+    riskController.getClassUser(jwtToken, classSourcedId, userSourcedId, "", 0);
   }
 
 

src/test/java/org/apereo/openlrw/risk/RiskServiceTest.java

@@ -63,7 +63,7 @@ public void testFindByClassAndUser() throws ParseException {
 
         unit.save("tenant-1","org-1", risk, true);
 
-        Collection<MongoRisk> found = unit.getRisksForUserAndClass("tenant-1","org-1", "class-id","user-id", "");
+        Collection<MongoRisk> found = unit.getRisksForUserAndClass("tenant-1","org-1", "class-id","user-id", "",0 );
         ArrayList<MongoRisk> list = new ArrayList<>(found);
 
         assertThat(found, is(notNullValue()));
@@ -74,7 +74,7 @@ public void testFindByClassAndUser() throws ParseException {
     @Test
     public void testFindByClassAndUserAndDate() throws ParseException {
         SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm");
-        Instant date = sdf.parse("2019-02-25 08:00").toInstant();
+        Instant date = sdf.parse("2019-02-24 08:00").toInstant();
 
         MongoRisk risk = new MongoRisk.Builder()
                 .withClassSourcedId("class-id")
@@ -85,7 +85,7 @@ public void testFindByClassAndUserAndDate() throws ParseException {
                 .build();
         unit.save("tenant-1","org-1", risk, true);
 
-        date = sdf.parse("2019-02-25 18:00").toInstant();
+        date = sdf.parse("2019-02-26 18:00").toInstant();
         risk = new MongoRisk.Builder()
                 .withClassSourcedId("class-id")
                 .withUserSourcedId("user-id")
@@ -106,22 +106,21 @@ public void testFindByClassAndUserAndDate() throws ParseException {
         unit.save("tenant-1","org-1", risk, true);
 
 
-        Collection<MongoRisk> found = unit.getRisksForUserAndClass("tenant-1", "org-1", "class-id", "user-id", "");
+        Collection<MongoRisk> found = unit.getRisksForUserAndClass("tenant-1", "org-1", "class-id", "user-id", "",0 );
         ArrayList<MongoRisk> list = new ArrayList<>(found);
 
 
         assertThat(found, is(notNullValue()));
         assertThat(list.size(), is(4));
 
-
-        found = unit.getRisksForUserAndClass("tenant-1", "org-1", "class-id", "user-id", "2019-02-25 05:00");
-       list = new ArrayList<>(found);
+        found = unit.getRisksForUserAndClass("tenant-1", "org-1", "class-id", "user-id", "2019-02-25", 0);
+        list = new ArrayList<>(found);
 
         assertThat(found, is(notNullValue()));
         assertThat(list.get(0).getName(), containsString("5am"));
         assertThat(list.size(), is(1));
 
-        found = unit.getRisksForUserAndClass("tenant-1", "org-1", "class-id", "user-id", "latest");
+        found = unit.getRisksForUserAndClass("tenant-1", "org-1", "class-id", "user-id", "latest", 0);
 
         list = new ArrayList<>(found);
 
@@ -132,4 +131,48 @@ public void testFindByClassAndUserAndDate() throws ParseException {
     }
 
 
+    @Test
+    public void testFindByClassLimit() throws ParseException {
+        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm");
+        Instant date = sdf.parse("2019-03-22 15:38").toInstant();
+
+        MongoRisk risk = new MongoRisk.Builder()
+                .withClassSourcedId("class-id")
+                .withUserSourcedId("user-id")
+                .withScore("80/100")
+                .withName("First risk")
+                .withDateTime(date)
+                .withVelocity("-1")
+                .build();
+
+        Instant date2 = sdf.parse("2019-03-28 15:38").toInstant();
+        MongoRisk risk2 = new MongoRisk.Builder()
+                .withClassSourcedId("class-id")
+                .withUserSourcedId("user-id")
+                .withScore("0.8")
+                .withName("Second risk")
+                .withDateTime(date2)
+                .withVelocity("1")
+                .build();
+
+        Instant date3 = sdf.parse("2020-03-28 15:38").toInstant();
+        MongoRisk risk3 = new MongoRisk.Builder()
+                .withClassSourcedId("class-id")
+                .withUserSourcedId("user-id")
+                .withName("Third risk")
+                .withDateTime(date3)
+                .withVelocity("0.5")
+                .build();
+
+        unit.save("tenant-1","org-1", risk, true);
+        unit.save("tenant-1","org-1", risk2, true);
+        unit.save("tenant-1","org-1", risk3, true);
+
+        Collection<MongoRisk> found = unit.getRisksForUserAndClass("tenant-1","org-1", "class-id","user-id", "",2 );
+        ArrayList<MongoRisk> list = new ArrayList<>(found);
+
+        assertThat(found, is(notNullValue()));
+        assertThat(list.size(), is(equalTo(2)));
+
+    }
 }