EnrollmentController.java

1
package edu.ucsb.cs156.courses.controllers;
2
3
import com.opencsv.bean.StatefulBeanToCsv;
4
import com.opencsv.exceptions.CsvDataTypeMismatchException;
5
import com.opencsv.exceptions.CsvRequiredFieldEmptyException;
6
import edu.ucsb.cs156.courses.entities.EnrollmentDataPoint;
7
import edu.ucsb.cs156.courses.models.EnrollmentCSV;
8
import edu.ucsb.cs156.courses.repositories.EnrollmentDataPointRepository;
9
import edu.ucsb.cs156.courses.services.EnrollmentCSVService;
10
import io.swagger.v3.oas.annotations.Operation;
11
import io.swagger.v3.oas.annotations.Parameter;
12
import io.swagger.v3.oas.annotations.media.Content;
13
import io.swagger.v3.oas.annotations.media.Schema;
14
import io.swagger.v3.oas.annotations.responses.ApiResponse;
15
import io.swagger.v3.oas.annotations.tags.Tag;
16
import java.io.IOException;
17
import java.io.OutputStreamWriter;
18
import java.io.Writer;
19
import java.nio.charset.StandardCharsets;
20
import java.util.List;
21
import java.util.stream.Collectors;
22
import lombok.extern.slf4j.Slf4j;
23
import org.springframework.beans.factory.annotation.Autowired;
24
import org.springframework.data.util.Streamable;
25
import org.springframework.http.HttpHeaders;
26
import org.springframework.http.MediaType;
27
import org.springframework.http.ResponseEntity;
28
import org.springframework.web.bind.annotation.GetMapping;
29
import org.springframework.web.bind.annotation.RequestMapping;
30
import org.springframework.web.bind.annotation.RequestParam;
31
import org.springframework.web.bind.annotation.RestController;
32
import org.springframework.web.servlet.mvc.method.annotation.StreamingResponseBody;
33
34
@Slf4j
35
@Tag(name = "API for enrollment data")
36
@RequestMapping("/api/enrollment")
37
@RestController
38
public class EnrollmentController extends ApiController {
39
40
  @Autowired EnrollmentDataPointRepository enrollmentDataPointRepository;
41
  @Autowired private EnrollmentCSVService enrollmentCSVService;
42
43
  @Operation(
44
      summary = "Download Enrollment Data as CSV File",
45
      description = "Returns a CSV file as a response",
46
      responses = {
47
        @ApiResponse(
48
            responseCode = "200",
49
            description = "CSV file",
50
            content =
51
                @Content(
52
                    mediaType = "text/csv",
53
                    schema = @Schema(type = "string", format = "binary"))),
54
        @ApiResponse(responseCode = "500", description = "Internal Server Error")
55
      })
56
  @GetMapping(value = "/csv/quarter", produces = "text/csv")
57
  public ResponseEntity<StreamingResponseBody> csvForQuarter(
58
      @Parameter(name = "yyyyq", description = "quarter in yyyyq format", example = "20252")
59
          @RequestParam
60
          String yyyyq)
61
      throws IOException {
62
63
    StreamingResponseBody stream =
64
        (outputStream) -> {
65
          Iterable<EnrollmentDataPoint> iterable = enrollmentDataPointRepository.findByYyyyq(yyyyq);
66
          List<EnrollmentCSV> list =
67
              Streamable.of(iterable).toList().stream()
68 1 1. lambda$csvForQuarter$0 : replaced return value with null for edu/ucsb/cs156/courses/controllers/EnrollmentController::lambda$csvForQuarter$0 → KILLED
                  .map(enrollmentDataPoint -> EnrollmentCSV.fromEntity(enrollmentDataPoint))
69
                  .collect(Collectors.toList());
70
71
          try (Writer writer = new OutputStreamWriter(outputStream, StandardCharsets.UTF_8)) {
72
            try {
73
              StatefulBeanToCsv<EnrollmentCSV> beanToCsvWriter =
74
                  enrollmentCSVService.getStatefulBeanToCSV(writer);
75 1 1. lambda$csvForQuarter$1 : removed call to com/opencsv/bean/StatefulBeanToCsv::write → KILLED
              beanToCsvWriter.write(list);
76
            } catch (CsvDataTypeMismatchException | CsvRequiredFieldEmptyException e) {
77
              log.error("Error writing CSV file", e);
78
              throw new IOException("Error writing CSV file: " + e.getMessage());
79
            }
80
          }
81
        };
82
83 1 1. csvForQuarter : replaced return value with null for edu/ucsb/cs156/courses/controllers/EnrollmentController::csvForQuarter → KILLED
    return ResponseEntity.ok()
84
        .contentType(MediaType.parseMediaType("text/csv; charset=UTF-8"))
85
        .header(
86
            HttpHeaders.CONTENT_DISPOSITION,
87
            String.format("attachment;filename=enrollment_%s.csv", yyyyq))
88
        .header(HttpHeaders.CONTENT_TYPE, "text/csv; charset=UTF-8")
89
        .header(HttpHeaders.ACCESS_CONTROL_EXPOSE_HEADERS, HttpHeaders.CONTENT_DISPOSITION)
90
        .body(stream);
91
  }
92
}

Mutations

68

1.1
Location : lambda$csvForQuarter$0
Killed by : edu.ucsb.cs156.courses.controllers.EnrollmentControllerTests.[engine:junit-jupiter]/[class:edu.ucsb.cs156.courses.controllers.EnrollmentControllerTests]/[method:testCsvForQuarter_mappingProducesNonNullValues()]
replaced return value with null for edu/ucsb/cs156/courses/controllers/EnrollmentController::lambda$csvForQuarter$0 → KILLED

75

1.1
Location : lambda$csvForQuarter$1
Killed by : edu.ucsb.cs156.courses.controllers.EnrollmentControllerTests.[engine:junit-jupiter]/[class:edu.ucsb.cs156.courses.controllers.EnrollmentControllerTests]/[method:testCsvForQuarter_success()]
removed call to com/opencsv/bean/StatefulBeanToCsv::write → KILLED

83

1.1
Location : csvForQuarter
Killed by : edu.ucsb.cs156.courses.controllers.EnrollmentControllerTests.[engine:junit-jupiter]/[class:edu.ucsb.cs156.courses.controllers.EnrollmentControllerTests]/[method:testCsvForQuarter_success()]
replaced return value with null for edu/ucsb/cs156/courses/controllers/EnrollmentController::csvForQuarter → KILLED

Active mutators

Tests examined


Report generated by PIT 1.17.0