1 /*
2 * Copyright (c) 2008, 2009, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24 /* @test
25 * @bug 4313887
26 * @summary Unit test for probeContentType method
27 * @library ..
28 * @build ContentType SimpleFileTypeDetector
29 * @run main/othervm ContentType
30 */
31
32 import java.nio.file.*;
33 import java.io.*;
34
35 /**
36 * Uses Files.probeContentType to probe html file and custom file type.
37 */
38
39 public class ContentType {
40
41 static Path createHtmlFile() throws IOException {
42 Path file = File.createTempFile("foo", ".html").toPath();
43 OutputStream out = file.newOutputStream();
44 try {
45 out.write("<html><body>foo</body></html>".getBytes());
46 } finally {
47 out.close();
48 }
49
50 return file;
51 }
52
53 static Path createGrapeFile() throws IOException {
54 return File.createTempFile("red", ".grape").toPath();
55 }
56
57 public static void main(String[] args) throws IOException {
58
59 // exercise default file type detector
60 Path file = createHtmlFile();
61 try {
62 String type = Files.probeContentType(file);
63 if (type == null) {
64 System.err.println("Content type cannot be determined - test skipped");
65 } else {
66 if (!type.equals("text/html"))
67 throw new RuntimeException("Unexpected type: " + type);
68 }
69 } finally {
70 file.delete();
71 }
72
73 // exercise custom file type detector
74 file = createGrapeFile();
75 try {
76 String type = Files.probeContentType(file);
77 if (type == null)
78 throw new RuntimeException("Custom file type detector not installed?");
79 if (!type.equals("grape/unknown"))
80 throw new RuntimeException("Unexpected type: " + type);
81 } finally {
82 file.delete();
83 }
84
85 }
86 }