Skip to content

Commit 08038f9

Browse files
committed
add test case
1 parent a229d65 commit 08038f9

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

datahub-upgrade/src/test/java/com/linkedin/datahub/upgrade/restoreindices/SendMAEStepTest.java

+26
Original file line numberDiff line numberDiff line change
@@ -389,4 +389,30 @@ public void testReportAddedLines() {
389389
// Verify report lines are added
390390
verify(mockReport, atLeastOnce()).addLine(anyString());
391391
}
392+
393+
@Test
394+
public void testExecutableWithCreateDefaultAspects() {
395+
// Setup
396+
parsedArgs.put(RestoreIndices.CREATE_DEFAULT_ASPECTS_ARG_NAME, Optional.of("true"));
397+
398+
// Insert test data
399+
insertTestRows(3, null);
400+
401+
// Execute
402+
UpgradeStepResult result = sendMAEStep.executable().apply(mockContext);
403+
404+
// Verify result
405+
assertTrue(result instanceof DefaultUpgradeStepResult);
406+
assertEquals(result.result(), DataHubUpgradeState.SUCCEEDED);
407+
assertEquals(result.stepId(), sendMAEStep.id());
408+
assertEquals(result.action(), UpgradeStepResult.Action.CONTINUE);
409+
410+
// Verify createDefaultAspects parameter
411+
ArgumentCaptor<RestoreIndicesArgs> argsCaptor =
412+
ArgumentCaptor.forClass(RestoreIndicesArgs.class);
413+
verify(mockEntityService).restoreIndices(eq(mockOpContext), argsCaptor.capture(), any());
414+
415+
RestoreIndicesArgs capturedArgs = argsCaptor.getValue();
416+
assertTrue(capturedArgs.createDefaultAspects);
417+
}
392418
}

0 commit comments

Comments
 (0)