| // Copyright 2015 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.devtools.build.lib.analysis.util; |
| |
| import com.google.common.collect.ImmutableList; |
| import com.google.common.collect.ImmutableMap; |
| import com.google.devtools.build.lib.analysis.BlazeDirectories; |
| import com.google.devtools.build.lib.analysis.ConfiguredRuleClassProvider; |
| import com.google.devtools.build.lib.analysis.config.ConfigurationFragmentFactory; |
| import com.google.devtools.build.lib.bazel.rules.android.AndroidNdkRepositoryFunction; |
| import com.google.devtools.build.lib.bazel.rules.android.AndroidNdkRepositoryRule; |
| import com.google.devtools.build.lib.bazel.rules.android.AndroidSdkRepositoryFunction; |
| import com.google.devtools.build.lib.bazel.rules.android.AndroidSdkRepositoryRule; |
| import com.google.devtools.build.lib.packages.util.LoadingMock; |
| import com.google.devtools.build.lib.packages.util.MockCcSupport; |
| import com.google.devtools.build.lib.packages.util.MockPythonSupport; |
| import com.google.devtools.build.lib.packages.util.MockToolsConfig; |
| import com.google.devtools.build.lib.rules.cpp.CcSkyframeFdoSupportFunction; |
| import com.google.devtools.build.lib.rules.cpp.CcSkyframeFdoSupportValue; |
| import com.google.devtools.build.lib.rules.repository.LocalRepositoryFunction; |
| import com.google.devtools.build.lib.rules.repository.LocalRepositoryRule; |
| import com.google.devtools.build.lib.rules.repository.ManagedDirectoriesKnowledge; |
| import com.google.devtools.build.lib.rules.repository.RepositoryDelegatorFunction; |
| import com.google.devtools.build.lib.rules.repository.RepositoryFunction; |
| import com.google.devtools.build.lib.rules.repository.RepositoryLoaderFunction; |
| import com.google.devtools.build.lib.skyframe.SkyFunctions; |
| import com.google.devtools.build.lib.skyframe.packages.PackageFactoryBuilderWithSkyframeForTesting; |
| import com.google.devtools.build.lib.testutil.TestConstants; |
| import com.google.devtools.build.lib.vfs.Path; |
| import com.google.devtools.build.skyframe.SkyFunction; |
| import com.google.devtools.build.skyframe.SkyFunctionName; |
| import java.io.IOException; |
| import java.lang.reflect.Field; |
| import java.util.List; |
| import java.util.concurrent.atomic.AtomicBoolean; |
| |
| /** Create a mock client for the analysis phase, as well as a configuration factory. */ |
| public abstract class AnalysisMock extends LoadingMock { |
| |
| public static AnalysisMock get() { |
| try { |
| Class<?> providerClass = Class.forName(TestConstants.TEST_ANALYSIS_MOCK); |
| Field instanceField = providerClass.getField("INSTANCE"); |
| return (AnalysisMock) instanceField.get(null); |
| } catch (Exception e) { |
| throw new IllegalStateException(e); |
| } |
| } |
| |
| @Override |
| public String getProductName() { |
| return TestConstants.PRODUCT_NAME; |
| } |
| |
| public ImmutableList<String> getEmbeddedTools() { |
| return TestConstants.EMBEDDED_TOOLS; |
| } |
| |
| @Override |
| public PackageFactoryBuilderWithSkyframeForTesting getPackageFactoryBuilderForTesting( |
| BlazeDirectories directories) { |
| return super.getPackageFactoryBuilderForTesting(directories) |
| .setExtraSkyFunctions(getSkyFunctions(directories)); |
| } |
| |
| /** |
| * This is called from test setup to create the mock directory layout needed to create the |
| * configuration. |
| */ |
| public void setupMockClient(MockToolsConfig mockToolsConfig) throws IOException { |
| List<String> workspaceContents = getWorkspaceContents(mockToolsConfig); |
| setupMockClient(mockToolsConfig, workspaceContents); |
| } |
| |
| public abstract void setupMockClient( |
| MockToolsConfig mockToolsConfig, List<String> getWorkspaceContents) throws IOException; |
| |
| /** |
| * Returns the contents of WORKSPACE. |
| */ |
| public abstract List<String> getWorkspaceContents(MockToolsConfig config); |
| |
| /** |
| * This is called from test setup to create any necessary mock workspace files in the |
| * <code>_embedded_binaries</code> directory. |
| */ |
| public abstract void setupMockWorkspaceFiles(Path embeddedBinariesRoot) throws IOException; |
| |
| /** Creates a mock tools repository. */ |
| public void setupMockToolsRepository(MockToolsConfig config) throws IOException { |
| // Do nothing by default. |
| } |
| |
| /** Returns the default factories for configuration fragments used in tests. */ |
| public abstract List<ConfigurationFragmentFactory> getDefaultConfigurationFragmentFactories(); |
| |
| @Override |
| public abstract ConfiguredRuleClassProvider createRuleClassProvider(); |
| |
| public abstract boolean isThisBazel(); |
| |
| public abstract MockCcSupport ccSupport(); |
| |
| public abstract MockPythonSupport pySupport(); |
| |
| public ImmutableMap<SkyFunctionName, SkyFunction> getSkyFunctions(BlazeDirectories directories) { |
| // Some tests require the local_repository rule so we need the appropriate SkyFunctions. |
| ImmutableMap.Builder<String, RepositoryFunction> repositoryHandlers = |
| new ImmutableMap.Builder<String, RepositoryFunction>() |
| .put(LocalRepositoryRule.NAME, new LocalRepositoryFunction()) |
| .put(AndroidSdkRepositoryRule.NAME, new AndroidSdkRepositoryFunction()) |
| .put(AndroidNdkRepositoryRule.NAME, new AndroidNdkRepositoryFunction()); |
| |
| addExtraRepositoryFunctions(repositoryHandlers); |
| |
| return ImmutableMap.of( |
| SkyFunctions.REPOSITORY_DIRECTORY, |
| new RepositoryDelegatorFunction( |
| repositoryHandlers.build(), |
| null, |
| new AtomicBoolean(true), |
| ImmutableMap::of, |
| directories, |
| ManagedDirectoriesKnowledge.NO_MANAGED_DIRECTORIES), |
| SkyFunctions.REPOSITORY, |
| new RepositoryLoaderFunction(), |
| CcSkyframeFdoSupportValue.SKYFUNCTION, |
| new CcSkyframeFdoSupportFunction(directories)); |
| } |
| |
| // Allow subclasses to add extra repository functions. |
| public abstract void addExtraRepositoryFunctions( |
| ImmutableMap.Builder<String, RepositoryFunction> repositoryHandlers); |
| |
| /** |
| * Stub class for tests to extend in order to update a small amount of {@link AnalysisMock} |
| * functionality. |
| */ |
| public static class Delegate extends AnalysisMock { |
| |
| private final AnalysisMock delegate; |
| |
| public Delegate(AnalysisMock delegate) { |
| this.delegate = delegate; |
| } |
| |
| @Override |
| public void setupMockClient(MockToolsConfig mockToolsConfig, List<String> workspaceContents) |
| throws IOException { |
| delegate.setupMockClient(mockToolsConfig, workspaceContents); |
| } |
| |
| @Override |
| public List<String> getWorkspaceContents(MockToolsConfig mockToolsConfig) { |
| return delegate.getWorkspaceContents(mockToolsConfig); |
| } |
| |
| @Override |
| public void setupMockWorkspaceFiles(Path embeddedBinariesRoot) throws IOException { |
| delegate.setupMockWorkspaceFiles(embeddedBinariesRoot); |
| } |
| |
| @Override |
| public void setupMockToolsRepository(MockToolsConfig config) throws IOException { |
| delegate.setupMockToolsRepository(config); |
| } |
| |
| @Override |
| public List<ConfigurationFragmentFactory> getDefaultConfigurationFragmentFactories() { |
| return delegate.getDefaultConfigurationFragmentFactories(); |
| } |
| |
| @Override |
| public ConfiguredRuleClassProvider createRuleClassProvider() { |
| return delegate.createRuleClassProvider(); |
| } |
| |
| @Override |
| public boolean isThisBazel() { |
| return delegate.isThisBazel(); |
| } |
| |
| @Override |
| public MockCcSupport ccSupport() { |
| return delegate.ccSupport(); |
| } |
| |
| @Override |
| public MockPythonSupport pySupport() { |
| return delegate.pySupport(); |
| } |
| |
| @Override |
| public ImmutableMap<SkyFunctionName, SkyFunction> getSkyFunctions( |
| BlazeDirectories directories) { |
| return delegate.getSkyFunctions(directories); |
| } |
| |
| @Override |
| public void addExtraRepositoryFunctions( |
| ImmutableMap.Builder<String, RepositoryFunction> repositoryHandlers) { |
| delegate.addExtraRepositoryFunctions(repositoryHandlers); |
| } |
| } |
| } |