import com.vuln.Package;
import com.vuln.DateService;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.Arguments;
import org.junit.jupiter.params.provider.MethodSource;

import java.util.stream.Stream;  

private static final String DETECTION_DATE = "2023-03-15T04:20:00";

private final DateService dateService = new DateServiceImpl();

private static Stream<Arguments> vulnerabilities() {
    return Stream.of(
      Arguments.of(0.0, DETECTION_DATE, null),
      Arguments.of(0.04, DETECTION_DATE, null),
    );
  }

  @ParameterizedTest
  @MethodSource("vulnerabilities")
  public void testGetDueDateForPackage(double score, String detectionDate, String expectedDueDate) {
    Package issue = new Package();
    issue.setScore(score);
    issue.setDetectionDate(detectionDate);

    String actualDueDate = dateService.getDueDate(issue);

    assertEquals(expectedDueDate, actualDueDate);
  }