test/com/facebook/buck/cli/ProjectCommandIntellijTest.java

/*
 * Copyright 2015-present Facebook, 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.facebook.buck.cli;

import static org.junit.Assert.assertEquals;

import com.facebook.buck.java.JavaBinaryRuleBuilder;
import com.facebook.buck.java.JavaLibraryBuilder;
import com.facebook.buck.java.JavaTestBuilder;
import com.facebook.buck.model.BuildTarget;
import com.facebook.buck.rules.ProjectConfigBuilder;
import com.facebook.buck.rules.TargetGraph;
import com.facebook.buck.rules.TargetGraphAndTargets;
import com.facebook.buck.rules.TargetNode;
import com.facebook.buck.testutil.TargetGraphFactory;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.ImmutableSortedSet;

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

public class ProjectCommandIntellijTest {

  private TargetNode<?> barLibNode;
  private TargetNode<?> fooLibNode;
  private TargetNode<?> fooBinNode;
  private TargetNode<?> bazLibNode;
  private TargetNode<?> bazTestNode;
  private TargetNode<?> fooTestNode;
  private TargetNode<?> fooBinTestNode;
  private TargetNode<?> quxBinNode;
  private TargetNode<?> fooProjectNode;
  private TargetNode<?> bazProjectNode;

  TargetGraph targetGraph;

  @Before
  public void buildGraph() {
    // Create the following dep tree:
    //
    // FooBin -has-test-> FooBinTest
    // |
    // V
    // FooLib -has-test-> FooLibTest
    // |                  |
    // V                  V
    // BarLib             BazLib -has-test-> BazLibTest
    // ^
    // |
    // QuxBin

    BuildTarget barLibTarget = BuildTarget.builder("//bar", "lib").build();
    barLibNode = JavaLibraryBuilder
        .createBuilder(barLibTarget)
        .build();

    BuildTarget fooLibTarget = BuildTarget.builder("//foo", "lib").build();
    fooLibNode = JavaLibraryBuilder
        .createBuilder(fooLibTarget)
        .addDep(barLibTarget)
        .build();

    BuildTarget fooBinTarget = BuildTarget.builder("//foo", "binbinary").build();
    fooBinNode = new JavaBinaryRuleBuilder(fooBinTarget)
        .setDeps(ImmutableSortedSet.of(fooLibTarget))
        .build();

    BuildTarget bazLibTarget = BuildTarget.builder("//baz", "lib").build();
    bazLibNode = JavaLibraryBuilder
        .createBuilder(bazLibTarget)
        .build();

    BuildTarget bazTestTarget = BuildTarget.builder("//baz", "xctest").build();
    bazTestNode = JavaTestBuilder
        .createBuilder(bazTestTarget)
        .setSourceUnderTest(ImmutableSortedSet.of(bazLibTarget))
        .build();

    BuildTarget fooTestTarget = BuildTarget.builder("//foo", "lib-xctest").build();
    fooTestNode = JavaTestBuilder
        .createBuilder(fooTestTarget)
        .setSourceUnderTest(ImmutableSortedSet.of(fooLibTarget))
        .addDep(bazLibTarget)
        .build();

    BuildTarget fooBinTestTarget = BuildTarget.builder("//foo", "bin-xctest").build();
    fooBinTestNode = JavaTestBuilder
        .createBuilder(fooBinTestTarget)
        .setSourceUnderTest(ImmutableSortedSet.of(fooBinTarget))
        .build();

    BuildTarget quxBinTarget = BuildTarget.builder("//qux", "bin").build();
    quxBinNode = new JavaBinaryRuleBuilder(quxBinTarget)
        .setDeps(ImmutableSortedSet.of(barLibTarget))
        .build();

    BuildTarget fooProjectTarget = BuildTarget.builder("//foo", "foo").build();
    fooProjectNode = ProjectConfigBuilder
        .newProjectConfigRuleBuilder(fooProjectTarget)
        .setSrcRule(fooBinTarget)
        .build();

    BuildTarget bazProjectTarget = BuildTarget.builder("//baz", "baz").build();
    bazProjectNode = ProjectConfigBuilder
        .newProjectConfigRuleBuilder(bazProjectTarget)
        .setSrcRule(bazLibTarget)
        .build();

    targetGraph = TargetGraphFactory.newInstance(
        barLibNode,
        fooLibNode,
        fooBinNode,
        bazLibNode,
        bazTestNode,
        fooTestNode,
        fooBinTestNode,
        quxBinNode,
        fooProjectNode,
        bazProjectNode);
  }

  @Test
  public void testCreateTargetGraphWithoutTests() {
    TargetGraphAndTargets targetGraphAndTargets = ProjectCommandTests.createTargetGraph(
        targetGraph,
        ProjectCommandOptions.Ide.INTELLIJ,
        ImmutableSet.<BuildTarget>of(),
        /* withTests = */ false);

    assertEquals(
        ImmutableSortedSet.<TargetNode<?>>of(
            fooProjectNode,
            fooBinNode,
            fooLibNode,
            barLibNode,
            bazProjectNode,
            bazLibNode),
        ImmutableSortedSet.copyOf(
            targetGraphAndTargets.getTargetGraph().getNodes()));
  }

  @Test
  public void testCreateTargetGraphWithTests() {
    TargetGraphAndTargets targetGraphAndTargets = ProjectCommandTests.createTargetGraph(
        targetGraph,
        ProjectCommandOptions.Ide.INTELLIJ,
        ImmutableSet.<BuildTarget>of(),
        /* withTests = */ true);

    assertEquals(
        ImmutableSortedSet.<TargetNode<?>>of(
            fooProjectNode,
            fooBinNode,
            fooLibNode,
            fooBinTestNode,
            fooTestNode,
            barLibNode,
            bazProjectNode,
            bazLibNode,
            bazTestNode),
        ImmutableSortedSet.copyOf(
            targetGraphAndTargets.getTargetGraph().getNodes()));
  }

  @Test
  public void testCreateTargetGraphForSliceWithoutTests() {
    TargetGraphAndTargets targetGraphAndTargets = ProjectCommandTests.createTargetGraph(
        targetGraph,
        ProjectCommandOptions.Ide.INTELLIJ,
        ImmutableSet.of(fooProjectNode.getBuildTarget()),
        /* withTests = */ false);

    assertEquals(
        ImmutableSortedSet.<TargetNode<?>>of(
            fooProjectNode,
            fooBinNode,
            fooLibNode,
            barLibNode),
        ImmutableSortedSet.copyOf(
            targetGraphAndTargets.getTargetGraph().getNodes()));
  }

  @Test
  public void testCreateTargetGraphForSliceWithTests() {
    TargetGraphAndTargets targetGraphAndTargets = ProjectCommandTests.createTargetGraph(
        targetGraph,
        ProjectCommandOptions.Ide.INTELLIJ,
        ImmutableSet.of(fooProjectNode.getBuildTarget()),
        /* withTests = */ true);

    assertEquals(
        ImmutableSortedSet.<TargetNode<?>>of(
            fooProjectNode,
            fooBinNode,
            fooLibNode,
            fooBinTestNode,
            fooTestNode,
            barLibNode,
            bazProjectNode,
            bazLibNode),
        ImmutableSortedSet.copyOf(
            targetGraphAndTargets.getTargetGraph().getNodes()));
  }

  @Test
  public void testCreateTargetGraphForSmallSliceWithoutTests() {
    TargetGraphAndTargets targetGraphAndTargets = ProjectCommandTests.createTargetGraph(
        targetGraph,
        ProjectCommandOptions.Ide.INTELLIJ,
        ImmutableSet.of(bazProjectNode.getBuildTarget()),
        /* withTests = */ false);

    assertEquals(
        ImmutableSortedSet.<TargetNode<?>>of(
            bazProjectNode,
            bazLibNode),
        ImmutableSortedSet.copyOf(
            targetGraphAndTargets.getTargetGraph().getNodes()));
  }

  @Test
  public void testCreateTargetGraphForSmallSliceWithTests() {
    TargetGraphAndTargets targetGraphAndTargets = ProjectCommandTests.createTargetGraph(
        targetGraph,
        ProjectCommandOptions.Ide.INTELLIJ,
        ImmutableSet.of(bazProjectNode.getBuildTarget()),
        /* withTests = */ true);

    assertEquals(
        ImmutableSortedSet.<TargetNode<?>>of(
            bazProjectNode,
            bazLibNode,
            bazTestNode),
        ImmutableSortedSet.copyOf(
            targetGraphAndTargets.getTargetGraph().getNodes()));
  }
}