ckaczor/sqltoolsservice
1
0
Fork 0
mirror of https://github.com/ckaczor/sqltoolsservice.git synced 2026-02-16 18:47:57 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add initial schedule request handlers (#638)

Browse Source
This commit is contained in:
Karl Burtram
2018-06-18 10:18:23 -07:00
committed by GitHub
parent d2cc376b87
commit 29fb715ad5
10 changed files with 555 additions and 279 deletions
Download Patch File Download Diff File Expand all files Collapse all files

83
test/Microsoft.SqlTools.ServiceLayer.IntegrationTests/Agent/AgentTestUtils.cs
View File

@@ -18,6 +18,8 @@ namespace Microsoft.SqlTools.ServiceLayer.IntegrationTests.Agent
{
public static class AgentTestUtils
{
public const string TestJobName = "Test Job";
internal static AgentJobStepInfo GetTestJobStepInfo(
TestConnectionResult connectionResult,
AgentJobInfo job,
@@ -40,7 +42,7 @@ namespace Microsoft.SqlTools.ServiceLayer.IntegrationTests.Agent
{
return new AgentJobInfo()
{
Name = "Test Job",
Name = TestJobName,
Owner = "sa",
Description = "Test job description",
CurrentExecutionStatus = 1,
@@ -79,7 +81,17 @@ namespace Microsoft.SqlTools.ServiceLayer.IntegrationTests.Agent
Description = "Test proxy description",
IsEnabled = true
};
}
}
internal static AgentScheduleInfo GetTestScheduleInfo()
{
return new AgentScheduleInfo()
{
Name = "Test Schedule",
JobName = TestJobName,
IsEnabled = true
};
}
internal static async Task CreateAgentJob(
AgentService service,
@@ -211,7 +223,7 @@ namespace Microsoft.SqlTools.ServiceLayer.IntegrationTests.Agent
Operator = operatorInfo
}, context.Object);
context.VerifyAll();
}
}
internal static async Task CreateAgentProxy(
AgentService service,
@@ -241,7 +253,7 @@ namespace Microsoft.SqlTools.ServiceLayer.IntegrationTests.Agent
Proxy = proxy
}, context.Object);
context.VerifyAll();
}
}
internal static async Task DeleteAgentProxy(
AgentService service,
@@ -255,7 +267,51 @@ namespace Microsoft.SqlTools.ServiceLayer.IntegrationTests.Agent
Proxy = proxy
}, context.Object);
context.VerifyAll();
}
}
internal static async Task CreateAgentSchedule(
AgentService service,
TestConnectionResult connectionResult,
AgentScheduleInfo schedule)
{
var context = new Mock<RequestContext<AgentScheduleResult>>();
await service.HandleCreateAgentScheduleRequest(new CreateAgentScheduleParams
{
OwnerUri = connectionResult.ConnectionInfo.OwnerUri,
Schedule = schedule
}, context.Object);
context.VerifyAll();
}
internal static async Task UpdateAgentSchedule(
AgentService service,
TestConnectionResult connectionResult,
string originalScheduleName,
AgentScheduleInfo schedule)
{
var context = new Mock<RequestContext<AgentScheduleResult>>();
await service.HandleUpdateAgentScheduleRequest(new UpdateAgentScheduleParams()
{
OwnerUri = connectionResult.ConnectionInfo.OwnerUri,
OriginalScheduleName = originalScheduleName,
Schedule = schedule
}, context.Object);
context.VerifyAll();
}
internal static async Task DeleteAgentSchedule(
AgentService service,
TestConnectionResult connectionResult,
AgentScheduleInfo schedule)
{
var context = new Mock<RequestContext<ResultStatus>>();
await service.HandleDeleteAgentScheduleRequest(new DeleteAgentScheduleParams()
{
OwnerUri = connectionResult.ConnectionInfo.OwnerUri,
Schedule = schedule
}, context.Object);
context.VerifyAll();
}
internal static async Task<AgentAlertInfo[]> GetAgentAlerts(string connectionUri)
{
@@ -274,5 +330,22 @@ namespace Microsoft.SqlTools.ServiceLayer.IntegrationTests.Agent
await service.HandleAgentAlertsRequest(requestParams, requestContext.Object);
return agentAlerts;
}
public static async Task<AgentJobInfo> SetupJob(TestConnectionResult connectionResult)
{
var service = new AgentService();
var job = GetTestJobInfo();
await DeleteAgentJob(service, connectionResult, job);
await CreateAgentJob(service, connectionResult, job);
return job;
}
public static async Task CleanupJob(
TestConnectionResult connectionResult,
AgentJobInfo job)
{
var service = new AgentService();
await DeleteAgentJob(service, connectionResult, job);
}
}
}