fix: PR remarks

Signed-off-by: LE SAULNIER Kevin <[email protected]>

Author: LE SAULNIER Kevin

src/main/java/org/gridsuite/securityanalysis/server/entities/AbstractLimitViolationEntity.java

@@ -137,14 +137,9 @@ protected static Integer calculateActualOverloadDuration(LimitViolation limitVio
             String equipmentId = limitViolation.getSubjectId();
             LoadingLimits.TemporaryLimit tempLimit = null;
 
-            Line line = network.getLine(equipmentId);
-            if (line != null) {
-                tempLimit = getBranchLimitViolationAboveCurrentValue(line, limitViolation);
-            } else {
-                TwoWindingsTransformer twoWindingsTransformer = network.getTwoWindingsTransformer(equipmentId);
-                if (twoWindingsTransformer != null) {
-                    tempLimit = getBranchLimitViolationAboveCurrentValue(twoWindingsTransformer, limitViolation);
-                }
+            Branch<?> branch = network.getBranch(equipmentId);
+            if (branch != null) {
+                tempLimit = getBranchLimitViolationAboveCurrentValue(branch, limitViolation);
             }
             return (tempLimit != null) ? tempLimit.getAcceptableDuration() : Integer.MAX_VALUE;
         }
@@ -159,7 +154,7 @@ protected static Integer calculateUpcomingOverloadDuration(LimitViolation limitV
 
     private static LoadingLimits.TemporaryLimit getBranchLimitViolationAboveCurrentValue(Branch<?> branch, LimitViolation limitViolation) {
         // limits are returned from the store by DESC duration / ASC value
-        Optional<CurrentLimits> currentLimits = limitViolation.getSideAsTwoSides().equals(TwoSides.ONE) ? branch.getCurrentLimits1() : branch.getCurrentLimits2();
+        Optional<CurrentLimits> currentLimits = branch.getCurrentLimits(limitViolation.getSideAsTwoSides());
         if (currentLimits.isEmpty() || limitViolation.getValue() < currentLimits.get().getPermanentLimit()) {
             return null;
         } else {