Fix mocking

linkedin · Jan 16, 2025 · 1e66fb6 · 1e66fb6
1 parent f2f3388
commit 1e66fb6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...inci-client/src/test/java/com/linkedin/davinci/kafka/consumer/StoreIngestionTaskTest.java b/...inci-client/src/test/java/com/linkedin/davinci/kafka/consumer/StoreIngestionTaskTest.java
@@ -4822,6 +4822,7 @@ public void testStoreIngestionRecordTransformer(AAConfig aaConfig) throws Except
     SchemaEntry valueSchemaEntry = mock(SchemaEntry.class);
     when(valueSchemaEntry.getSchema()).thenReturn(myValueSchema);
     when(mockSchemaRepo.getValueSchema(eq(storeNameWithoutVersionInfo), anyInt())).thenReturn(valueSchemaEntry);
+    when(mockSchemaRepo.getSupersetOrLatestValueSchema(eq(storeNameWithoutVersionInfo))).thenReturn(valueSchemaEntry);
 
     StoreIngestionTaskTestConfig config = new StoreIngestionTaskTestConfig(Collections.singleton(PARTITION_FOO), () -> {
       TestUtils.waitForNonDeterministicAssertion(
@@ -4894,6 +4895,7 @@ public void testStoreIngestionRecordTransformerError(AAConfig aaConfig) throws E
     SchemaEntry valueSchemaEntry = mock(SchemaEntry.class);
     when(valueSchemaEntry.getSchema()).thenReturn(myValueSchema);
     when(mockSchemaRepo.getValueSchema(eq(storeNameWithoutVersionInfo), anyInt())).thenReturn(valueSchemaEntry);
+    when(mockSchemaRepo.getSupersetOrLatestValueSchema(eq(storeNameWithoutVersionInfo))).thenReturn(valueSchemaEntry);
 
     StoreIngestionTaskTestConfig config = new StoreIngestionTaskTestConfig(Collections.singleton(PARTITION_FOO), () -> {
       TestUtils.waitForNonDeterministicAssertion(