Function TEST_F

src/tabletnode/test/remote_tabletnode_test.cc:57–112

}

TEST_F(RemoteTabletNodeTest, LoadTablet) {
  std::unique_ptr<LoadTabletRequest> request(new LoadTabletRequest);
  request->set_path("test/tablet00000001");
  std::unique_ptr<LoadTabletResponse> response(new LoadTabletResponse);
  std::unique_ptr<MockClosure> done(new MockClosure);
  std::unique_ptr<google::protobuf::RpcController> controller(new sofa::pbrpc::RpcController);
  remote_ts_->LoadTablet(controller.get(), request.get(), response.get(), done.get());
  EXPECT_EQ(remote_ts_->ctrl_thread_pool_->queue_.size(), 1);
  EXPECT_EQ(remote_ts_->ctrl_thread_pool_->PendingNum(), 1);
  EXPECT_EQ(remote_ts_->tablets_ctrl_status_.size(), 1);
  EXPECT_EQ(remote_ts_->tablets_ctrl_status_[request->path()],
            RemoteTabletNode::TabletCtrlStatus::kCtrlWaitLoad);

  std::unique_ptr<LoadTabletRequest> request4(new LoadTabletRequest);
  request4->set_path("test/tablet00000001");
  std::unique_ptr<LoadTabletResponse> response4(new LoadTabletResponse);
  std::unique_ptr<MockClosure> done4(new MockClosure);
  std::unique_ptr<google::protobuf::RpcController> controller4(new sofa::pbrpc::RpcController);
  remote_ts_->LoadTablet(controller4.get(), request4.get(), response4.get(), done4.get());
  EXPECT_EQ(remote_ts_->ctrl_thread_pool_->queue_.size(), 1);
  EXPECT_EQ(remote_ts_->ctrl_thread_pool_->PendingNum(), 1);
  EXPECT_EQ(remote_ts_->tablets_ctrl_status_.size(), 1);
  EXPECT_EQ(remote_ts_->tablets_ctrl_status_[request4->path()],
            RemoteTabletNode::TabletCtrlStatus::kCtrlWaitLoad);
  EXPECT_EQ(remote_ts_->lightweight_ctrl_thread_pool_->queue_.size(), 1);

  std::unique_ptr<LoadTabletRequest> request2(new LoadTabletRequest);
  request2->set_path("test/tablet00000002");
  std::unique_ptr<LoadTabletResponse> response2(new LoadTabletResponse);
  std::unique_ptr<MockClosure> done2(new MockClosure);
  std::unique_ptr<google::protobuf::RpcController> controller2(new sofa::pbrpc::RpcController);
  remote_ts_->LoadTablet(controller2.get(), request2.get(), response2.get(), done2.get());
  EXPECT_EQ(remote_ts_->ctrl_thread_pool_->queue_.size(), 2);
  EXPECT_EQ(remote_ts_->ctrl_thread_pool_->PendingNum(), 2);
  EXPECT_EQ(remote_ts_->tablets_ctrl_status_.size(), 2);
  EXPECT_EQ(remote_ts_->tablets_ctrl_status_[request2->path()],
            RemoteTabletNode::TabletCtrlStatus::kCtrlWaitLoad);

  std::unique_ptr<LoadTabletRequest> request3(new LoadTabletRequest);
  request3->set_path("test/tablet00000003");
  std::unique_ptr<LoadTabletResponse> response3(new LoadTabletResponse);
  std::unique_ptr<MockClosure> done3(new MockClosure);
  std::unique_ptr<google::protobuf::RpcController> controller3(new sofa::pbrpc::RpcController);
  remote_ts_->LoadTablet(controller3.get(), request3.get(), response3.get(), done3.get());
  EXPECT_EQ(remote_ts_->ctrl_thread_pool_->queue_.size(), 2);
  EXPECT_EQ(remote_ts_->ctrl_thread_pool_->PendingNum(), 2);
  EXPECT_EQ(remote_ts_->tablets_ctrl_status_.size(), 2);
  EXPECT_TRUE(remote_ts_->tablets_ctrl_status_.find(request3->path()) ==
              remote_ts_->tablets_ctrl_status_.end());
  EXPECT_EQ(response3->status(), kTabletNodeIsBusy);

  remote_ts_->ctrl_thread_pool_->Start();
  remote_ts_->ctrl_thread_pool_->Stop(true);
  EXPECT_EQ(remote_ts_->ctrl_thread_pool_->PendingNum(), 0);
  EXPECT_EQ(remote_ts_->tablets_ctrl_status_.size(), 0);
}

TEST_F(RemoteTabletNodeTest, UnloadTablet) {