| // Copyright (C) 2020 The Android Open Source Project |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| package com.googlesource.gerrit.plugins.uploadvalidator; |
| |
| import static com.google.common.truth.Truth.assertThat; |
| |
| import java.nio.charset.StandardCharsets; |
| import java.util.ArrayList; |
| import java.util.List; |
| import org.junit.Before; |
| import org.junit.Test; |
| |
| public class MimeTypeDetectionTest extends ValidatorTestCase { |
| private class FileContent { |
| public FileContent(String fileName, byte[] content, String contentType) { |
| this.fileName = fileName; |
| this.content = content; |
| this.contentType = contentType; |
| } |
| |
| public String fileName; |
| public byte[] content; |
| public String contentType; |
| } |
| |
| private static final byte[] TEST_PDF = |
| ("%PDF-1.4\n" |
| + "1 0 obj << /Type /Catalog /Outlines 2 0 R /Pages 3 0 R >>\n" |
| + "endobj 2 0 obj << /Type /Outlines /Count 0 >>\n" |
| + "endobj 3 0 obj << /Type /Pages /Kids [4 0 R] /Count 1\n" |
| + ">> endobj 4 0 obj << /Type /Page /Parent 3 0 R\n" |
| + "/MediaBox [0 0 612 144] /Contents 5 0 R /Resources << /ProcSet 6 0 R\n" |
| + "/Font << /F1 7 0 R >> >> >> endobj 5 0 obj\n" |
| + "<< /Length 73 >> stream BT\n" |
| + "/F1 24 Tf\n" |
| + "100 100 Td\n" |
| + "(Small pdf) Tj\n" |
| + "ET endstream endobj 6 0 obj [/PDF /Text] endobj 7 0 obj\n" |
| + "<< /Type /Font /Subtype /Type1 /Name /F1 /BaseFont /Helvetica\n" |
| + "/Encoding /MacRomanEncoding >> endobj xref 0 8\n" |
| + "0000000000 65535 f 0000000009 00000 n 0000000074 00000 n\n" |
| + "0000000120 00000 n 0000000179 00000 n 0000000364 00000 n\n" |
| + "0000000466 00000 n 0000000496 00000 n\n" |
| + "trailer << /Size 8 /Root 1 0 R >> startxref 625\n" |
| + "%%EOF") |
| .getBytes(StandardCharsets.UTF_8); |
| |
| private MimeTypeDetection detection; |
| |
| @Before |
| public void setUp() { |
| detection = new MimeTypeDetection(); |
| } |
| |
| @Test |
| public void testMimeTypeDetection() throws Exception { |
| List<FileContent> files = createFiles(); |
| for (FileContent file : files) { |
| assertThat(detection.getMimeType(file.fileName, file.content).toString()) |
| .isEqualTo(file.contentType); |
| } |
| } |
| |
| private List<FileContent> createFiles() { |
| List<FileContent> files = new ArrayList<>(); |
| |
| String content = "<?xml version=\"1.0\"?><a><b>c</b></a>"; |
| files.add( |
| new FileContent("foo.xml", content.getBytes(StandardCharsets.UTF_8), "application/xml")); |
| |
| content = "<html><body><h1>Hello World!</h1></body></html>"; |
| files.add(new FileContent("foo.html", content.getBytes(StandardCharsets.UTF_8), "text/html")); |
| |
| content = "Hello,World"; |
| files.add(new FileContent("foo.csv", content.getBytes(StandardCharsets.UTF_8), "text/csv")); |
| |
| content = "hello=world"; |
| files.add(new FileContent("foo", content.getBytes(StandardCharsets.UTF_8), "text/plain")); |
| |
| files.add(new FileContent("foo.pdf", TEST_PDF, "application/pdf")); |
| return files; |
| } |
| } |