re-implement hide of nuget api key in build logs, add more tests for it
[teamcity/dotNetPackagesSupport.git] / nuget-tests / src / jetbrains / buildServer / nuget / tests / agent / NuGetUpdatePackageActionFactoryTest.java
1 /*\r
2  * Copyright 2000-2011 JetBrains s.r.o.\r
3  *\r
4  * Licensed under the Apache License, Version 2.0 (the "License");\r
5  * you may not use this file except in compliance with the License.\r
6  * You may obtain a copy of the License at\r
7  *\r
8  * http://www.apache.org/licenses/LICENSE-2.0\r
9  *\r
10  * Unless required by applicable law or agreed to in writing, software\r
11  * distributed under the License is distributed on an "AS IS" BASIS,\r
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
13  * See the License for the specific language governing permissions and\r
14  * limitations under the License.\r
15  */\r
16 \r
17 package jetbrains.buildServer.nuget.tests.agent;\r
18 \r
19 import jetbrains.buildServer.BaseTestCase;\r
20 import jetbrains.buildServer.RunBuildException;\r
21 import jetbrains.buildServer.agent.BuildParametersMap;\r
22 import jetbrains.buildServer.agent.BuildRunnerContext;\r
23 import jetbrains.buildServer.nuget.agent.commands.impl.CommandFactoryImpl;\r
24 import jetbrains.buildServer.nuget.agent.commands.impl.NuGetActionFactoryImpl;\r
25 import jetbrains.buildServer.nuget.agent.dependencies.PackageUsages;\r
26 import jetbrains.buildServer.nuget.agent.parameters.NuGetFetchParameters;\r
27 import jetbrains.buildServer.nuget.agent.parameters.PackagesUpdateParameters;\r
28 import jetbrains.buildServer.nuget.agent.util.CommandlineBuildProcessFactory;\r
29 import org.jmock.Expectations;\r
30 import org.jmock.Mockery;\r
31 import org.testng.annotations.BeforeMethod;\r
32 import org.testng.annotations.Test;\r
33 \r
34 import java.io.File;\r
35 import java.io.IOException;\r
36 import java.util.Arrays;\r
37 import java.util.Collections;\r
38 \r
39 /**\r
40  * Created by Eugene Petrenko (eugene.petrenko@gmail.com)\r
41  * Date: 10.07.11 14:29\r
42  */\r
43 public class NuGetUpdatePackageActionFactoryTest extends BaseTestCase {\r
44   private Mockery m;\r
45   private CommandlineBuildProcessFactory myProcessFactory;\r
46   private NuGetActionFactoryImpl i;\r
47   private BuildRunnerContext ctx;\r
48   private NuGetFetchParameters nugetParams;\r
49   private PackagesUpdateParameters ps;\r
50   private File myTarget;\r
51   private File myConfig;\r
52   private String cmd;\r
53   private BuildParametersMap myBuildParametersMap;\r
54 \r
55 \r
56   @BeforeMethod\r
57   @Override\r
58   protected void setUp() throws Exception {\r
59     super.setUp();\r
60     m = new Mockery();\r
61     myProcessFactory = m.mock(CommandlineBuildProcessFactory.class);\r
62     PackageUsages pu = m.mock(PackageUsages.class);\r
63     i = new NuGetActionFactoryImpl(myProcessFactory, pu, new CommandFactoryImpl());\r
64     ctx = m.mock(BuildRunnerContext.class);\r
65     ps = m.mock(PackagesUpdateParameters.class);\r
66     nugetParams = m.mock(NuGetFetchParameters.class);\r
67 \r
68     myTarget = createTempDir();\r
69     myConfig = createTempFile();\r
70 \r
71     myBuildParametersMap = m.mock(BuildParametersMap.class);\r
72     cmd = System.getenv("ComSpec");\r
73 \r
74     m.checking(new Expectations(){{\r
75       allowing(ctx).getBuildParameters(); will(returnValue(myBuildParametersMap));\r
76       allowing(myBuildParametersMap).getEnvironmentVariables(); will(returnValue(Collections.singletonMap("ComSpec", cmd)));\r
77 \r
78       allowing(ps).getNuGetParameters(); will(returnValue(nugetParams));\r
79     }});\r
80   }\r
81 \r
82   @Test\r
83   public void test_no_sources() throws RunBuildException, IOException {\r
84     final File nuget = createTempFile();\r
85     m.checking(new Expectations(){{\r
86       allowing(nugetParams).getNuGetPackageSources(); will(returnValue(Collections.<String>emptyList()));\r
87       allowing(nugetParams).getNuGetExeFile();  will(returnValue(nuget));\r
88       allowing(ps).getUseSafeUpdate(); will(returnValue(false));\r
89       allowing(ps).getPackagesToUpdate(); will(returnValue(Collections.<String>emptyList()));\r
90 \r
91       oneOf(myProcessFactory).executeCommandLine(\r
92               ctx,\r
93               cmd,\r
94               Arrays.asList("/c", nuget.getPath(), "update", myConfig.getPath(), "-Verbose", "-RepositoryPath", myTarget.getPath()),\r
95               myConfig.getParentFile(),\r
96               Collections.<String, String>emptyMap()\r
97       );\r
98     }});\r
99 \r
100     i.createUpdate(ctx, ps, myConfig, myTarget);\r
101     m.assertIsSatisfied();\r
102   }\r
103 \r
104   @Test\r
105   public void test_packageIds() throws RunBuildException, IOException {\r
106     final File nuget = createTempFile();\r
107     m.checking(new Expectations(){{\r
108       allowing(nugetParams).getNuGetPackageSources(); will(returnValue(Collections.<String>emptyList()));\r
109       allowing(nugetParams).getNuGetExeFile();  will(returnValue(nuget));\r
110       allowing(ps).getUseSafeUpdate(); will(returnValue(false));\r
111       allowing(ps).getPackagesToUpdate(); will(returnValue(Arrays.asList("aaa", "bbb")));\r
112 \r
113       oneOf(myProcessFactory).executeCommandLine(\r
114               ctx,\r
115               cmd,\r
116               Arrays.asList("/c", nuget.getPath(), "update", myConfig.getPath(), "-Verbose", "-RepositoryPath", myTarget.getPath(), "-Id", "aaa", "-Id", "bbb"),\r
117               myConfig.getParentFile(),\r
118               Collections.<String, String>emptyMap()\r
119       );\r
120     }});\r
121 \r
122     i.createUpdate(ctx, ps, myConfig, myTarget);\r
123     m.assertIsSatisfied();\r
124   }\r
125 \r
126   @Test\r
127   public void test_safe() throws RunBuildException, IOException {\r
128     final File nuget = createTempFile();\r
129     m.checking(new Expectations(){{\r
130       allowing(nugetParams).getNuGetPackageSources(); will(returnValue(Collections.<String>emptyList()));\r
131       allowing(nugetParams).getNuGetExeFile();  will(returnValue(nuget));\r
132       allowing(ps).getUseSafeUpdate(); will(returnValue(true));\r
133       allowing(ps).getPackagesToUpdate(); will(returnValue(Collections.<String>emptyList()));\r
134 \r
135       oneOf(myProcessFactory).executeCommandLine(\r
136               ctx,\r
137               cmd,\r
138               Arrays.asList("/c", nuget.getPath(), "update", myConfig.getPath(), "-Safe", "-Verbose", "-RepositoryPath", myTarget.getPath()),\r
139               myConfig.getParentFile(),\r
140               Collections.<String, String>emptyMap()\r
141       );\r
142     }});\r
143 \r
144     i.createUpdate(ctx, ps, myConfig, myTarget);\r
145     m.assertIsSatisfied();\r
146   }\r
147 \r
148   @Test\r
149   public void test_sources() throws RunBuildException, IOException {\r
150     final File nuget = createTempFile();\r
151     m.checking(new Expectations(){{\r
152       allowing(NuGetUpdatePackageActionFactoryTest.this.nugetParams).getNuGetPackageSources(); will(returnValue(Arrays.asList("aaa", "bbb")));\r
153       allowing(NuGetUpdatePackageActionFactoryTest.this.nugetParams).getNuGetExeFile();  will(returnValue(nuget));\r
154       allowing(ps).getUseSafeUpdate(); will(returnValue(false));\r
155       allowing(ps).getPackagesToUpdate(); will(returnValue(Collections.<String>emptyList()));\r
156 \r
157       oneOf(myProcessFactory).executeCommandLine(\r
158               ctx,\r
159               cmd,\r
160               Arrays.asList("/c", nuget.getPath(), "update", myConfig.getPath(), "-Verbose", "-RepositoryPath", myTarget.getPath(), "-Source", "aaa", "-Source", "bbb"),\r
161               myConfig.getParentFile(),\r
162               Collections.<String, String>emptyMap()\r
163       );\r
164     }});\r
165 \r
166     i.createUpdate(ctx, ps, myConfig, myTarget);\r
167     m.assertIsSatisfied();\r
168   }\r
169 \r
170 }\r