DocumentHttpLiveTest.java

package uk.gov.hmcts.cp.cdk.http;

import static org.assertj.core.api.Assertions.assertThat;
import static uk.gov.hmcts.cp.cdk.util.UtilConstants.CJSCPPUID;
import static uk.gov.hmcts.cp.cdk.util.UtilConstants.USER_WITH_PERMISSIONS;

import uk.gov.hmcts.cp.cdk.testsupport.AbstractHttpLiveTest;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.time.OffsetDateTime;
import java.util.UUID;

import org.junit.jupiter.api.Test;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;

 class DocumentHttpLiveTest extends AbstractHttpLiveTest {
    public static final MediaType VND_TYPE_MATERIAL = MediaType.valueOf("application/vnd.casedocumentknowledge-service.document-content+json");


    @Test
    void materialContentUrl_returnsExpectedUrl() throws Exception {

        final UUID docId = UUID.randomUUID();
        final UUID materialId = UUID.randomUUID();

        try (Connection c = DriverManager.getConnection(jdbcUrl, jdbcUser, jdbcPass);
             PreparedStatement ps = c.prepareStatement(
                     "INSERT INTO case_documents (doc_id, case_id, material_id, source,doc_name ,blob_uri,uploaded_at) " +
                             "VALUES (?, ?, ?, ?, ?,?,?)"
             )) {
            ps.setObject(1, docId);
            ps.setObject(2, UUID.randomUUID());
            ps.setObject(3, materialId);
            ps.setString(4, "IDPC");
            ps.setString(5, "docnmae");
            ps.setString(6, "IDPC");
            ps.setObject(7, OffsetDateTime.now());
            ps.executeUpdate();
        }

        final HttpHeaders headers = new HttpHeaders();
        headers.setContentType(VND_TYPE_MATERIAL);
        headers.set(CJSCPPUID, USER_WITH_PERMISSIONS); // add required header

        // --- Act: perform GET request to the endpoint ---
        final ResponseEntity<String> res = http.exchange(
                baseUrl + "/document/" + docId + "/content",
                HttpMethod.GET,
                new HttpEntity<>(headers),
                String.class
        );


        assertThat(res.getStatusCode()).isEqualTo(HttpStatus.OK);
        assertThat(res.getBody()).contains("\"url\"");
        assertThat(res.getBody()).contains("http"); // the URL should look like a URI


        try (Connection c = DriverManager.getConnection(jdbcUrl, jdbcUser, jdbcPass);
             PreparedStatement del = c.prepareStatement("DELETE FROM case_documents WHERE doc_id = ?")) {
            del.setObject(1, docId);
            del.executeUpdate();
        }
    }
}