1 /*
   2  * Copyright (c) 2008, 2011, 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 Basic SimpleFileTypeDetector
  29  * @run main/othervm Basic
  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 Basic {
  40 



  41     static Path createHtmlFile() throws IOException {
  42         Path file = Files.createTempFile("foo", ".html");
  43         try (OutputStream out = Files.newOutputStream(file)) {
  44             out.write("<html><body>foo</body></html>".getBytes());
  45         }
  46 
  47         return file;
  48     }
  49 
  50     static Path createGrapeFile() throws IOException {
  51         return Files.createTempFile("red", ".grape");
  52     }
  53 


















































  54     public static void main(String[] args) throws IOException {

  55 
  56         // exercise default file type detector
  57         Path file = createHtmlFile();
  58         try {
  59             String type = Files.probeContentType(file);
  60             if (type == null) {
  61                 System.err.println("Content type cannot be determined - test skipped");
  62             } else {
  63                 if (!type.equals("text/html"))
  64                     throw new RuntimeException("Unexpected type: " + type);
  65             }
  66         } finally {
  67             Files.delete(file);
  68         }
  69 
  70         // exercise custom file type detector
  71         file = createGrapeFile();
  72         try {
  73             String type = Files.probeContentType(file);
  74             if (type == null)
  75                 throw new RuntimeException("Custom file type detector not installed?");
  76             if (!type.equals("grape/unknown"))
  77                 throw new RuntimeException("Unexpected type: " + type);
  78         } finally {
  79             Files.delete(file);
  80         }
  81 



  82     }
  83 }
--- EOF ---