1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
// 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.exec;
import static com.google.common.truth.Truth.assertThat;
import static com.google.devtools.build.lib.actions.util.ActionsTestUtil.NULL_ACTION_OWNER;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;
import com.google.common.collect.Lists;
import com.google.common.collect.Sets;
import com.google.devtools.build.lib.actions.Action;
import com.google.devtools.build.lib.actions.Artifact;
import com.google.devtools.build.lib.actions.MiddlemanAction;
import com.google.devtools.build.lib.actions.MiddlemanFactory;
import com.google.devtools.build.lib.analysis.util.AnalysisTestUtil;
import com.google.devtools.build.lib.analysis.util.BuildViewTestCase;
import com.google.devtools.build.lib.testutil.Suite;
import com.google.devtools.build.lib.testutil.TestSpec;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;
import java.util.ArrayList;
import java.util.Arrays;
/**
* A test for {@link MiddlemanAction}.
*/
@TestSpec(size = Suite.SMALL_TESTS)
@RunWith(JUnit4.class)
public class MiddlemanActionTest extends BuildViewTestCase {
private AnalysisTestUtil.CollectingAnalysisEnvironment analysisEnvironment;
private MiddlemanFactory middlemanFactory;
private Artifact a, b, middle;
@Before
public final void initializeMiddleman() throws Exception {
scratch.file("a/BUILD",
"testing_dummy_rule(name='a', outs=['a.out'])");
scratch.file("b/BUILD",
"testing_dummy_rule(name='b', outs=['b.out'])");
a = getFilesToBuild(getConfiguredTarget("//a")).iterator().next();
b = getFilesToBuild(getConfiguredTarget("//b")).iterator().next();
analysisEnvironment =
new AnalysisTestUtil.CollectingAnalysisEnvironment(
AnalysisTestUtil.STUB_ANALYSIS_ENVIRONMENT);
middlemanFactory = new MiddlemanFactory(view.getArtifactFactory(), analysisEnvironment);
middle = middlemanFactory.createAggregatingMiddleman(
NULL_ACTION_OWNER, "middleman_test",
Arrays.asList(a, b),
targetConfig.getMiddlemanDirectory());
analysisEnvironment.registerWith(getMutableActionGraph());
}
@Test
public void testActionIsAMiddleman() {
Action middleman = getGeneratingAction(middle);
assertTrue("Encountered instance of " + middleman.getClass(),
middleman.getActionType().isMiddleman());
}
@Test
public void testAAndBAreInputsToMiddleman() {
MiddlemanAction middleman = (MiddlemanAction) getGeneratingAction(middle);
assertThat(middleman.getInputs()).containsExactly(a, b);
}
@Test
public void testMiddleIsOutputOfMiddleman() {
MiddlemanAction middleman = (MiddlemanAction) getGeneratingAction(middle);
assertThat(middleman.getOutputs()).containsExactly(middle);
}
@Test
public void testMiddlemanIsNullForEmptyInputs() throws Exception {
assertNull(middlemanFactory.createAggregatingMiddleman(NULL_ACTION_OWNER,
"middleman_test", new ArrayList<Artifact>(), targetConfig.getMiddlemanDirectory()));
}
@Test
public void testMiddlemanIsIdentityForLonelyInput() throws Exception {
assertEquals(a,
middlemanFactory.createAggregatingMiddleman(
NULL_ACTION_OWNER, "middleman_test",
Lists.newArrayList(a),
targetConfig.getMiddlemanDirectory()));
}
@Test
public void testDifferentExecutablesForRunfilesMiddleman() throws Exception {
scratch.file("c/BUILD",
"testing_dummy_rule(name='c', outs=['c.out', 'd.out', 'common.out'])");
Artifact c = getFilesToBuild(getConfiguredTarget("//c:c.out")).iterator().next();
Artifact d = getFilesToBuild(getConfiguredTarget("//c:d.out")).iterator().next();
Artifact common = getFilesToBuild(getConfiguredTarget("//c:common.out")).iterator().next();
analysisEnvironment.clear();
Artifact middlemanForC = middlemanFactory.createRunfilesMiddleman(
NULL_ACTION_OWNER, c, Arrays.asList(c, common), targetConfig.getMiddlemanDirectory());
Artifact middlemanForD = middlemanFactory.createRunfilesMiddleman(
NULL_ACTION_OWNER, d, Arrays.asList(d, common), targetConfig.getMiddlemanDirectory());
analysisEnvironment.registerWith(getMutableActionGraph());
MiddlemanAction middlemanActionForC = (MiddlemanAction) getGeneratingAction(middlemanForC);
MiddlemanAction middlemanActionForD = (MiddlemanAction) getGeneratingAction(middlemanForD);
assertThat(Sets.newHashSet(middlemanActionForD.getInputs()))
.isNotEqualTo(Sets.newHashSet(middlemanActionForC.getInputs()));
assertThat(Sets.newHashSet(middlemanActionForD.getOutputs()))
.isNotEqualTo(Sets.newHashSet(middlemanActionForC.getOutputs()));
}
}
|
