lang/java/avro/src/test/java/org/apache/avro/util/TestCaseFinder.java

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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 org.apache.avro.util;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.io.BufferedReader;
import java.io.StringReader;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import org.junit.runners.Parameterized.Parameters;
import org.junit.experimental.runners.Enclosed;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

@RunWith(Enclosed.class)
public class TestCaseFinder {

  @RunWith(Parameterized.class)
  public static class SimpleCases {
    String input, label;
    List<Object[]> expectedOutput;

    public SimpleCases(String input, String label, Object[][] ex) {
      this.input = input;
      this.label = label;
      this.expectedOutput = Arrays.asList(ex);
    }

    @Parameters
    public static List<Object[]> cases() {
      List<Object[]> result = new ArrayList<Object[]>();
      result.add(new Object[] { "", "foo", new Object[][] { } });
      result.add(new Object[] { "<<INPUT a\n<<OUTPUT b", "OUTPUT",
                                new Object[][] { {"a","b"} } });
      result.add(new Object[] { "<<INPUT a\n<<OUTPUT b\n", "OUTPUT",
                                new Object[][] { {"a","b"} } });
      result.add(new Object[] { "<<INPUT a\n<<OUTPUT b\n\n", "OUTPUT",
                                new Object[][] { {"a","b"} } });
      result.add(new Object[] { "<<INPUT a\r<<OUTPUT b", "OUTPUT",
                                new Object[][] { {"a","b"} } });
      result.add(new Object[] { "// This is a test\n<<INPUT a\n\n\n<<OUTPUT b",
                                "OUTPUT",
                                new Object[][] { {"a","b"} } });
      result.add(new Object[] { "<<INPUT a\n<<OUTPUT\nb\nOUTPUT", "OUTPUT",
                                new Object[][] { {"a","b"} } });
      result.add(new Object[] { "<<INPUT a\n<<OUTPUT\nb\nOUTPUT", "OUTPUT",
                                new Object[][] { {"a","b"} } });
      result.add(new Object[] { "<<INPUT a\n<<OUTPUT\nb\n\nOUTPUT", "OUTPUT",
                                new Object[][] { {"a","b\n"} } });
      result.add(new Object[] { "<<INPUT a\n<<OUTPUT\n\n  b  \n\nOUTPUT",
                                "OUTPUT",
                                new Object[][] { {"a","\n  b  \n"} } });
      result.add(new Object[] { "<<INPUT a\n<<O b\n<<INPUT c\n<<O d", "O",
                                new Object[][] { {"a","b"}, {"c","d"} } });
      result.add(new Object[] { "<<INPUT a\n<<O b\n<<F z\n<<INPUT c\n<<O d",
                                "O",
                                new Object[][] { {"a","b"}, {"c","d"} } });
      result.add(new Object[] { "<<INPUT a\n<<O b\n<<F z\n<<INPUT c\n<<O d",
                                "F",
                                new Object[][] { {"a","z"} } });
      result.add(new Object[] { "<<INPUT a\n<<O b\n<<F z\n<<INPUT\nc\nINPUT\n<<O d\n<<INPUT e",
                                "INPUT",
                                new Object[][] { {"a",null}, {"c",null}, {"e", null} } });
      return result;
    }

    @Test public void testOutput() throws Exception {
      List<Object[]> result = new ArrayList<Object[]>();
      CaseFinder.find(mk(input), label, result);
      assertTrue(pr(result), eq(result, expectedOutput));
    }
  }

  public static class NonParameterized {
    @Test (expected=java.lang.IllegalArgumentException.class)
    public void testBadDocLabel1() throws Exception {
      List<Object[]> result = new ArrayList<Object[]>();
      CaseFinder.find(mk("<<INPUT blah"), "", result);
    }

    public void testBadDocLabel2() throws Exception {
      List<Object[]> result = new ArrayList<Object[]>();
      CaseFinder.find(mk("<<INPUT blah"), "kill-er", result);
    }

    @Test (expected=java.io.IOException.class)
    public void testBadSingleLineHeredoc() throws Exception {
      List<Object[]> result = new ArrayList<Object[]>();
      CaseFinder.find(mk("<<INPUTblah"), "foo", result);
    }

    @Test (expected=java.io.IOException.class)
    public void testUnterminatedHeredoc() throws Exception {
      List<Object[]> result = new ArrayList<Object[]>();
      CaseFinder.find(mk("<<INPUT"), "foo", result);
    }
  }

  private static BufferedReader mk(String s)
  { return new BufferedReader(new StringReader(s)); }

  private static String pr(List<Object[]> t) {
    StringBuilder b = new StringBuilder();
    b.append("{ ");
    boolean firstTime = true;
    for (Object[] p: t) {
      if (! firstTime) b.append(", "); else firstTime = false;
      b.append("{ \"").append(p[0]).append("\", \"")
        .append(p[1]).append("\" }");
    }
    b.append("}");
    return b.toString();
  }

  private static boolean eq(List<Object[]> l1, List<Object[]> l2) {
    if (l1 == null || l2 == null) return l1 == l2;
    if (l1.size() != l2.size()) return false;
    for (int i = 0; i < l1.size(); i++)
      if (! Arrays.equals(l1.get(i), l2.get(i))) return false;
    return true;
  }
}