Skip to content

Commit ad83ab8

Browse files
author
jasper-schneider
committed
Get Task scenario tests
1 parent d689b14 commit ad83ab8

File tree

10 files changed

+6298
-0
lines changed

10 files changed

+6298
-0
lines changed

‎src/ResourceManager/Batch/Commands.Batch.Test/Commands.Batch.Test.csproj

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -160,6 +160,7 @@
160160
<Compile Include="ScenarioTests\JobTests.cs" />
161161
<Compile Include="ScenarioTests\PoolTests.cs" />
162162
<Compile Include="ScenarioTests\ScenarioTestHelpers.cs" />
163+
<Compile Include="ScenarioTests\TaskTests.cs" />
163164
<Compile Include="ScenarioTests\WorkItemTests.cs" />
164165
<Compile Include="Tasks\GetBatchTaskCommandTests.cs" />
165166
<Compile Include="WorkItems\GetBatchWorkItemCommandTests.cs" />
@@ -181,6 +182,9 @@
181182
<None Include="ScenarioTests\PoolTests.ps1">
182183
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
183184
</None>
185+
<None Include="ScenarioTests\TaskTests.ps1">
186+
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
187+
</None>
184188
<None Include="ScenarioTests\WorkItemTests.ps1">
185189
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
186190
</None>
@@ -235,6 +239,24 @@
235239
<None Include="SessionRecords\Microsoft.Azure.Commands.Batch.Test.ScenarioTests.PoolTests\TestListPoolsWithMaxCount.json">
236240
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
237241
</None>
242+
<None Include="SessionRecords\Microsoft.Azure.Commands.Batch.Test.ScenarioTests.TaskTests\TestGetTaskByName.json">
243+
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
244+
</None>
245+
<None Include="SessionRecords\Microsoft.Azure.Commands.Batch.Test.ScenarioTests.TaskTests\TestGetTaskRequiredParameters.json">
246+
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
247+
</None>
248+
<None Include="SessionRecords\Microsoft.Azure.Commands.Batch.Test.ScenarioTests.TaskTests\TestListAllTasks.json">
249+
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
250+
</None>
251+
<None Include="SessionRecords\Microsoft.Azure.Commands.Batch.Test.ScenarioTests.TaskTests\TestListTaskPipeline.json">
252+
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
253+
</None>
254+
<None Include="SessionRecords\Microsoft.Azure.Commands.Batch.Test.ScenarioTests.TaskTests\TestListTasksByFilter.json">
255+
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
256+
</None>
257+
<None Include="SessionRecords\Microsoft.Azure.Commands.Batch.Test.ScenarioTests.TaskTests\TestListTasksWithMaxCount.json">
258+
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
259+
</None>
238260
<None Include="SessionRecords\Microsoft.Azure.Commands.Batch.Test.ScenarioTests.WorkItemTests\TestGetWorkItemByName.json">
239261
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
240262
</None>

‎src/ResourceManager/Batch/Commands.Batch.Test/ScenarioTests/ScenarioTestHelpers.cs

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -148,6 +148,22 @@ public static string WaitForRecentJob(BatchController controller, BatchAccountCo
148148
return workItem.ExecutionInformation.RecentJob.Name;
149149
}
150150

151+
/// <summary>
152+
/// Creates a test Task for use in Scenario tests.
153+
/// TODO: Replace with new Task client method when it exists.
154+
/// </summary>
155+
public static void CreateTestTask(BatchAccountContext context, string workItemName, string jobName, string taskName)
156+
{
157+
if (HttpMockServer.Mode == HttpRecorderMode.Record)
158+
{
159+
using (IWorkItemManager wiManager = context.BatchOMClient.OpenWorkItemManager())
160+
{
161+
ICloudTask task = new CloudTask(taskName, "cmd /c dir /s");
162+
wiManager.AddTask(workItemName, jobName, task);
163+
}
164+
}
165+
}
166+
151167
/// <summary>
152168
/// Deletes a WorkItem used in a Scenario test.
153169
/// TODO: Replace with remove WorkItem client method when it exists.

