Skip to content

Commit 5be60ea

Browse files
Added unit test
1 parent a82c5f6 commit 5be60ea

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

orchestration/src/test/java/com/sap/ai/sdk/orchestration/OrchestrationModuleConfigTest.java

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,11 @@ void testThrowOnEmptyFilterConfig() {
7777

7878
@Test
7979
void testDpiMaskingConfig() {
80-
var maskingConfig = DpiMasking.anonymization().withEntities(DPIEntities.ADDRESS);
80+
var maskingConfig =
81+
DpiMasking.anonymization()
82+
.withEntities(DPIEntities.ADDRESS)
83+
.withMaskGroundingEnabled()
84+
.withAllowList(List.of("Alice"));
8185
var config =
8286
new OrchestrationModuleConfig().withLlmConfig(GPT_4O).withMaskingConfig(maskingConfig);
8387

@@ -87,6 +91,8 @@ void testDpiMaskingConfig() {
8791
assertThat(dpiConfig.getMethod()).isEqualTo(DPIConfig.MethodEnum.ANONYMIZATION);
8892
assertThat(dpiConfig.getEntities()).hasSize(1);
8993
assertThat(dpiConfig.getEntities().get(0).getType()).isEqualTo(DPIEntities.ADDRESS);
94+
assertThat(dpiConfig.getMaskGroundingInput().isEnabled()).isEqualTo(true);
95+
assertThat(dpiConfig.getAllowlist()).containsExactly("Alice");
9096

9197
var configModified = config.withMaskingConfig(maskingConfig);
9298
assertThat(configModified.getMaskingConfig()).isNotNull();

0 commit comments

Comments
 (0)