joshua-core/src/test/java/org/apache/joshua/decoder/io/TranslationRequestStreamTest.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.joshua.decoder.io;

import org.apache.joshua.decoder.Decoder;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import com.typesafe.config.Config;

import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.nio.charset.Charset;

import static org.mockito.Mockito.mock;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.fail;

/**
 * This class verifies the following behaviors:
 * <p>
 * - A blank input, i.e. "", does not cause a translation to be created.
 * <p>
 * - A non-blank input that is not followed by a newline, e.g. "1", causes a translation to be
 * created.
 * <p>
 * - An input that contains whitespace or nothing followed by a newline causes a translation to be
 * created, with "" as the source.
 */

public class TranslationRequestStreamTest {

  private static final Config CONFIG = Decoder.getDefaultFlags();

  @BeforeMethod
  public void createTranslationRequest() throws Exception {
  }

  /**
   * @throws java.lang.Exception
   */
  @BeforeMethod
  protected void setUp() {
  }

  /**
   * @throws java.lang.Exception
   */
  @AfterMethod
  protected void tearDown() throws Exception {
  }

  /**
   * Test method for {@link org.apache.joshua.decoder.io.TranslationRequestStream#TranslationRequestStream(BufferedReader, JoshuaConfiguration)}.
   */
  @Test(enabled = false)
  public void testTranslationRequest() {
    fail("Not yet implemented");
  }

  /**
   * Test method for {@link org.apache.joshua.decoder.io.TranslationRequestStream#size()}.
   */
  @Test
  public void testSize_uponConstruction() {
    InputStream in = mock(InputStream.class);
    TranslationRequestStream request = new TranslationRequestStream(
            new BufferedReader(new InputStreamReader(in, Charset.defaultCharset())), CONFIG);
    assertEquals(request.size(), 0);
  }

  /**
   * Test method for {@link org.apache.joshua.decoder.io.TranslationRequestStream#size()}.
   *
   * @throws Exception
   */
  @Test
  public void testSize_1() throws Exception {
    byte[] data = "1".getBytes();
    ByteArrayInputStream input = new ByteArrayInputStream(data);
    TranslationRequestStream request = new TranslationRequestStream(
            new BufferedReader(new InputStreamReader(input, Charset.defaultCharset())), CONFIG);
    request.next();
    assertEquals(request.size(), 1);
  }

  /**
   * Test method for {@link org.apache.joshua.decoder.io.TranslationRequestStream##size()}.
   *
   * @throws Exception
   */
  @Test
  public void testSize_newline() throws Exception {
    byte[] data = "\n".getBytes();
    ByteArrayInputStream input = new ByteArrayInputStream(data);
    TranslationRequestStream request = new TranslationRequestStream(
            new BufferedReader(new InputStreamReader(input, Charset.defaultCharset())), CONFIG);
    request.next();
    assertEquals(request.size(), 1);
  }

  /**
   * Test method for {@link org.apache.joshua.decoder.io.TranslationRequestStream##size()}.
   *
   * @throws Exception
   */
  @Test
  public void testSize_2newlines() throws Exception {
    byte[] data = "\n\n".getBytes();
    ByteArrayInputStream input = new ByteArrayInputStream(data);
    TranslationRequestStream request = new TranslationRequestStream(
            new BufferedReader(new InputStreamReader(input, Charset.defaultCharset())), CONFIG);
    request.next();
    request.next();
    assertEquals(request.size(), 2);
  }

  /**
   * Test method for {@link org.apache.joshua.decoder.io.TranslationRequestStream##next()}.
   *
   * @throws Exception
   */
  @Test
  public void testNext_2Newlines() throws Exception {
    byte[] data = "\n\n".getBytes();
    ByteArrayInputStream input = new ByteArrayInputStream(data);
    TranslationRequestStream request = new TranslationRequestStream(
            new BufferedReader(new InputStreamReader(input, Charset.defaultCharset())), CONFIG);
    assertEquals(request.next().source(), "");
    assertEquals(request.next().source(), "");
  }

  /**
   * Test method for {@link org.apache.joshua.decoder.io.TranslationRequestStream##remove()}.
   */
  @Test(enabled = false)
  public void testRemove() {
    fail("Not yet implemented");
  }

}