Code review fix

Author: KarnaiahPesula

sormas-backend/src/main/java/de/symeda/sormas/backend/caze/CaseFacadeEjb.java

@@ -2204,15 +2204,23 @@ public void onCaseChanged(CaseDataDto existingCase, Case newCase, boolean syncSh
 			}
 		}
 
-		if (Objects.nonNull(existingCase)) {
-			// Contact with the source case known with the confirmed status. Github#13507 fix
-			boolean isConfirmedCaseExists = contactService.getAllByResultingCase(caseService.getByUuid(existingCase.getUuid())).stream().anyMatch(e -> CaseClassification.CONFIRMED == e.getCaze().getCaseClassification());
-			// If contact with source case is known by epidemiolist, && its classified as confirmed, then set epidemiological confirmation to YES
-			// Otherwise set to No
-			if ((YesNoUnknown.YES == newCase.getEpiData().getContactWithSourceCaseKnown() && isConfirmedCaseExists)) {
-				newCase.setEpidemiologicalConfirmation(YesNoUnknown.YES);
-			} else {
+		if (existingCase != null && newCase.getEpiData() != null) {
+			// If contact with source case is known and at least one source case (same disease) is confirmed, set YES.
+			final YesNoUnknown contactKnown = newCase.getEpiData().getContactWithSourceCaseKnown();
+			if (contactKnown == YesNoUnknown.YES) {
+				// Check if at least one source case is confirmed
+				// We only check for contacts that have this case as resulting case
+				final boolean hasConfirmedSourceCase = contactService.getAllByResultingCase(caseService.getByUuid(existingCase.getUuid())).stream()
+						.map(Contact::getCaze)
+						.filter(Objects::nonNull)
+						.anyMatch(src ->
+								src.getDisease() == newCase.getDisease()
+										&& CaseClassification.getConfirmedClassifications().contains(src.getCaseClassification()));
+				newCase.setEpidemiologicalConfirmation(hasConfirmedSourceCase ? YesNoUnknown.YES : YesNoUnknown.NO);
+			} else if (contactKnown == YesNoUnknown.NO) {
 				newCase.setEpidemiologicalConfirmation(YesNoUnknown.NO);
+			} else {
+				newCase.setEpidemiologicalConfirmation(YesNoUnknown.UNKNOWN);
 			}
 		}