From 775e950dbae0bc34339c91bd9def081c89f66308 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?J=C3=A1nos=20Benjamin=20Antal?= <benjamin.antal@memgraph.io>
Date: Mon, 16 Jan 2023 10:16:12 +0100
Subject: [PATCH] Update unit tests to test the new logic

---
 tests/unit/query_v2_create_node_multiframe.cpp | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/tests/unit/query_v2_create_node_multiframe.cpp b/tests/unit/query_v2_create_node_multiframe.cpp
index b04b00e96..014fff6ce 100644
--- a/tests/unit/query_v2_create_node_multiframe.cpp
+++ b/tests/unit/query_v2_create_node_multiframe.cpp
@@ -56,10 +56,9 @@ TEST(CreateNodeTest, CreateNodeCursor) {
   auto create_expand = plan::CreateNode(once_op, node);
   auto cursor = create_expand.MakeCursor(utils::NewDeleteResource());
   MockedRequestRouter router;
-  EXPECT_CALL(router, CreateVertices(testing::_))
-      .Times(1)
-      .WillOnce(::testing::Return(std::vector<msgs::CreateVerticesResponse>{}));
-  EXPECT_CALL(router, IsPrimaryKey(testing::_, testing::_)).WillRepeatedly(::testing::Return(true));
+  EXPECT_CALL(router, CreateVertices(_)).Times(1).WillOnce(Return(std::vector<msgs::CreateVerticesResponse>{}));
+  EXPECT_CALL(router, IsPrimaryLabel(_)).WillRepeatedly(Return(true));
+  EXPECT_CALL(router, IsPrimaryKey(_, _)).WillRepeatedly(Return(true));
   auto context = MakeContext(ast, symbol_table, &router, &id_alloc);
   auto multi_frame = CreateMultiFrame(context.symbol_table.max_position());
   cursor->PullMultiple(multi_frame, context);