slhaf/Partner
1
0
Fork 0
mirror of https://github.com/slhaf/Partner.git synced 2026-05-12 16:53:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(test): adapt ActionExecutor tests to Result-based extractor/corrector execution flow

Browse Source
This commit is contained in:
slhafzjw
2026-04-12 19:19:07 +08:00
parent 04d6ad6d80
commit e37a282141
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
Partner-Core/src/test/java/work/slhaf/partner/module/action/executor/ActionExecutorTest.java
View File

@@ -138,7 +138,7 @@ class ActionExecutorTest {
ExtractorResult extractorResult = new ExtractorResult(); ExtractorResult extractorResult = new ExtractorResult();
extractorResult.setOk(true); extractorResult.setOk(true);
extractorResult.setParams(Map.of("fresh", "value")); extractorResult.setParams(Map.of("fresh", "value"));
when(paramsExtractor.execute(any())).thenReturn(extractorResult); when(paramsExtractor.execute(any())).thenReturn(Result.success(extractorResult));
doAnswer(invocation -> { doAnswer(invocation -> {
MetaAction metaAction = invocation.getArgument(0); MetaAction metaAction = invocation.getArgument(0);
metaAction.getResult().setStatus(MetaAction.Result.Status.SUCCESS); metaAction.getResult().setStatus(MetaAction.Result.Status.SUCCESS);
@@ -209,6 +209,11 @@ class ActionExecutorTest {
"META_ACTION_INFO" "META_ACTION_INFO"
))); )));
when(cognitionCapability.contextWorkspace()).thenReturn(new ContextWorkspace()); when(cognitionCapability.contextWorkspace()).thenReturn(new ContextWorkspace());
work.slhaf.partner.module.action.executor.entity.CorrectorResult correctorResult =
new work.slhaf.partner.module.action.executor.entity.CorrectorResult();
correctorResult.setMetaInterventionList(List.of());
correctorResult.setCorrectionReason("no-op");
when(actionCorrector.execute(any())).thenReturn(Result.success(correctorResult));
ActionExecutor actionExecutor = new ActionExecutor(); ActionExecutor actionExecutor = new ActionExecutor();
inject(actionExecutor, "actionCapability", actionCapability); inject(actionExecutor, "actionCapability", actionCapability);
@@ -235,7 +240,7 @@ class ActionExecutorTest {
verify(runnerClient, never()).submit(any(MetaAction.class)); verify(runnerClient, never()).submit(any(MetaAction.class));
assertEquals(Action.Status.SUCCESS, action.getStatus()); assertEquals(Action.Status.SUCCESS, action.getStatus());
assertEquals(MetaAction.Result.Status.FAILED, metaAction.getResult().getStatus()); assertEquals(MetaAction.Result.Status.FAILED, metaAction.getResult().getStatus());
assertTrue(metaAction.getResult().getData().contains("参数提取失败")); assertTrue(metaAction.getResult().getData().contains("missing"));
assertEquals(metaAction.getResult().getData(), action.getResult()); assertEquals(metaAction.getResult().getData(), action.getResult());
} }
@@ -278,7 +283,7 @@ class ActionExecutorTest {
ExtractorResult extractorResult = new ExtractorResult(); ExtractorResult extractorResult = new ExtractorResult();
extractorResult.setOk(true); extractorResult.setOk(true);
extractorResult.setParams(Map.of("fresh", "value")); extractorResult.setParams(Map.of("fresh", "value"));
when(paramsExtractor.execute(any())).thenReturn(extractorResult); when(paramsExtractor.execute(any())).thenReturn(Result.success(extractorResult));
doAnswer(invocation -> { doAnswer(invocation -> {
MetaAction metaAction = invocation.getArgument(0); MetaAction metaAction = invocation.getArgument(0);
metaAction.getResult().setStatus(MetaAction.Result.Status.SUCCESS); metaAction.getResult().setStatus(MetaAction.Result.Status.SUCCESS);