‎src/ResourceManager/Batch/Commands.Batch.Test/ScenarioTests/TaskTests.cs

Lines changed: 230 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,230 @@
1+
// ----------------------------------------------------------------------------------
2+
//
3+
// Copyright Microsoft Corporation
4+
// Licensed under the Apache License, Version 2.0 (the "License");
5+
// you may not use this file except in compliance with the License.
6+
// You may obtain a copy of the License at
7+
// http://www.apache.org/licenses/LICENSE-2.0
8+
// Unless required by applicable law or agreed to in writing, software
9+
// distributed under the License is distributed on an "AS IS" BASIS,
10+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11+
// See the License for the specific language governing permissions and
12+
// limitations under the License.
13+
// ----------------------------------------------------------------------------------
14+
15+
using System;
16+
using Microsoft.Azure.Batch;
17+
using Microsoft.Azure.Commands.Batch.Models;
18+
using Microsoft.Azure.Test;
19+
using Microsoft.WindowsAzure.Commands.ScenarioTest;
20+
using System.Collections.Generic;
21+
using System.Management.Automation;
22+
using Xunit;
23+
using Constants = Microsoft.Azure.Commands.Batch.Utils.Constants;
24+
25+
namespace Microsoft.Azure.Commands.Batch.Test.ScenarioTests
26+
{
27+
public class TaskTests
28+
{
29+
[Fact]
30+
[Trait(Category.AcceptanceType, Category.CheckIn)]
31+
public void TestGetTaskRequiredParameters()
32+
{
33+
BatchController controller = BatchController.NewInstance;
34+
string resourceGroupName = "test-get-task-params";
35+
string accountName = "testgettaskparams";
36+
string location = "eastus";
37+
BatchAccountContext context = null;
38+
controller.RunPsTestWorkflow(
39+
() => { return new string[] { string.Format("Test-GetTaskRequiredParameters '{0}'", accountName) }; },
40+
() =>
41+
{
42+
context = ScenarioTestHelpers.CreateTestAccountAndResourceGroup(controller, resourceGroupName, accountName, location);
43+
},
44+
() =>
45+
{
46+
ScenarioTestHelpers.CleanupTestAccount(controller, resourceGroupName, accountName);
47+
},
48+
TestUtilities.GetCallingClass(),
49+
TestUtilities.GetCurrentMethodName());
50+
}
51+
52+
[Fact]
53+
[Trait(Category.AcceptanceType, Category.CheckIn)]
54+
public void TestGetTaskByName()
55+
{
56+
BatchController controller = BatchController.NewInstance;
57+
string resourceGroupName = "test-get-task";
58+
string accountName = "testgettaskbyname";
59+
string location = "eastus";
60+
string workItemName = "testName";
61+
string jobName = null;
62+
string taskName = "testTask";
63+
BatchAccountContext context = null;
64+
controller.RunPsTestWorkflow(
65+
() => { return new string[] { string.Format("Test-GetTaskByName '{0}' '{1}' '{2}' '{3}'", accountName, workItemName, jobName, taskName) }; },
66+
() =>
67+
{
68+
context = ScenarioTestHelpers.CreateTestAccountAndResourceGroup(controller, resourceGroupName, accountName, location);
69+
ScenarioTestHelpers.CreateTestWorkItem(context, workItemName);
70+
jobName = ScenarioTestHelpers.WaitForRecentJob(controller, context, workItemName);
71+
ScenarioTestHelpers.CreateTestTask(context, workItemName, jobName, taskName);
72+
},
73+
() =>
74+
{
75+
ScenarioTestHelpers.DeleteWorkItem(context, workItemName);
76+
ScenarioTestHelpers.CleanupTestAccount(controller, resourceGroupName, accountName);
77+
},
78+
TestUtilities.GetCallingClass(),
79+
TestUtilities.GetCurrentMethodName());
80+
}
81+
82+
[Fact]
83+
[Trait(Category.AcceptanceType, Category.CheckIn)]
84+
public void TestListTasksByFilter()
85+
{
86+
BatchController controller = BatchController.NewInstance;
87+
string resourceGroupName = "test-list-task-filter";
88+
string accountName = "testlisttaskfilter";
89+
string location = "eastus";
90+
string workItemName = "testWorkItem";
91+
string jobName = null;
92+
string taskName1 = "testTask1";
93+
string taskName2 = "testTask2";
94+
string taskName3 = "thirdTestTask";
95+
string taskPrefix = "testTask";
96+
int matches = 2;
97+
BatchAccountContext context = null;
98+
controller.RunPsTestWorkflow(
99+
() => { return new string[] { string.Format("Test-ListTasksByFilter '{0}' '{1}' '{2}' '{3}' '{4}'", accountName, workItemName, jobName, taskPrefix, matches) }; },
100+
() =>
101+
{
102+
context = ScenarioTestHelpers.CreateTestAccountAndResourceGroup(controller, resourceGroupName, accountName, location);
103+
ScenarioTestHelpers.CreateTestWorkItem(context, workItemName);
104+
jobName = ScenarioTestHelpers.WaitForRecentJob(controller, context, workItemName);
105+
ScenarioTestHelpers.CreateTestTask(context, workItemName, jobName, taskName1);
106+
ScenarioTestHelpers.CreateTestTask(context, workItemName, jobName, taskName2);
107+
ScenarioTestHelpers.CreateTestTask(context, workItemName, jobName, taskName3);
108+
},
109+
() =>
110+
{
111+
ScenarioTestHelpers.DeleteWorkItem(context, workItemName);
112+
ScenarioTestHelpers.CleanupTestAccount(controller, resourceGroupName, accountName);
113+
},
114+
TestUtilities.GetCallingClass(),
115+
TestUtilities.GetCurrentMethodName());
116+
}
117+
118+
[Fact]
119+
[Trait(Category.AcceptanceType, Category.CheckIn)]
120+
public void TestListTasksWithMaxCount()
121+
{
122+
BatchController controller = BatchController.NewInstance;
123+
string resourceGroupName = "test-list-task-maxcount";
124+
string accountName = "testlisttaskmaxcount";
125+
string location = "eastus";
126+
string workItemName = "testWorkItem";
127+
string jobName = null;
128+
string taskName1 = "testTask1";
129+
string taskName2 = "testTask2";
130+
string taskName3 = "testTask3";
131+
int maxCount = 1;
132+
BatchAccountContext context = null;
133+
controller.RunPsTestWorkflow(
134+
() => { return new string[] { string.Format("Test-ListTasksWithMaxCount '{0}' '{1}' '{2}' '{3}'", accountName, workItemName, jobName, maxCount) }; },
135+
() =>
136+
{
137+
context = ScenarioTestHelpers.CreateTestAccountAndResourceGroup(controller, resourceGroupName, accountName, location);
138+
ScenarioTestHelpers.CreateTestWorkItem(context, workItemName);
139+
jobName = ScenarioTestHelpers.WaitForRecentJob(controller, context, workItemName);
140+
ScenarioTestHelpers.CreateTestTask(context, workItemName, jobName, taskName1);
141+
ScenarioTestHelpers.CreateTestTask(context, workItemName, jobName, taskName2);
142+
ScenarioTestHelpers.CreateTestTask(context, workItemName, jobName, taskName3);
143+
},
144+
() =>
145+
{
146+
ScenarioTestHelpers.DeleteWorkItem(context, workItemName);
147+
ScenarioTestHelpers.CleanupTestAccount(controller, resourceGroupName, accountName);
148+
},
149+
TestUtilities.GetCallingClass(),
150+
TestUtilities.GetCurrentMethodName());
151+
}
152+
153+
[Fact]
154+
[Trait(Category.AcceptanceType, Category.CheckIn)]
155+
public void TestListAllTasks()
156+
{
157+
BatchController controller = BatchController.NewInstance;
158+
string resourceGroupName = "test-list-task";
159+
string accountName = "testlisttask";
160+
string location = "eastus";
161+
string workItemName = "testWorkItem";
162+
string jobName = null;
163+
string taskName1 = "testTask1";
164+
string taskName2 = "testTask2";
165+
string taskName3 = "testTask3";
166+
int count = 3;
167+
BatchAccountContext context = null;
168+
controller.RunPsTestWorkflow(
169+
() => { return new string[] { string.Format("Test-ListAllTasks '{0}' '{1}' '{2}' '{3}'", accountName, workItemName, jobName, count) }; },
170+
() =>
171+
{
172+
context = ScenarioTestHelpers.CreateTestAccountAndResourceGroup(controller, resourceGroupName, accountName, location);
173+
ScenarioTestHelpers.CreateTestWorkItem(context, workItemName);
174+
jobName = ScenarioTestHelpers.WaitForRecentJob(controller, context, workItemName);
175+
ScenarioTestHelpers.CreateTestTask(context, workItemName, jobName, taskName1);
176+
ScenarioTestHelpers.CreateTestTask(context, workItemName, jobName, taskName2);
177+
ScenarioTestHelpers.CreateTestTask(context, workItemName, jobName, taskName3);
178+
},
179+
() =>
180+
{
181+
ScenarioTestHelpers.DeleteWorkItem(context, workItemName);
182+
ScenarioTestHelpers.CleanupTestAccount(controller, resourceGroupName, accountName);
183+
},
184+
TestUtilities.GetCallingClass(),
185+
TestUtilities.GetCurrentMethodName());
186+
}
187+
188+
[Fact]
189+
[Trait(Category.AcceptanceType, Category.CheckIn)]
190+
public void TestListTaskPipeline()
191+
{
192+
BatchController controller = BatchController.NewInstance;
193+
string resourceGroupName = "test-list-task-pipe";
194+
string accountName = "testlisttaskpipe";
195+
string location = "eastus";
196+
string workItemName = "testWorkItem";
197+
string jobName = null;
198+
string taskName = "testTask";
199+
BatchAccountContext context = null;
200+
controller.RunPsTestWorkflow(
201+
() => { return new string[] { string.Format("Test-ListTaskPipeline '{0}' '{1}' '{2}' '{3}'", accountName, workItemName, jobName, taskName) }; },
202+
() =>
203+
{
204+
context = ScenarioTestHelpers.CreateTestAccountAndResourceGroup(controller, resourceGroupName, accountName, location);
205+
ScenarioTestHelpers.CreateTestWorkItem(context, workItemName);
206+
jobName = ScenarioTestHelpers.WaitForRecentJob(controller, context, workItemName);
207+
ScenarioTestHelpers.CreateTestTask(context, workItemName, jobName, taskName);
208+
},
209+
() =>
210+
{
211+
ScenarioTestHelpers.DeleteWorkItem(context, workItemName);
212+
ScenarioTestHelpers.CleanupTestAccount(controller, resourceGroupName, accountName);
213+
},
214+
TestUtilities.GetCallingClass(),
215+
TestUtilities.GetCurrentMethodName());
216+
}
217+
218+
}
219+
220+
// Cmdlets that use the HTTP Recorder interceptor for use with scenario tests
221+
[Cmdlet(VerbsCommon.Get, "AzureBatchTask_ST", DefaultParameterSetName = Constants.ODataFilterParameterSet)]
222+
public class GetBatchTaskScenarioTestCommand : GetBatchTaskCommand
223+
{
224+
public override void ExecuteCmdlet()
225+
{
226+
AdditionalBehaviors = new List<BatchClientBehavior>() { ScenarioTestHelpers.CreateHttpRecordingInterceptor() };
227+
base.ExecuteCmdlet();
228+
}
229+
}
230+
}

0 commit comments

Comments
 (0)