|
35 | 35 | import io.airbyte.workers.temporal.scheduling.activities.ConfigFetchActivity;
|
36 | 36 | import io.airbyte.workers.temporal.scheduling.activities.ConfigFetchActivity.GetMaxAttemptOutput;
|
37 | 37 | import io.airbyte.workers.temporal.scheduling.activities.ConfigFetchActivity.ScheduleRetrieverOutput;
|
38 |
| -import io.airbyte.workers.temporal.scheduling.activities.ConnectionDeletionActivity; |
39 | 38 | import io.airbyte.workers.temporal.scheduling.activities.GenerateInputActivity.GeneratedJobInput;
|
40 | 39 | import io.airbyte.workers.temporal.scheduling.activities.GenerateInputActivity.SyncInputWithAttemptNumber;
|
41 | 40 | import io.airbyte.workers.temporal.scheduling.activities.GenerateInputActivityImpl;
|
@@ -114,8 +113,6 @@ class ConnectionManagerWorkflowTest {
|
114 | 113 | mock(ConfigFetchActivity.class, Mockito.withSettings().withoutAnnotations());
|
115 | 114 | private final CheckConnectionActivity mCheckConnectionActivity =
|
116 | 115 | mock(CheckConnectionActivity.class, Mockito.withSettings().withoutAnnotations());
|
117 |
| - private static final ConnectionDeletionActivity mConnectionDeletionActivity = |
118 |
| - mock(ConnectionDeletionActivity.class, Mockito.withSettings().withoutAnnotations()); |
119 | 116 | private static final GenerateInputActivityImpl mGenerateInputActivityImpl =
|
120 | 117 | mock(GenerateInputActivityImpl.class, Mockito.withSettings().withoutAnnotations());
|
121 | 118 | private static final JobCreationAndStatusUpdateActivity mJobCreationAndStatusUpdateActivity =
|
@@ -152,7 +149,6 @@ static Stream<Arguments> getMaxAttemptForResetRetry() {
|
152 | 149 | void setUp() {
|
153 | 150 | Mockito.reset(mConfigFetchActivity);
|
154 | 151 | Mockito.reset(mCheckConnectionActivity);
|
155 |
| - Mockito.reset(mConnectionDeletionActivity); |
156 | 152 | Mockito.reset(mGenerateInputActivityImpl);
|
157 | 153 | Mockito.reset(mJobCreationAndStatusUpdateActivity);
|
158 | 154 | Mockito.reset(mAutoDisableConnectionActivity);
|
@@ -533,8 +529,6 @@ void deleteSync() throws InterruptedException {
|
533 | 529 | && changedStateEvent.getField() != StateField.DONE_WAITING
|
534 | 530 | && changedStateEvent.isValue())
|
535 | 531 | .isEmpty();
|
536 |
| - |
537 |
| - Mockito.verify(mConnectionDeletionActivity, Mockito.times(0)).deleteConnection(Mockito.any()); |
538 | 532 | }
|
539 | 533 |
|
540 | 534 | @Test
|
@@ -900,8 +894,8 @@ void setup() {
|
900 | 894 |
|
901 | 895 | final Worker managerWorker = testEnv.newWorker(TemporalJobType.CONNECTION_UPDATER.name());
|
902 | 896 | managerWorker.registerWorkflowImplementationTypes(temporalProxyHelper.proxyWorkflowClass(ConnectionManagerWorkflowImpl.class));
|
903 |
| - managerWorker.registerActivitiesImplementations(mConfigFetchActivity, mCheckConnectionActivity, mConnectionDeletionActivity, |
904 |
| - mGenerateInputActivityImpl, mJobCreationAndStatusUpdateActivity, mAutoDisableConnectionActivity, mRecordMetricActivity, |
| 897 | + managerWorker.registerActivitiesImplementations(mConfigFetchActivity, mCheckConnectionActivity, mGenerateInputActivityImpl, |
| 898 | + mJobCreationAndStatusUpdateActivity, mAutoDisableConnectionActivity, mRecordMetricActivity, |
905 | 899 | mWorkflowConfigActivity, mRouteToSyncTaskQueueActivity);
|
906 | 900 |
|
907 | 901 | client = testEnv.getWorkflowClient();
|
@@ -998,8 +992,8 @@ void setup() {
|
998 | 992 |
|
999 | 993 | final Worker managerWorker = testEnv.newWorker(TemporalJobType.CONNECTION_UPDATER.name());
|
1000 | 994 | managerWorker.registerWorkflowImplementationTypes(temporalProxyHelper.proxyWorkflowClass(ConnectionManagerWorkflowImpl.class));
|
1001 |
| - managerWorker.registerActivitiesImplementations(mConfigFetchActivity, mCheckConnectionActivity, mConnectionDeletionActivity, |
1002 |
| - mGenerateInputActivityImpl, mJobCreationAndStatusUpdateActivity, mAutoDisableConnectionActivity, mRecordMetricActivity, |
| 995 | + managerWorker.registerActivitiesImplementations(mConfigFetchActivity, mCheckConnectionActivity, mGenerateInputActivityImpl, |
| 996 | + mJobCreationAndStatusUpdateActivity, mAutoDisableConnectionActivity, mRecordMetricActivity, |
1003 | 997 | mWorkflowConfigActivity, mRouteToSyncTaskQueueActivity);
|
1004 | 998 |
|
1005 | 999 | client = testEnv.getWorkflowClient();
|
@@ -1586,8 +1580,8 @@ private <T extends SyncWorkflow> void setupSpecificChildWorkflow(final Class<T>
|
1586 | 1580 |
|
1587 | 1581 | final Worker managerWorker = testEnv.newWorker(TemporalJobType.CONNECTION_UPDATER.name());
|
1588 | 1582 | managerWorker.registerWorkflowImplementationTypes(temporalProxyHelper.proxyWorkflowClass(ConnectionManagerWorkflowImpl.class));
|
1589 |
| - managerWorker.registerActivitiesImplementations(mConfigFetchActivity, mCheckConnectionActivity, mConnectionDeletionActivity, |
1590 |
| - mGenerateInputActivityImpl, mJobCreationAndStatusUpdateActivity, mAutoDisableConnectionActivity, mRecordMetricActivity, |
| 1583 | + managerWorker.registerActivitiesImplementations(mConfigFetchActivity, mCheckConnectionActivity, mGenerateInputActivityImpl, |
| 1584 | + mJobCreationAndStatusUpdateActivity, mAutoDisableConnectionActivity, mRecordMetricActivity, |
1591 | 1585 | mWorkflowConfigActivity, mRouteToSyncTaskQueueActivity);
|
1592 | 1586 |
|
1593 | 1587 | client = testEnv.getWorkflowClient();
|
|
0 commit comments