DetectIT.java

/*
  Copyright 2016, Google, Inc.

  Licensed 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 com.example.vision;

import static com.google.common.truth.Truth.assertThat;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.PrintStream;

/** Tests for vision "Detect" sample. */
@RunWith(JUnit4.class)
@SuppressWarnings("checkstyle:abbreviationaswordinname")
public class DetectIT {
  private ByteArrayOutputStream bout;
  private PrintStream out;
  private Detect app;
  private static final String PROJECT_ID = System.getenv("GOOGLE_CLOUD_PROJECT");
  private static final String BUCKET = PROJECT_ID;

  @Before
  public void setUp() throws IOException {
    bout = new ByteArrayOutputStream();
    out = new PrintStream(bout);
    System.setOut(out);
    app = new Detect();
  }

  @After
  public void tearDown() {
    System.setOut(null);
  }

  @Test
  public void testFaces() throws Exception {
    // Act
    String[] args = {"faces", "./resources/face_no_surprise.jpg"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("anger: POSSIBLE");
    assertThat(got).contains("joy: POSSIBLE");
    assertThat(got).contains("surprise: LIKELY");
  }

  @Test
  public void testFacesGcs() throws Exception {
    // Act
    String[] args = {"faces", "gs://" + BUCKET + "/vision/face_no_surprise.jpg"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("anger: POSSIBLE");
    assertThat(got).contains("joy: POSSIBLE");
    assertThat(got).contains("surprise: LIKELY");
  }

  @Test
  public void testLabels() throws Exception {
    // Act
    String[] args = {"labels", "./resources/wakeupcat.jpg"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("whiskers");
  }

  @Test
  public void testLabelsGcs() throws Exception {
    // Act
    String[] args = {"labels", "gs://" + BUCKET + "/vision/wakeupcat.jpg"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("whiskers");
  }

  @Test
  public void testLandmarks() throws Exception {
    // Act
    String[] args = {"landmarks", "./resources/landmark.jpg"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("Palace of Fine Arts");
  }

  @Test
  public void testLandmarksGcs() throws Exception {
    // Act
    String[] args = {"landmarks", "gs://" + BUCKET + "/vision/landmark.jpg"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("Palace of Fine Arts");
  }

  @Test
  public void testLandmarksUrl() throws Exception {
    // Act
    String uri = "https://storage-download.googleapis.com/"
        + BUCKET + "/vision/landmark.jpg";
    String[] args = {"landmarks", uri};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("Palace of Fine Arts");
  }

  @Test
  public void testLogos() throws Exception {
    // Act
    String[] args = {"logos", "./resources/logos.png"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("Google");
  }

  @Test
  public void testLogosGcs() throws Exception {
    // Act
    String[] args = {"logos", "gs://" + BUCKET + "/vision/logos.png"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("Google");
  }

  @Test
  public void testText() throws Exception {
    // Act
    String[] args = {"text", "./resources/text.jpg"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("37%");
  }

  @Test
  public void testTextGcs() throws Exception {
    // Act
    String[] args = {"text", "gs://" + BUCKET + "/vision/text.jpg"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("37%");
  }

  @Test
  public void testSafeSearch() throws Exception {
    // Act
    String[] args = {"safe-search", "./resources/wakeupcat.jpg"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("adult: VERY_UNLIKELY");
  }

  @Test
  public void testSafeSearchGcs() throws Exception {
    // Act
    String[] args = {"safe-search", "gs://" + BUCKET + "/vision/wakeupcat.jpg"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("adult: VERY_UNLIKELY");
  }

  @Test
  public void testProperties() throws Exception {
    // Act
    String[] args = {"properties", "./resources/landmark.jpg"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("fraction:");
    assertThat(got).contains("r:");
    assertThat(got).contains("g:");
    assertThat(got).contains("b:");
  }

  @Test
  public void testPropertiesGcs() throws Exception {
    // Act
    String[] args = {"properties", "gs://" + BUCKET + "/vision/landmark.jpg"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("fraction:");
    assertThat(got).contains("r:");
    assertThat(got).contains("g:");
    assertThat(got).contains("b:");
  }

  @Test
  public void detectWebAnnotations() throws Exception {
    // Act
    String[] args = {"web", "./resources/landmark.jpg"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("Palace");
    assertThat(got).contains("Francisco");
  }

  @Test
  public void detectWebAnnotationsGcs() throws Exception {
    // Act
    String[] args = {"web", "gs://" + BUCKET + "/vision/landmark.jpg"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("Palace");
    assertThat(got).contains("Francisco");
  }

  @Test
  public void testCropHints() throws Exception {
    // Act
    String[] args = {"crop", "./resources/wakeupcat.jpg"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("vertices {");
    assertThat(got).contains("x: 599");
    assertThat(got).contains("y: 475");
  }

  @Test
  public void testCropHintsGcs() throws Exception {
    // Act
    String[] args = {"crop", "gs://" + BUCKET + "/vision/wakeupcat.jpg"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("vertices {");
    assertThat(got).contains("x: 599");
    assertThat(got).contains("y: 475");
  }

  @Test
  public void testDocumentText() throws Exception {
    // Act
    String[] args = {"fulltext", "./resources/text.jpg"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("After preparation is complete, the ");
    assertThat(got).contains("37%");
  }

  @Test
  public void testDocumentTextGcs() throws Exception {
    // Act
    String[] args = {"fulltext", "gs://" + BUCKET + "/vision/text.jpg"};
    Detect.argsHelper(args, out);

    // Assert
    String got = bout.toString();
    assertThat(got).contains("After preparation is complete, the ");
    assertThat(got).contains("37%");
  }
}