SinkTest.java

/*
 * Copyright 2016 Google LLC
 *
 * 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.google.cloud.logging;

import static org.easymock.EasyMock.createMock;
import static org.easymock.EasyMock.createStrictMock;
import static org.easymock.EasyMock.expect;
import static org.easymock.EasyMock.replay;
import static org.easymock.EasyMock.verify;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;

import com.google.api.core.ApiFutures;
import com.google.cloud.logging.SinkInfo.Destination.BucketDestination;
import com.google.cloud.logging.SinkInfo.Destination.DatasetDestination;
import com.google.cloud.logging.SinkInfo.VersionFormat;

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

import java.util.concurrent.ExecutionException;

public class SinkTest {

  private static final String NAME = "sink";
  private static final String FILTER =
      "logName=projects/my-projectid/logs/syslog AND severity>=ERROR";
  private static final VersionFormat VERSION = VersionFormat.V1;
  private static final String NEW_NAME = "newSink";
  private static final String NEW_FILTER =
      "logName=projects/my-projectid/logs/syslog";
  private static final VersionFormat NEW_VERSION = VersionFormat.V2;
  private static final BucketDestination BUCKET_DESTINATION = BucketDestination.of("bucket");
  private static final DatasetDestination DATASET_DESTINATION = DatasetDestination.of("dataset");
  private static final SinkInfo SINK_INFO = SinkInfo.newBuilder(NAME, BUCKET_DESTINATION)
      .setFilter(FILTER)
      .setVersionFormat(VERSION)
      .build();
  private final Logging serviceMockReturnsOptions = createStrictMock(Logging.class);
  private final LoggingOptions mockOptions = createMock(LoggingOptions.class);
  private Logging logging;
  private Sink expectedSink;
  private Sink sink;

  private void initializeExpectedSink(int optionsCalls) {
    expect(serviceMockReturnsOptions.getOptions()).andReturn(mockOptions).times(optionsCalls);
    replay(serviceMockReturnsOptions);
    logging = createStrictMock(Logging.class);
    expectedSink = new Sink(serviceMockReturnsOptions, new Sink.BuilderImpl(SINK_INFO));
  }

  private void initializeSink() {
    sink = new Sink(logging, new Sink.BuilderImpl(SINK_INFO));
  }

  @After
  public void tearDown() throws Exception {
    verify(logging, serviceMockReturnsOptions);
  }

  @Test
  public void testBuilder() {
    initializeExpectedSink(2);
    replay(logging);
    Sink builtSink = expectedSink.toBuilder()
        .setName(NEW_NAME)
        .setFilter(NEW_FILTER)
        .setDestination(DATASET_DESTINATION)
        .setVersionFormat(NEW_VERSION)
        .build();
    assertEquals(NEW_NAME, builtSink.getName());
    assertEquals(DATASET_DESTINATION, builtSink.getDestination());
    assertEquals(NEW_FILTER, builtSink.getFilter());
    assertEquals(NEW_VERSION, builtSink.getVersionFormat());
  }


  @Test
  public void testToBuilder() {
    initializeExpectedSink(2);
    replay(logging);
    compareSink(expectedSink, expectedSink.toBuilder().build());
  }

  @Test
  public void testReload() {
    initializeExpectedSink(2);
    SinkInfo updatedInfo = SINK_INFO.toBuilder().setFilter(NEW_FILTER).build();
    Sink expectedSink =
        new Sink(serviceMockReturnsOptions, new SinkInfo.BuilderImpl(updatedInfo));
    expect(logging.getOptions()).andReturn(mockOptions);
    expect(logging.getSink(NAME)).andReturn(expectedSink);
    replay(logging);
    initializeSink();
    Sink updatedSink = sink.reload();
    compareSink(expectedSink, updatedSink);
  }

  @Test
  public void testReloadNull() {
    initializeExpectedSink(1);
    expect(logging.getOptions()).andReturn(mockOptions);
    expect(logging.getSink(NAME)).andReturn(null);
    replay(logging);
    initializeSink();
    assertNull(sink.reload());
  }

  @Test
  public void testReloadAsync() throws ExecutionException, InterruptedException {
    initializeExpectedSink(2);
    SinkInfo updatedInfo = SINK_INFO.toBuilder().setFilter(NEW_FILTER).build();
    Sink expectedSink = new Sink(serviceMockReturnsOptions, new SinkInfo.BuilderImpl(updatedInfo));
    expect(logging.getOptions()).andReturn(mockOptions);
    expect(logging.getSinkAsync(NAME))
        .andReturn(ApiFutures.immediateFuture(expectedSink));
    replay(logging);
    initializeSink();
    Sink updatedSink = sink.reloadAsync().get();
    compareSink(expectedSink, updatedSink);
  }

  @Test
  public void testReloadAsyncNull() throws ExecutionException, InterruptedException {
    initializeExpectedSink(1);
    expect(logging.getOptions()).andReturn(mockOptions);
    expect(logging.getSinkAsync(NAME)).andReturn(ApiFutures.<Sink>immediateFuture(null));
    replay(logging);
    initializeSink();
    assertNull(sink.reloadAsync().get());
  }

  @Test
  public void testUpdate() {
    initializeExpectedSink(2);
    SinkInfo updatedInfo = SINK_INFO.toBuilder().setFilter(NEW_FILTER).build();
    Sink expectedSink = new Sink(serviceMockReturnsOptions, new SinkInfo.BuilderImpl(updatedInfo));
    expect(logging.getOptions()).andReturn(mockOptions).times(2);
    expect(logging.update(expectedSink)).andReturn(expectedSink);
    replay(logging);
    initializeSink();
    Sink updatedSink = sink.toBuilder().setFilter(NEW_FILTER).build().update();
    compareSink(expectedSink, updatedSink);
  }

  @Test
  public void testUpdateAsync() throws ExecutionException, InterruptedException {
    initializeExpectedSink(2);
    SinkInfo updatedInfo = SINK_INFO.toBuilder().setFilter(NEW_FILTER).build();
    Sink expectedSink = new Sink(serviceMockReturnsOptions, new SinkInfo.BuilderImpl(updatedInfo));
    expect(logging.getOptions()).andReturn(mockOptions).times(2);
    expect(logging.updateAsync(expectedSink)).andReturn(ApiFutures.immediateFuture(expectedSink));
    replay(logging);
    initializeSink();
    Sink updatedSink = sink.toBuilder().setFilter(NEW_FILTER).build().updateAsync().get();
    compareSink(expectedSink, updatedSink);
  }

  @Test
  public void testDeleteTrue() {
    initializeExpectedSink(1);
    expect(logging.getOptions()).andReturn(mockOptions);
    expect(logging.deleteSink(NAME)).andReturn(true);
    replay(logging);
    initializeSink();
    assertTrue(sink.delete());
  }

  @Test
  public void testDeleteFalse() {
    initializeExpectedSink(1);
    expect(logging.getOptions()).andReturn(mockOptions);
    expect(logging.deleteSink(NAME)).andReturn(false);
    replay(logging);
    initializeSink();
    assertFalse(sink.delete());
  }

  @Test
  public void testDeleteAsyncTrue() throws ExecutionException, InterruptedException {
    initializeExpectedSink(1);
    expect(logging.getOptions()).andReturn(mockOptions);
    expect(logging.deleteSinkAsync(NAME)).andReturn(ApiFutures.immediateFuture(true));
    replay(logging);
    initializeSink();
    assertTrue(sink.deleteAsync().get());
  }

  @Test
  public void testDeleteAsyncFalse() throws ExecutionException, InterruptedException {
    initializeExpectedSink(1);
    expect(logging.getOptions()).andReturn(mockOptions);
    expect(logging.deleteSinkAsync(NAME)).andReturn(ApiFutures.immediateFuture(false));
    replay(logging);
    initializeSink();
    assertFalse(sink.deleteAsync().get());
  }

  private void compareSink(Sink expected, Sink value) {
    assertEquals(expected, value);
    assertEquals(expected.getName(), value.getName());
    assertEquals(expected.getDestination(), value.getDestination());
    assertEquals(expected.getFilter(), value.getFilter());
    assertEquals(expected.getVersionFormat(), value.getVersionFormat());
    assertEquals(expected.hashCode(), value.hashCode());
    assertEquals(expected.toString(), value.toString());
  }
}