src/test/java/de/ids_mannheim/korap/plkexport/CsvExporterTest.java

package de.ids_mannheim.korap.plkexport;

import java.io.IOException;

import static org.junit.Assert.assertEquals;
import org.junit.Test;

import jakarta.ws.rs.core.Response;

public class CsvExporterTest {
    
    @Test
    public void testInit () throws IOException {
        CsvExporter csv = new CsvExporter();
        csv.init("{\"query\":\"cool\"}");
        csv.finish();

        Response resp = csv.serve().build();
        String x = (String) resp.getEntity();
        resp.close();
        assertEquals(x, "HasMoreLeft,leftContext,Match,rightContext,HasMoreRight,"+
                     "isCutted,textSigle,author,pubDate,title\n");
    };

    @Test
    public void testInitFull () throws IOException {
        CsvExporter csv = new CsvExporter();
        csv.init("{\"meta\":\"ja\",\"collection\":\"hm\",\"query\":\"cool\"," +
                 "\"matches\":["+
                 "{\"author\":\"Goethe\","+
                 "\"title\":\"Title1\","+
                 "\"pubDate\":\"20051103\","+
                 "\"textSigle\":\"RTF/G59/34284\","+
                 "\"snippet\":\"Simple <mark>match1</mark> Snippet\"}"+
                 ","+
                 "{\"author\":\"Schiller\","+
                 "\"title\":\"Title2, the\","+
                 "\"pubDate\":\"20051104\","+
                 "\"textSigle\":\"RTF/G59/34285\","+
                 "\"snippet\":\"<span class=\\\"context-left\\\"><span class=\\\"more\\\"></span>"+
                 "Simpler, \\\"faster\\\" </span><span class=\\\"match\\\"><mark>&quot;match2&quot;</mark></span>"+
                 "<span class=\\\"context-right\\\"> Snippet"+
                 "<span class=\\\"more\\\"></span></span>\"}"+
                 "]}");
        csv.finish();

        Response resp = csv.serve().build();
        String x = (String) resp.getEntity();
        resp.close();
        String[] lines = x.split("\n");
        assertEquals(lines[0],"HasMoreLeft,leftContext,Match,rightContext,HasMoreRight,isCutted,textSigle,author,pubDate,title");
        assertEquals(lines[1],",Simple,match1,Snippet,,,RTF/G59/34284,Goethe,20051103,Title1");
        assertEquals(lines[2],"...,\"Simpler, \"\"faster\"\"\",\" \"\"match2\"\"\",Snippet,...,,RTF/G59/34285,Schiller,20051104,\"Title2, the\"");
        assertEquals(lines.length,3);

        csv = new CsvExporter();
        csv.init("{\"matches\":["+
                 "{\"title\":\"Title1\","+
                 "\"pubDate\":\"20051103\","+
                 "\"textSigle\":\"RTF/G59/34284\","+
                 "\"snippet\":\"Simple <mark>match1</mark> Snippet\"}"+
                 ","+
                 "{\"author\":\"Schiller\","+
                 "\"title\":\"Title2, the\","+
                 "\"pubDate\":\"20051104\","+
                 "\"textSigle\":\"RTF/G59/34285\","+
                 "\"snippet\":\"<span class=\\\"context-left\\\"><span class=\\\"more\\\"></span>"+
                 "Simpler, \\\"faster\\\" </span><span class=\\\"match\\\"><mark>&quot;match2&quot;</mark></span>"+
                 "<span class=\\\"context-right\\\"> Snippet"+
                 "<span class=\\\"more\\\"></span></span>\"}"+
                 "]}");
        csv.finish();
        resp = csv.serve().build();
        x = (String) resp.getEntity();
        resp.close();

    };

    @Test
    public void testCsvSanitize () throws IOException {
        CsvExporter csv = new CsvExporter();
        csv.init("{\"meta\":\"ja\",\"collection\":\"hm\",\"query\":\"cool\"," +
                 "\"matches\":["+
                 "{\"author\":\"Goethe\","+
                 "\"title\":\"@Title1\","+
                 "\"pubDate\":\"20051103\","+
                 "\"textSigle\":\"RTF/G59/34284\","+
                 "\"snippet\":\"=Simple <mark>-match1</mark> +Snippet\"}"+
                 ","+
                 "{\"author\":\"+Schiller\","+
                 "\"title\":\"-Title2,\\tthe\","+
                 "\"pubDate\":\"20051104\","+
                 "\"textSigle\":\"RTF/G59/34285\","+
                 "\"snippet\":\"<span class=\\\"context-left\\\"><span class=\\\"more\\\"></span>"+
                 "Simpler, \\\"faster\\\" </span><span class=\\\"match\\\"><mark>&quot;match2&quot;</mark></span>"+
                 "<span class=\\\"context-right\\\"> Snippet"+
                 "<span class=\\\"more\\\"></span></span>\"}"+
                 "]}");
        csv.finish();

        Response resp = csv.serve().build();
        String x = (String) resp.getEntity();
        resp.close();
        String[] lines = x.split("\n");
        assertEquals(lines[0],"HasMoreLeft,leftContext,Match,rightContext,HasMoreRight,isCutted,textSigle,author,pubDate,title");
        assertEquals(lines[1],",\" =Simple\",\" -match1\",\" +Snippet\",,,RTF/G59/34284,Goethe,20051103,\" @Title1\"");
        assertEquals(lines[2],"...,\"Simpler, \"\"faster\"\"\",\" \"\"match2\"\"\",Snippet,...,,RTF/G59/34285,\" +Schiller\",20051104,\" -Title2, the\"");
        assertEquals(lines.length,3);
    };
  
    @Test
    public void testAttributes () throws IOException {
        CsvExporter csv = new CsvExporter();
        csv.setFileName("Beispiel");
        assertEquals(csv.getFileName(),"Beispiel");
        assertEquals(csv.getMimeType(),"text/csv");
        assertEquals(csv.getSuffix(),"csv");
    };
};