test again

vesoft-inc · Nov 26, 2021 · 9503f68 · 9503f68
1 parent d1ed559
commit 9503f68
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 1 deletion.
diff --git a/include/nebula/sclient/StorageClient.h b/include/nebula/sclient/StorageClient.h
@@ -76,6 +76,8 @@ class StorageClient {
                                 bool onlyLatestVersion = false,
                                 bool enableReadFromFollower = true);  // plato needed
 
+  MetaClient* getMetaClient() { return mClient_.get(); }
+
  private:
   std::pair<bool, storage::cpp2::ScanEdgeResponse> doScanEdge(
       const storage::cpp2::ScanEdgeRequest& req);

diff --git a/src/sclient/tests/StorageClientTest.cpp b/src/sclient/tests/StorageClientTest.cpp
@@ -44,11 +44,34 @@ class StorageClientTest : public ClientTest {
         "'103'->'201':(43), '201'->'202':(56), '202'->'203':(-13), '203'->'301':(431), "
         "'301'->'302':(457)");
     ASSERT_EQ(result3.errorCode, nebula::ErrorCode::SUCCEEDED)
-        << (result2.errorMsg ? *result2.errorMsg : "");
+        << (result3.errorMsg ? *result3.errorMsg : "");
 
     ::sleep(10);
   }
 
+  static void runOnce(nebula::MetaClient &c) {
+    auto ret = c.getSpaceIdByNameFromCache("storage_client_test");
+    ASSERT_TRUE(ret.first);
+    nebula::GraphSpaceID spaceId = ret.second;
+    LOG(INFO) << "spaceId of nba: " << spaceId;
+    EXPECT_GT(spaceId, 0);
+
+    auto ret2 = c.getEdgeTypeByNameFromCache(spaceId, "like");
+    ASSERT_TRUE(ret2.first);
+    nebula::EdgeType edgeType = ret2.second;
+    LOG(INFO) << "edgeType of like: " << edgeType;
+    EXPECT_GT(edgeType, 0);
+
+    auto ret3 = c.getPartsFromCache(spaceId);
+    ASSERT_TRUE(ret3.first);
+    auto parts = ret3.second;
+    EXPECT_EQ(parts, (std::vector<nebula::PartitionID>{1}));
+
+    auto ret4 = c.getPartLeaderFromCache(spaceId, 1);
+    ASSERT_TRUE(ret4.first);
+    EXPECT_EQ(ret4.second, nebula::HostAddr(kServerHost, 9779));
+  }
+
   static void runGetParts(nebula::StorageClient &c) {
     auto parts = c.getParts("storage_client_test");
     EXPECT_EQ(parts, (std::vector<nebula::PartitionID>{1}));
@@ -113,6 +136,9 @@ TEST_F(StorageClientTest, Basic) {
   LOG(INFO) << "Prepare data.";
   prepare();
   nebula::StorageClient c({kServerHost ":9559"});
+  auto* m = c.getMetaClient();
+  LOG(INFO) << "Testing run once of meta client";
+  runOnce(*m);
   LOG(INFO) << "Testing run get parts.";
   runGetParts(c);
   LOG(INFO) << "Testing run scan edge with part.";