CloudMonitoringAuthSampleTest.java

/*
 * Copyright 2015 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.
 */

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

import org.junit.After;
import org.junit.Before;
import org.junit.Test;

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

/**
 * Tests the Cloud Monitoring auth sample.
 */
public class CloudMonitoringAuthSampleTest {
  private final ByteArrayOutputStream stdout =
      new ByteArrayOutputStream();
  private final ByteArrayOutputStream stderr =
      new ByteArrayOutputStream();
  private static final PrintStream REAL_OUT = System.out;
  private static final PrintStream REAL_ERR = System.err;
  private static final String PROJECT_ID = System.getenv("GOOGLE_CLOUD_PROJECT");

  @Before
  public void setUp() {
    System.setOut(new PrintStream(stdout));
    System.setErr(new PrintStream(stderr));
  }

  @After
  public void tearDown() {
    System.setOut(this.REAL_OUT);
    System.setErr(this.REAL_ERR);
  }

  @Test
  public void testUsage() throws Exception {
    CloudMonitoringAuthSample.main(new String[] { });
    assertThat(stderr.toString())
        .named("stderr")
        .isEqualTo("Usage: CloudMonitoringAuthSample <project-name>\n");
  }

  @Test
  public void testListTimeSeries() throws Exception {
    CloudMonitoringAuthSample.main(new String[] { PROJECT_ID });
    String out = stdout.toString();
    assertThat(out).named("stdout").contains("Timeseries.list raw response:");
    assertThat(out)
        .named("stdout")
        .containsMatch("\\{\\s*\"kind\" *: *\"cloudmonitoring#listTimeseriesResponse\",");
    assertThat(out).named("stdout").containsMatch(".*oldest.*");
  }
}