From fd0df49008e15060a20c964a3563965068afae17 Mon Sep 17 00:00:00 2001 From: Caio Camatta Date: Wed, 28 Feb 2024 15:25:30 -0500 Subject: [PATCH] Fix stub --- .../src/test/scala/ai/chronon/online/FetcherBaseTest.scala | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/online/src/test/scala/ai/chronon/online/FetcherBaseTest.scala b/online/src/test/scala/ai/chronon/online/FetcherBaseTest.scala index bc7b60d1b..6eda6c8ac 100644 --- 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])