base/tests/unittests/com/google/idea/blaze/base/sync/BlazeSyncManagerTest.java

/*
 * Copyright 2016 The Bazel Authors. All rights reserved.
 *
 * 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.idea.blaze.base.sync;

import static com.google.common.truth.Truth.assertThat;
import static org.mockito.Matchers.any;
import static org.mockito.Mockito.doNothing;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;

import com.google.common.collect.ImmutableList;
import com.google.idea.blaze.base.BlazeTestCase;
import com.google.idea.blaze.base.model.primitives.TargetExpression;
import com.google.idea.blaze.base.settings.BlazeUserSettings;
import com.google.idea.blaze.base.sync.BlazeSyncParams.SyncMode;
import java.io.IOException;
import java.util.List;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;
import org.mockito.ArgumentCaptor;
import org.mockito.Captor;
import org.mockito.MockitoAnnotations;
import org.mockito.Spy;

/** Test cases for {@link BlazeSyncManager}. */
@RunWith(JUnit4.class)
public class BlazeSyncManagerTest extends BlazeTestCase {
  @Spy BlazeSyncManager manager = new BlazeSyncManager(project);
  @Captor ArgumentCaptor<BlazeSyncParams> paramsCaptor;

  @Override
  protected void initTest(Container applicationServices, Container projectServices) {
    super.initTest(applicationServices, projectServices);
    MockitoAnnotations.initMocks(this);
    applicationServices.register(BlazeUserSettings.class, mock(BlazeUserSettings.class));
    doNothing().when(manager).requestProjectSync(any());
    projectServices.register(BlazeSyncManager.class, manager);
    assertThat(BlazeSyncManager.getInstance(project)).isSameAs(manager);
  }

  @Test
  public void testFullProjectSync() throws IOException {
    manager.fullProjectSync();
    verify(manager).requestProjectSync(paramsCaptor.capture());
    BlazeSyncParams params = paramsCaptor.getValue();
    assertThat(params).isNotNull();
    assertThat(params.title).isEqualTo("Full Sync");
    assertThat(params.syncMode).isEqualTo(SyncMode.FULL);
    assertThat(params.backgroundSync).isFalse();
    assertThat(params.addProjectViewTargets).isTrue();
    assertThat(params.addWorkingSet)
        .isEqualTo(BlazeUserSettings.getInstance().getExpandSyncToWorkingSet());
    assertThat(params.targetExpressions).isEmpty();
  }

  @Test
  public void testIncrementalProjectSync() throws IOException {
    manager.incrementalProjectSync();
    verify(manager).requestProjectSync(paramsCaptor.capture());
    BlazeSyncParams params = paramsCaptor.getValue();
    assertThat(params).isNotNull();
    assertThat(params.title).isEqualTo("Sync");
    assertThat(params.syncMode).isEqualTo(SyncMode.INCREMENTAL);
    assertThat(params.backgroundSync).isFalse();
    assertThat(params.addProjectViewTargets).isTrue();
    assertThat(params.addWorkingSet)
        .isEqualTo(BlazeUserSettings.getInstance().getExpandSyncToWorkingSet());
    assertThat(params.targetExpressions).isEmpty();
  }

  @Test
  public void testPartialSync() {
    List<TargetExpression> targets =
        ImmutableList.of(
            TargetExpression.fromString("//foo:bar"), TargetExpression.fromString("//foo:baz"));
    manager.partialSync(targets);
    verify(manager).requestProjectSync(paramsCaptor.capture());
    BlazeSyncParams params = paramsCaptor.getValue();
    assertThat(params).isNotNull();
    assertThat(params.title).isEqualTo("Partial Sync");
    assertThat(params.syncMode).isEqualTo(SyncMode.PARTIAL);
    assertThat(params.backgroundSync).isFalse();
    assertThat(params.addProjectViewTargets).isFalse();
    assertThat(params.addWorkingSet).isFalse();
    assertThat(params.targetExpressions).isEqualTo(targets);
  }

  @Test
  public void testWorkingSetSync() throws IOException {
    manager.workingSetSync();
    verify(manager).requestProjectSync(paramsCaptor.capture());
    BlazeSyncParams params = paramsCaptor.getValue();
    assertThat(params).isNotNull();
    assertThat(params.title).isEqualTo("Sync Working Set");
    assertThat(params.syncMode).isEqualTo(SyncMode.PARTIAL);
    assertThat(params.backgroundSync).isFalse();
    assertThat(params.addProjectViewTargets).isFalse();
    assertThat(params.addWorkingSet).isTrue();
    assertThat(params.targetExpressions).isEmpty();
  }
}