Skip to content

Commit 06f3f12

Browse files
committed
Fix test
Signed-off-by: Felix Wang <[email protected]>
1 parent f2c6595 commit 06f3f12

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

serving/src/test/java/feast/serving/service/OnlineServingServiceTest.java

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ public class OnlineServingServiceTest {
5353
@Mock CachedSpecService specService;
5454
@Mock Tracer tracer;
5555
@Mock OnlineRetriever retrieverV2;
56-
private String featureTransformationServer;
56+
private String transformationServiceEndpoint;
5757

5858
private OnlineServingServiceV2 onlineServingServiceV2;
5959

@@ -63,12 +63,12 @@ public class OnlineServingServiceTest {
6363
@Before
6464
public void setUp() {
6565
initMocks(this);
66+
CoreFeatureSpecRetriever coreFeatureSpecRetriever = new CoreFeatureSpecRetriever(specService);
67+
OnlineTransformationService onlineTransformationService =
68+
new OnlineTransformationService(transformationServiceEndpoint, coreFeatureSpecRetriever);
6669
onlineServingServiceV2 =
6770
new OnlineServingServiceV2(
68-
retrieverV2,
69-
tracer,
70-
new CoreFeatureSpecRetriever(specService),
71-
featureTransformationServer);
71+
retrieverV2, tracer, coreFeatureSpecRetriever, onlineTransformationService);
7272

7373
mockedFeatureRows = new ArrayList<>();
7474
mockedFeatureRows.add(

0 commit comments

Comments
 (0)