test/com/facebook/buck/android/NdkLibraryTest.java

/*
 * Copyright 2012-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.android;

import static com.facebook.buck.rules.BuildableProperties.Kind.ANDROID;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import com.facebook.buck.io.ProjectFilesystem;
import com.facebook.buck.model.BuildTargetFactory;
import com.facebook.buck.rules.BuildContext;
import com.facebook.buck.rules.BuildRuleResolver;
import com.facebook.buck.rules.FakeBuildableContext;
import com.facebook.buck.rules.SourcePathResolver;
import com.facebook.buck.step.ExecutionContext;
import com.facebook.buck.step.Step;
import com.facebook.buck.step.TestExecutionContext;
import com.facebook.buck.testutil.MoreAsserts;
import com.facebook.buck.util.BuckConstant;
import com.facebook.buck.util.DefaultPropertyFinder;
import com.google.common.base.Optional;
import com.google.common.base.Suppliers;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;

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

import java.io.File;
import java.io.IOException;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.List;

/**
 * Unit test for {@link NdkLibrary}.
 */
public class NdkLibraryTest {

  private ExecutionContext executionContext;
  private String ndkBuildCommand;

  @Before
  public void setUp() {
    AssumeAndroidPlatform.assumeNdkIsAvailable();
    ProjectFilesystem projectFilesystem = new ProjectFilesystem(Paths.get("."));
    AndroidDirectoryResolver resolver = new DefaultAndroidDirectoryResolver(projectFilesystem,
        Optional.<String>absent(),
        new DefaultPropertyFinder(projectFilesystem, ImmutableMap.copyOf(System.getenv())));

    AndroidPlatformTarget androidPlatformTarget = AndroidPlatformTarget.getDefaultPlatformTarget(
        resolver,
        Optional.<Path>absent());
    executionContext = TestExecutionContext.newBuilder()
        .setAndroidPlatformTargetSupplier(Suppliers.ofInstance(androidPlatformTarget))
        .build();
    ndkBuildCommand = executionContext.resolveExecutable(
        resolver.findAndroidNdkDir().get(),
        "ndk-build").get().toAbsolutePath().toString();
  }

  @Test
  public void testSimpleNdkLibraryRule() throws IOException {
    BuildRuleResolver ruleResolver = new BuildRuleResolver();
    SourcePathResolver pathResolver = new SourcePathResolver(ruleResolver);
    BuildContext context = null;

    String basePath = "java/src/com/facebook/base";
    NdkLibrary ndkLibrary =
        NdkLibraryBuilder.createNdkLibrary(
            BuildTargetFactory.newInstance(
                String.format("//%s:base", basePath)),
            pathResolver)
            .setNdkVersion("r8b")
            .addSrc(Paths.get(basePath + "/Application.mk"))
            .addSrc(Paths.get(basePath + "/main.cpp"))
            .addSrc(Paths.get(basePath + "/Android.mk"))
            .addFlag("flag1")
            .addFlag("flag2")
            .setIsAsset(true)
            .build();

    ruleResolver.addToIndex(ndkLibrary);

    assertEquals(NdkLibraryDescription.TYPE, ndkLibrary.getType());

    assertTrue(ndkLibrary.getProperties().is(ANDROID));
    assertTrue(ndkLibrary.isAsset());
    assertEquals(Paths.get(BuckConstant.GEN_DIR, basePath, "__libbase"),
        ndkLibrary.getLibraryPath());

    MoreAsserts.assertListEquals(
        ImmutableList.of(
            Paths.get(basePath + "/Android.mk"),
            Paths.get(basePath + "/Application.mk"),
            Paths.get(basePath + "/main.cpp")),
        ImmutableList.copyOf(ndkLibrary.getInputsToCompareToOutput()));

    List<Step> steps = ndkLibrary.getBuildSteps(context, new FakeBuildableContext());

    String libbase = Paths.get(BuckConstant.BIN_DIR, basePath, "__libbase").toString();
    MoreAsserts.assertShellCommands(
        "ndk_library() should invoke ndk-build on the given path with some -j value",
        ImmutableList.of(
            String.format(
                "%s -j %d -C %s flag1 flag2 " +
                    "APP_PROJECT_PATH=%s " +
                    "APP_BUILD_SCRIPT=%s " +
                    "NDK_OUT=%s " +
                    "NDK_LIBS_OUT=%s " +
                    "BUCK_PROJECT_DIR=. " +
                    "host-echo-build-step=@# " +
                    "--silent",
                ndkBuildCommand,
                Runtime.getRuntime().availableProcessors(),
                Paths.get(basePath).toString(),
                /* APP_PROJECT_PATH */ libbase + File.separator,
                /* APP_BUILD_SCRIPT */ Paths.get(basePath, "Android.mk"),
                /* NDK_OUT */ libbase + File.separator,
                /* NDK_LIBS_OUT */ Paths.get(libbase, "libs"),
                Paths.get(libbase, "libs").toString())
        ),
        steps.subList(0, 1),
        executionContext);
  }
}