|
1 | | -// SPDX-FileCopyrightText: © 2021-2022 MONAI Consortium |
2 | | -// SPDX-License-Identifier: Apache License 2.0 |
3 | | - |
4 | | -using BoDi; |
5 | | -using FluentAssertions; |
6 | | -using Monai.Deploy.Messaging.Events; |
7 | | -using Monai.Deploy.Messaging.Messages; |
8 | | -using Monai.Deploy.WorkflowManager.IntegrationTests.Support; |
9 | | -using Polly; |
10 | | -using Polly.Retry; |
11 | | - |
12 | | -namespace Monai.Deploy.WorkflowManager.IntegrationTests.StepDefinitions |
13 | | -{ |
14 | | - [Binding] |
15 | | - public class TaskStatusUpdateStepDefinitions |
16 | | - { |
17 | | - private MongoClientUtil MongoClient { get; set; } |
18 | | - private RabbitPublisher TaskUpdatePublisher { get; set; } |
19 | | - private RetryPolicy RetryPolicy { get; set; } |
20 | | - private DataHelper DataHelper { get; set; } |
21 | | - |
22 | | - public TaskStatusUpdateStepDefinitions(ObjectContainer objectContainer) |
23 | | - { |
24 | | - TaskUpdatePublisher = objectContainer.Resolve<RabbitPublisher>("TaskUpdatePublisher"); |
25 | | - MongoClient = objectContainer.Resolve<MongoClientUtil>(); |
26 | | - DataHelper = objectContainer.Resolve<DataHelper>(); |
27 | | - RetryPolicy = Policy.Handle<Exception>().WaitAndRetry(retryCount: 10, sleepDurationProvider: _ => TimeSpan.FromMilliseconds(500)); |
28 | | - } |
29 | | - |
30 | | - [When(@"I publish a Task Update Message (.*) with status (.*)")] |
31 | | - public void WhenIPublishATaskUpdateMessageTaskUpdateMessage(string name, string updateStatus) |
32 | | - { |
33 | | - var message = new JsonMessage<TaskUpdateEvent>( |
34 | | - DataHelper.GetTaskUpdateTestData(name, updateStatus), |
35 | | - "16988a78-87b5-4168-a5c3-2cfc2bab8e54", |
36 | | - Guid.NewGuid().ToString(), |
37 | | - string.Empty); |
38 | | - |
39 | | - TaskUpdatePublisher.PublishMessage(message.ToMessage()); |
40 | | - } |
41 | | - |
42 | | - [Then(@"I can see the status of the Tasks are updated")] |
43 | | - [Then(@"I can see the status of the Task is updated")] |
44 | | - public void ThenICanSeeTheStatusOfTheTaskIsUpdated() |
45 | | - { |
46 | | - RetryPolicy.Execute(() => |
47 | | - { |
48 | | - var workflowInstance = MongoClient.GetWorkflowInstanceById(DataHelper.TaskUpdateEvent.WorkflowInstanceId); |
49 | | - |
50 | | - var taskUpdated = workflowInstance.Tasks.FirstOrDefault(x => x.TaskId.Equals(DataHelper.TaskUpdateEvent.TaskId)); |
51 | | - |
52 | | - taskUpdated.Status.Should().Be(DataHelper.TaskUpdateEvent.Status); |
53 | | - |
54 | | - if (DataHelper.TaskDispatchEvents.Count > 0) |
55 | | - { |
56 | | - foreach (var e in DataHelper.TaskDispatchEvents) |
57 | | - { |
58 | | - var taskDispatched = workflowInstance.Tasks.FirstOrDefault(x => x.TaskId.Equals(e.TaskId)); |
59 | | - |
60 | | - taskDispatched.Status.Should().Be(TaskExecutionStatus.Dispatched); |
61 | | - } |
62 | | - } |
63 | | - }); |
64 | | - } |
65 | | - |
66 | | - [Then(@"I can see the status of the Task is not updated")] |
67 | | - public void ThenICanSeeTheStatusOfTheTaskIsNotUpdated() |
68 | | - { |
69 | | - for (int i = 0; i < 2; i++) |
70 | | - { |
71 | | - Thread.Sleep(2000); |
72 | | - |
73 | | - var updatedWorkflowInstance = MongoClient.GetWorkflowInstanceById(DataHelper.TaskUpdateEvent.WorkflowInstanceId); |
74 | | - |
75 | | - var orignalWorkflowInstance = DataHelper.WorkflowInstances.FirstOrDefault(x => x.Id.Equals(DataHelper.TaskUpdateEvent.WorkflowInstanceId)); |
76 | | - |
77 | | - updatedWorkflowInstance.Tasks[0].Status.Should().Be(orignalWorkflowInstance?.Tasks[0].Status); |
78 | | - } |
79 | | - } |
80 | | - |
81 | | - [Then(@"I can see the status of the Task is Succeeded")] |
82 | | - public void ThenICanSeeTheStatusOfTheTaskIsSucceeded() |
83 | | - { |
84 | | - var counter = 0; |
85 | | - var updatedWorkflowInstance = MongoClient.GetWorkflowInstanceById(DataHelper.TaskUpdateEvent.WorkflowInstanceId); |
86 | | - while (updatedWorkflowInstance.Tasks[0].Status == TaskExecutionStatus.Dispatched || updatedWorkflowInstance.Tasks.Count < 2) |
87 | | - { |
88 | | - updatedWorkflowInstance = MongoClient.GetWorkflowInstanceById(DataHelper.TaskUpdateEvent.WorkflowInstanceId); |
89 | | - Thread.Sleep(1000); |
90 | | - counter++; |
91 | | - if (counter == 10) |
92 | | - { |
93 | | - throw new Exception("Task Update Status did not complete in sufficient time."); |
94 | | - } |
95 | | - } |
96 | | - updatedWorkflowInstance.Tasks[0].Status.Should().Be(TaskExecutionStatus.Succeeded); |
97 | | - |
98 | | - var orignalWorkflowInstance = DataHelper.WorkflowInstances.FirstOrDefault(x => x.Id.Equals(DataHelper.TaskUpdateEvent.WorkflowInstanceId)); |
99 | | - } |
100 | | - } |
101 | | -} |
| 1 | +// SPDX-FileCopyrightText: © 2021-2022 MONAI Consortium |
| 2 | +// SPDX-License-Identifier: Apache License 2.0 |
| 3 | + |
| 4 | +using BoDi; |
| 5 | +using FluentAssertions; |
| 6 | +using Monai.Deploy.Messaging.Events; |
| 7 | +using Monai.Deploy.Messaging.Messages; |
| 8 | +using Monai.Deploy.WorkflowManager.IntegrationTests.Support; |
| 9 | +using Polly; |
| 10 | +using Polly.Retry; |
| 11 | +using TechTalk.SpecFlow.Infrastructure; |
| 12 | + |
| 13 | +namespace Monai.Deploy.WorkflowManager.IntegrationTests.StepDefinitions |
| 14 | +{ |
| 15 | + [Binding] |
| 16 | + public class TaskStatusUpdateStepDefinitions |
| 17 | + { |
| 18 | + private MongoClientUtil MongoClient { get; set; } |
| 19 | + private RabbitPublisher TaskUpdatePublisher { get; set; } |
| 20 | + private RetryPolicy RetryPolicy { get; set; } |
| 21 | + private DataHelper DataHelper { get; set; } |
| 22 | + private readonly ISpecFlowOutputHelper _outputHelper; |
| 23 | + |
| 24 | + public TaskStatusUpdateStepDefinitions(ObjectContainer objectContainer, ISpecFlowOutputHelper outputHelper) |
| 25 | + { |
| 26 | + TaskUpdatePublisher = objectContainer.Resolve<RabbitPublisher>("TaskUpdatePublisher"); |
| 27 | + MongoClient = objectContainer.Resolve<MongoClientUtil>(); |
| 28 | + DataHelper = objectContainer.Resolve<DataHelper>(); |
| 29 | + RetryPolicy = Policy.Handle<Exception>().WaitAndRetry(retryCount: 10, sleepDurationProvider: _ => TimeSpan.FromMilliseconds(500)); |
| 30 | + _outputHelper = outputHelper; |
| 31 | + } |
| 32 | + |
| 33 | + [When(@"I publish a Task Update Message (.*) with status (.*)")] |
| 34 | + public void WhenIPublishATaskUpdateMessageTaskUpdateMessage(string name, string updateStatus) |
| 35 | + { |
| 36 | + var message = new JsonMessage<TaskUpdateEvent>( |
| 37 | + DataHelper.GetTaskUpdateTestData(name, updateStatus), |
| 38 | + "16988a78-87b5-4168-a5c3-2cfc2bab8e54", |
| 39 | + Guid.NewGuid().ToString(), |
| 40 | + string.Empty); |
| 41 | + |
| 42 | + TaskUpdatePublisher.PublishMessage(message.ToMessage()); |
| 43 | + } |
| 44 | + |
| 45 | + [Then(@"I can see the status of the Tasks are updated")] |
| 46 | + [Then(@"I can see the status of the Task is updated")] |
| 47 | + public void ThenICanSeeTheStatusOfTheTaskIsUpdated() |
| 48 | + { |
| 49 | + RetryPolicy.Execute(() => |
| 50 | + { |
| 51 | + _outputHelper.WriteLine($"Retrieving workflow instance by id={DataHelper.TaskUpdateEvent.WorkflowInstanceId}"); |
| 52 | + var workflowInstance = MongoClient.GetWorkflowInstanceById(DataHelper.TaskUpdateEvent.WorkflowInstanceId); |
| 53 | + _outputHelper.WriteLine("Retrieved workflow instance"); |
| 54 | + |
| 55 | + var taskUpdated = workflowInstance.Tasks.FirstOrDefault(x => x.TaskId.Equals(DataHelper.TaskUpdateEvent.TaskId)); |
| 56 | + |
| 57 | + taskUpdated.Status.Should().Be(DataHelper.TaskUpdateEvent.Status); |
| 58 | + |
| 59 | + if (DataHelper.TaskDispatchEvents.Count > 0) |
| 60 | + { |
| 61 | + foreach (var e in DataHelper.TaskDispatchEvents) |
| 62 | + { |
| 63 | + var taskDispatched = workflowInstance.Tasks.FirstOrDefault(x => x.TaskId.Equals(e.TaskId)); |
| 64 | + |
| 65 | + taskDispatched.Status.Should().Be(TaskExecutionStatus.Dispatched); |
| 66 | + } |
| 67 | + } |
| 68 | + }); |
| 69 | + } |
| 70 | + |
| 71 | + [Then(@"I can see the status of the Task is not updated")] |
| 72 | + public void ThenICanSeeTheStatusOfTheTaskIsNotUpdated() |
| 73 | + { |
| 74 | + for (int i = 0; i < 2; i++) |
| 75 | + { |
| 76 | + Thread.Sleep(2000); |
| 77 | + var updatedWorkflowInstance = MongoClient.GetWorkflowInstanceById(DataHelper.TaskUpdateEvent.WorkflowInstanceId); |
| 78 | + var orignalWorkflowInstance = DataHelper.WorkflowInstances.FirstOrDefault(x => x.Id.Equals(DataHelper.TaskUpdateEvent.WorkflowInstanceId)); |
| 79 | + updatedWorkflowInstance.Tasks[0].Status.Should().Be(orignalWorkflowInstance?.Tasks[0].Status); |
| 80 | + } |
| 81 | + } |
| 82 | + |
| 83 | + [Then(@"I can see the status of the Task is Succeeded")] |
| 84 | + public void ThenICanSeeTheStatusOfTheTaskIsSucceeded() |
| 85 | + { |
| 86 | + var counter = 0; |
| 87 | + _outputHelper.WriteLine($"Retrieving workflow instance by id={DataHelper.TaskUpdateEvent.WorkflowInstanceId}"); |
| 88 | + var updatedWorkflowInstance = MongoClient.GetWorkflowInstanceById(DataHelper.TaskUpdateEvent.WorkflowInstanceId); |
| 89 | + _outputHelper.WriteLine("Retrieved workflow instance"); |
| 90 | + |
| 91 | + while (updatedWorkflowInstance.Tasks[0].Status == TaskExecutionStatus.Dispatched || updatedWorkflowInstance.Tasks.Count < 2) |
| 92 | + { |
| 93 | + updatedWorkflowInstance = MongoClient.GetWorkflowInstanceById(DataHelper.TaskUpdateEvent.WorkflowInstanceId); |
| 94 | + Thread.Sleep(1000); |
| 95 | + counter++; |
| 96 | + if (counter == 10) |
| 97 | + { |
| 98 | + throw new Exception("Task Update Status did not complete in sufficient time."); |
| 99 | + } |
| 100 | + } |
| 101 | + updatedWorkflowInstance.Tasks[0].Status.Should().Be(TaskExecutionStatus.Succeeded); |
| 102 | + |
| 103 | + var orignalWorkflowInstance = DataHelper.WorkflowInstances.FirstOrDefault(x => x.Id.Equals(DataHelper.TaskUpdateEvent.WorkflowInstanceId)); |
| 104 | + } |
| 105 | + } |
| 106 | +} |
0 commit comments