[JENKINS-20884] Test case contributed in pull-14

jenkinsci · Oct 9, 2014 · 4d63a43 · 4d63a43
1 parent cab3151
commit 4d63a43
Showing 1 changed file with 141 additions and 0 deletions.
diff --git a/src/test/java/hudson/maven/MavenEnvironmentContributingActionFromBuilderTest.java b/src/test/java/hudson/maven/MavenEnvironmentContributingActionFromBuilderTest.java
@@ -0,0 +1,141 @@
+package hudson.maven;
+
+import java.io.IOException;
+import java.util.Collection;
+import java.util.Collections;
+
+import org.junit.Assert;
+import org.junit.Rule;
+import org.junit.Test;
+import org.jvnet.hudson.test.Bug;
+import org.jvnet.hudson.test.ExtractResourceSCM;
+import org.jvnet.hudson.test.JenkinsRule;
+import org.kohsuke.stapler.StaplerRequest;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import hudson.EnvVars;
+import hudson.Extension;
+import hudson.Launcher;
+import hudson.model.AbstractBuild;
+import hudson.model.AbstractProject;
+import hudson.model.Action;
+import hudson.model.BuildListener;
+import hudson.model.Cause;
+import hudson.model.EnvironmentContributingAction;
+import hudson.model.InvisibleAction;
+import hudson.model.Result;
+import hudson.tasks.BuildWrapper;
+import hudson.tasks.BuildWrapperDescriptor;
+import hudson.tasks.Builder;
+import hudson.util.ArgumentListBuilder;
+import net.sf.json.JSONObject;
+
+/**
+ * This test case verifies that a maven build also takes EnvironmentContributingAction into account to resolve variables on the command line
+ * 
+ * @see hudson.model.EnvironmentContributingAction
+ * @author Marcin Cylke (mcl)
+ */
+public class MavenEnvironmentContributingActionFromBuilderTest {
+
+    @Rule
+    public JenkinsRule j = new JenkinsRule();
+
+    @Test
+    @Bug(20844)
+    public void builderInjectedVariableFromEnvironmentContributingActionMustBeAvailableInMavenModuleSetBuild() throws Exception {
+        j.jenkins.getInjector().injectMembers(this);
+
+        final MavenModuleSet p = j.createMavenProject("mvn");
+
+        p.setMaven(j.configureMaven3().getName());
+        p.setScm(new ExtractResourceSCM(getClass().getResource("maven3-project.zip")));
+        p.setGoals("initialize -Dval=${KEY}");
+
+        String keyValue = "MY_VALUE";
+
+        p.getPrebuilders().add(new TestMvnBuilder(keyValue));
+        p.getBuildWrappersList().add(new AssertingBuildWrapper("-Dval=" + keyValue));
+
+        j.assertBuildStatus(Result.SUCCESS, p.scheduleBuild2(0, new Cause.UserIdCause()).get());
+    }
+
+    /**
+     * This action contributes env variables
+     */
+    private static final class TestAction extends InvisibleAction implements EnvironmentContributingAction {
+        private final String key, value;
+
+        public TestAction(String key, String value) {
+            this.key = key;
+            this.value = value;
+        }
+
+        @Override
+        public void buildEnvVars(AbstractBuild<?, ?> arg0, EnvVars vars) {
+            vars.put(key, value);
+        }
+
+    }
+
+    /**
+     * This action verifies that the variable in the maven arguments got replaced
+     */
+    private static class MvnCmdLineVerifier extends InvisibleAction implements MavenArgumentInterceptorAction {
+        private String containsString;
+
+        public MvnCmdLineVerifier(String containsString) {
+            this.containsString = containsString;
+        }
+
+        @Override
+        public ArgumentListBuilder intercept(ArgumentListBuilder cli, MavenModuleSetBuild arg1) {
+            String all = cli.toString();
+            Assert.assertTrue(containsString + " was not found in the goals arguments(" + all + ")",
+                all.contains(containsString));
+            return cli;
+        }
+
+        @Override
+        public String getGoalsAndOptions(MavenModuleSetBuild arg0) {
+            return null;
+        }
+    }
+
+    /**
+     * This wrapper adds MvnCmdLineVerifier to the build to test whether the variable really got replaced
+     */
+    public static class AssertingBuildWrapper extends BuildWrapper {
+        private String containsString;
+
+        public AssertingBuildWrapper(String expectedString) {
+            this.containsString = expectedString;
+        }
+
+        @Override
+        public Environment setUp(AbstractBuild build, Launcher launcher, BuildListener listener) throws IOException, InterruptedException {
+
+            build.addAction(new MvnCmdLineVerifier(containsString));
+
+            return new Environment() { };
+        }
+    }
+
+    /**
+     * This builder returns an action with variable replacement to build process
+     */
+    public static class TestMvnBuilder extends Builder {
+
+        private final String envVariableValue;
+
+        public TestMvnBuilder(String envVariableValue) {
+            this.envVariableValue = envVariableValue;
+        }
+
+        @Override
+        public Collection<? extends Action> getProjectActions(AbstractProject<?, ?> project) {
+            return Collections.singletonList(new TestAction("KEY", envVariableValue));
+        }
+    }
+}