Fix stub

airbnb · Feb 28, 2024 · a7c7618 · a7c7618
1 parent 2048bb2
commit a7c7618
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/online/src/test/scala/ai/chronon/online/FetcherBaseTest.scala b/online/src/test/scala/ai/chronon/online/FetcherBaseTest.scala
@@ -36,7 +36,7 @@ import scala.concurrent.{Await, ExecutionContext, Future}
 import scala.util.{Failure, Success}
 import scala.util.Try
 
-class FetcherBaseTest extends MockitoSugar with Matchers with MockitoHelper {
+class FetcherBaseTest extends MockitoSugar with Matchers {
   val GroupBy = "relevance.short_term_user_features"
   val Column = "pdp_view_count_14d"
   val GuestKey = "guest"
@@ -177,8 +177,8 @@ class FetcherBaseTest extends MockitoSugar with Matchers with MockitoHelper {
 
     val metaDataMock = mock[MetaData]
     val groupByOpsMock = mock[GroupByOps]
-    doReturn("test").when(metaDataMock).name
-    doReturn(metaDataMock).when(groupByOpsMock).metaData
+    metaDataMock.name = "test"
+    groupByOpsMock.metaData = metaDataMock
     doReturn(groupByOpsMock).when(oldServingInfo).groupByOps
 
     val cachedBatchResponses = BatchResponses(mock[FinalBatchIr])