Skip to content

Commit 55024da

Browse files
Added unit test
1 parent c7843d0 commit 55024da

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

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

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,8 @@
55
import static org.assertj.core.api.Assertions.assertThatThrownBy;
66

77
import com.sap.ai.sdk.orchestration.client.model.ChatMessage;
8+
import com.sap.ai.sdk.orchestration.client.model.DPIConfig;
9+
import com.sap.ai.sdk.orchestration.client.model.DPIEntities;
810
import com.sap.ai.sdk.orchestration.client.model.Template;
911
import java.util.List;
1012
import java.util.Map;
@@ -75,4 +77,28 @@ void testMessagesHistory() {
7577

7678
assertThat(actual.getMessagesHistory()).containsExactly(systemMessage);
7779
}
80+
81+
@Test
82+
void testDpiMaskingConfig() {
83+
var maskingConfig = DpiMasking.anonymization().withEntities(DPIEntities.ADDRESS);
84+
var config =
85+
new OrchestrationModuleConfig()
86+
.withLlmConfig(CUSTOM_GPT_35)
87+
.withMaskingConfig(maskingConfig);
88+
89+
var actual = ConfigToRequestTransformer.toModuleConfigs(config);
90+
91+
assertThat(actual.getMaskingModuleConfig()).isNotNull();
92+
assertThat(actual.getMaskingModuleConfig().getMaskingProviders()).hasSize(1);
93+
DPIConfig dpiConfig = (DPIConfig) actual.getMaskingModuleConfig().getMaskingProviders().get(0);
94+
assertThat(dpiConfig.getMethod()).isEqualTo(DPIConfig.MethodEnum.ANONYMIZATION);
95+
assertThat(dpiConfig.getEntities()).hasSize(1);
96+
assertThat(dpiConfig.getEntities().get(0).getType()).isEqualTo(DPIEntities.ADDRESS);
97+
98+
var configModified = config.withMaskingConfig(maskingConfig);
99+
assertThat(configModified.getMaskingConfig()).isNotNull();
100+
assertThat(configModified.getMaskingConfig().getMaskingProviders())
101+
.withFailMessage("withMaskingConfig() should overwrite the existing config and not append")
102+
.hasSize(1);
103+
}
78104
}

0 commit comments

Comments
 (0)