src/test/java/org/z3950/zing/cql/CQLParserTest.java

   1 /*
   2  * Copyright (c) 1995-2014, Index Datassss
   3  * All rights reserved.
   4  * See the file LICENSE for details.
   5  */
   6 package org.z3950.zing.cql;
   7
   8 import java.io.BufferedInputStream;
   9 import java.io.BufferedReader;
  10 import java.io.File;
  11 import java.io.IOException;
  12 import java.io.InputStream;
  13 import java.io.InputStreamReader;
  14 import java.io.Reader;
  15 import java.io.StringReader;
  16 import java.net.URISyntaxException;
  17 import java.net.URL;
  18 import java.net.URLDecoder;
  19 import java.util.Enumeration;
  20 import java.util.HashSet;
  21 import java.util.Set;
  22 import java.util.jar.JarEntry;
  23 import java.util.jar.JarFile;
  24 import org.junit.After;
  25 import org.junit.AfterClass;
  26 import org.junit.Before;
  27 import org.junit.BeforeClass;
  28 import org.junit.Test;
  29 import static org.junit.Assert.*;
  30
  31 import static java.lang.System.out;
  32 import java.util.Properties;
  33
  34 /**
  35  *
  36  * @author jakub
  37  */
  38 public class CQLParserTest {
  39   public CQLParserTest() {
  40   }
  41
  42   @BeforeClass
  43   public static void setUpClass() {
  44   }
  45
  46   @AfterClass
  47   public static void tearDownClass() {
  48   }
  49
  50   @Before
  51   public void setUp() {
  52   }
  53
  54   @After
  55   public void tearDown() {
  56   }
  57
  58   /**
  59    * Test of main method, of class CQLParser.
  60    */
  61   @Test
  62   public void testRegressionQueries() throws IOException {
  63     System.out.println("Testing the parser using pre-canned regression queries...");
  64     //we might be running the test from within the jar
  65     //list all resource dirs, then traverse them
  66     String[] dirs = getResourceListing(this.getClass(), "regression");
  67     for (String dir : dirs) {
  68       String files[] = getResourceListing(this.getClass(), "regression/" + dir);
  69       for (String file : files) {
  70         if (!file.endsWith(".cql")) continue;
  71         out.println("Parsing "+dir+"/"+file);
  72         InputStream is = this.getClass().getResourceAsStream("/regression/"+dir+"/"+file);
  73         BufferedReader reader = null, reader2 = null;
  74         try {
  75           reader = new BufferedReader(new InputStreamReader(is));
  76           String input = reader.readLine();
  77           out.println("Query: "+input);
  78           CQLParser parser = new CQLParser();
  79           CQLNode parsed = parser.parse(input);
  80           String xcql = parsed.toXCQL();
  81           out.println("Parsed:");
  82           out.println(xcql);
  83           //read the expected xcql output
  84           String prefix = file.substring(0, file.length()-4);
  85           reader2 = new BufferedReader(new InputStreamReader(
  86             this.getClass().getResourceAsStream("/regression/"+dir+"/"+prefix+".xcql")));
  87           StringBuilder sb = new StringBuilder();
  88           String line;
  89           while ((line = reader2.readLine()) != null) {
  90             sb.append(line).append("\n");
  91           }
  92           String expected = sb.toString();
  93           out.println("Expected: ");
  94           out.println(expected);
  95           assertEquals("Assertion failure for "+dir+"/"+file, expected, xcql);
  96         } catch (CQLParseException pe) {
  97           fail("Parsing failed with: "+pe.toString());
  98         } finally {
  99           if (reader != null) reader.close();
 100           if (reader2 != null) reader2.close();
 101         }
 102       }
 103     }
 104   }
 105
 106   @Test
 107   public void testRandomQueries() throws IOException, MissingParameterException {
 108     out.println("Testing the parser using 100 randomly generated queries...");
 109     Properties params = new Properties();
 110     InputStream is = getClass().getResourceAsStream("/generate.properties");
 111     if (is == null)
 112       fail("Cannot locate generate.properties");
 113     params.load(is);
 114     is.close();
 115     CQLGenerator generator = new CQLGenerator(params);
 116     for (int i=0; i<100; i++) {
 117       CQLNode random = generator.generate();
 118       String expected = random.toCQL();
 119       out.println("Generated query: "+expected);
 120       CQLParser parser = new CQLParser();
 121       try {
 122         CQLNode parsed = parser.parse(expected);
 123         String result = parsed.toCQL();
 124         assertEquals(expected, result);
 125       } catch (CQLParseException pe) {
 126         fail("Generated query failed to parse: "+pe.getMessage());
 127       }
 128     }
 129   }
 130
 131
 132   //helper methods follow
 133   //TODO move to masterkey-common
 134
 135   @SuppressWarnings("rawtypes")
 136   public static String[] getResourceListing(Class clazz, String path) throws
 137     IOException {
 138     URL dirURL = clazz.getClassLoader().getResource(path);
 139     if (dirURL != null && dirURL.getProtocol().equals("file")) {
 140       /* A file path: easy enough */
 141       try {
 142         return new File(dirURL.toURI()).list();
 143       } catch (URISyntaxException use) {
 144         throw new UnsupportedOperationException(use);
 145       }
 146     }
 147
 148     if (dirURL == null) {
 149       /*
 150        * In case of a jar file, we can't actually find a directory.
 151        * Have to assume the same jar as clazz.
 152        */
 153       String me = clazz.getName().replace(".", "/") + ".class";
 154       dirURL = clazz.getClassLoader().getResource(me);
 155     }
 156
 157     if (dirURL.getProtocol().equals("jar")) {
 158       /* A JAR path */
 159       String jarPath = dirURL.getPath().substring(5, dirURL.getPath().indexOf(
 160         "!")); //strip out only the JAR file
 161       JarFile jar = new JarFile(URLDecoder.decode(jarPath, "UTF-8"));
 162       Enumeration<JarEntry> entries = jar.entries(); //gives ALL entries in jar
 163       Set<String> result = new HashSet<String>(); //avoid duplicates in case it is a subdirectory
 164       while (entries.hasMoreElements()) {
 165         String name = entries.nextElement().getName();
 166         if (name.startsWith(path)) { //filter according to the path
 167           String entry = name.substring(path.length());
 168           int checkSubdir = entry.indexOf("/");
 169           if (checkSubdir >= 0) {
 170             // if it is a subdirectory, we just return the directory name
 171             entry = entry.substring(0, checkSubdir);
 172           }
 173           result.add(entry);
 174         }
 175       }
 176       return result.toArray(new String[result.size()]);
 177     }
 178
 179     throw new UnsupportedOperationException("Cannot list files for URL "
 180       + dirURL);
 181   }
 182 }