Fix: Check for invalid visit number.

opensrp · May 17, 2017 · a206423 · a206423
1 parent cb5ee3c
commit a206423
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/...test/java/org/opensrp/register/mcare/report/mis1/maternityCare/data/ANCVisitTestData.java b/...test/java/org/opensrp/register/mcare/report/mis1/maternityCare/data/ANCVisitTestData.java
@@ -61,8 +61,10 @@ public List<Members> createANCVisitIsReferredData() {
 
 
         for (VisitNumber visitNumber : VisitNumber.values()) {
-            allMembers.add(createValidMemberWithStartDateTime(visitNumber));
-            allMembers.add(createValidMemberWithEndDateTime(visitNumber));
+            if(visitNumber != VisitNumber.INVALID) {
+                allMembers.add(createValidMemberWithStartDateTime(visitNumber));
+                allMembers.add(createValidMemberWithEndDateTime(visitNumber));
+            }
         }
 
         for (int i = 0; i < totalCount; i++) {
@@ -164,7 +166,7 @@ private Members getInvalidMemberBasedOn(int count) {
         }
     }
 
-    private Members createValidMemberWithValidRandomDateTime(VisitNumber visitNumber) {
+    private Members     createValidMemberWithValidRandomDateTime(VisitNumber visitNumber) {
         Long randomDateTimeBetweenStartAndEndDateTime = getRandomNumberBetween(startDateTime, endDateTime);
         builder.clientVersion(randomDateTimeBetweenStartAndEndDateTime).isReferred(YES).misoprostolReceived(YES);
         return createMemberWithANCVisit(visitNumber, builder.build().getVisitData());