patch 4.0

unittest/storage/tx/test_simple_tx_ctx.cpp

@@ -0,0 +1,192 @@
+/**
+ * Copyright (c) 2021 OceanBase
+ * OceanBase CE is licensed under Mulan PubL v2.
+ * You can use this software according to the terms and conditions of the Mulan PubL v2.
+ * You may obtain a copy of Mulan PubL v2 at:
+ *          http://license.coscl.org.cn/MulanPubL-2.0
+ * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+ * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+ * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+ * See the Mulan PubL v2 for more details.
+ */
+
+#include <gtest/gtest.h>
+#include <vector>
+#include "storage/tx/ob_mock_tx_ctx.h"
+
+namespace oceanbase
+{
+using namespace ::testing;
+using namespace transaction;
+using namespace share;
+
+namespace unittest
+{
+class TestMockObTxCtx : public ::testing::Test
+{
+protected:
+  virtual void SetUp() override
+  {
+    mailbox_mgr_.reset();
+  }
+  virtual void TearDown() override
+  {
+    mailbox_mgr_.reset();
+  }
+  int check_mail(ObMailBox<ObTxMsg> &mailbox,
+                 int64_t from,
+                 int64_t to,
+                 int64_t type)
+  {
+    int ret = OB_SUCCESS;
+    ObMail<ObTxMsg> mail;
+    EXPECT_EQ(OB_SUCCESS, mailbox.fetch_mail(mail));
+
+    EXPECT_EQ(from, mail.from_);
+    EXPECT_EQ(to, mail.to_);
+    EXPECT_EQ(type, mail.mail_->get_msg_type());
+
+    switch (type) {
+    case TX_COMMIT_RESP: {
+      const ObTxCommitRespMsg *tx_commit_resp = dynamic_cast<const ObTxCommitRespMsg*>(mail.mail_);
+      EXPECT_EQ(OB_SUCCESS, tx_commit_resp->ret_);
+      TRANS_LOG(INFO, "check tx commit resp", K(mail), KPC(tx_commit_resp));
+      break;
+    }
+    default: {
+      TRANS_LOG(ERROR, "invalid msg type", K(mail));
+      ret = OB_TRANS_INVALID_STATE;
+      break;
+    }
+    }
+
+    return ret;
+  }
+
+  int build_scheduler_mailbox()
+  {
+    int ret = OB_SUCCESS;
+
+    if (OB_FAIL(mailbox_mgr_.register_mailbox(scheduler_addr_,
+                                              scheduler_mailbox_,
+                                              NULL))) {
+      TRANS_LOG(ERROR, "mock ctx register mailbox failed");
+    }
+
+    return ret;
+  }
+
+public:
+  ObMailBoxMgr<ObTxMsg> mailbox_mgr_;
+  int64_t scheduler_addr_ = 0;
+  ObMailBox<ObTxMsg> scheduler_mailbox_;
+};
+
+TEST_F(TestMockObTxCtx, test_simple_tx_ctx1)
+{
+  int64_t ls_id_gen = ObLSID::MIN_USER_LS_ID;
+  MockObTxCtx ctx1;
+  MockObTxCtx ctx2;
+  ObTransID trans_id1(1);
+  ObLSID ls_id0(++ls_id_gen);
+  ObLSID ls_id1(++ls_id_gen);
+  ObLSID ls_id2(++ls_id_gen);
+
+  ObTxData data1;
+  ObTxData data2;
+  ctx1.change_to_leader();
+  ctx2.change_to_leader();
+  EXPECT_EQ(OB_SUCCESS, ctx1.init(ls_id1, trans_id1, nullptr, &data1, &mailbox_mgr_));
+  EXPECT_EQ(OB_SUCCESS, ctx2.init(ls_id2, trans_id1, nullptr, &data2, &mailbox_mgr_));
+
+  std::vector<ObLSID> participants;
+  participants.push_back(ls_id1);
+  participants.push_back(ls_id2);
+
+  EXPECT_EQ(OB_SUCCESS, build_scheduler_mailbox());
+
+  ObTxCommitMsg tx_commit_msg;
+  MockObTxCtx::mock_tx_commit_msg(trans_id1,
+                                  ls_id1,
+                                  participants,
+                                  tx_commit_msg);
+  ObMail<ObTxMsg> tx_commit_mail;
+  EXPECT_EQ(OB_SUCCESS, tx_commit_mail.init(scheduler_addr_ /*from*/,
+                                            ctx1.get_mailbox_addr() /*to*/,
+                                            sizeof(ObTxCommitMsg),
+                                            tx_commit_msg));
+
+  // ========== Two Phase Commit prepare Phase ==========
+  // ctx1 start to commit
+  mailbox_mgr_.send_to_head(tx_commit_mail, tx_commit_mail.to_);
+  EXPECT_EQ(OB_SUCCESS, ctx1.handle());
+
+    // ctx2 handle prepare request
+  EXPECT_EQ(OB_SUCCESS, ctx2.handle());
+  // ctx2 handle prepare request
+  EXPECT_EQ(OB_SUCCESS, ctx2.apply());
+  // ctx1 handle prepare response
+  EXPECT_EQ(OB_SUCCESS, ctx1.handle());
+  // ctx1 apply prepare log
+  EXPECT_EQ(OB_SUCCESS, ctx1.apply());
+
+  // TODO shanyan.g
+  /*
+  // ========== Two Phase Commit pre commit Phase ======
+  // ctx2 handle pre commit request
+  EXPECT_EQ(OB_SUCCESS, ctx2.handle());
+  // ctx1 handle pre commit response
+  EXPECT_EQ(OB_SUCCESS, ctx1.handle());
+  EXPECT_EQ(OB_SUCCESS, ctx1.handle());
+  */
+
+  //EXPECT_EQ(OB_SUCCESS, check_mail(scheduler_mailbox_,
+  //                                 ctx1.get_mailbox_addr() /*from*/,
+  //                                 scheduler_addr_,
+  //                                 TX_COMMIT_RESP));
+
+  /*
+  // ========== Two Phase Commit commit Phase ==========
+  // ctx2 handle commit request
+  EXPECT_EQ(OB_SUCCESS, ctx2.handle());
+  // ctx2 apply commit log
+  EXPECT_EQ(OB_SUCCESS, ctx2.apply());
+  // ctx1 handle commit response
+  EXPECT_EQ(OB_SUCCESS, ctx1.handle());
+  // ctx1 apply commit log
+  EXPECT_EQ(OB_SUCCESS, ctx1.apply());
+
+  // ========== Two Phase Commit clear Phase ==========
+  // ctx2 handle clear request
+  EXPECT_EQ(OB_SUCCESS, ctx2.handle());
+  // ctx2 apply clear log
+  EXPECT_EQ(OB_SUCCESS, ctx2.apply());
+  // ctx1 apply clear log
+  EXPECT_EQ(OB_SUCCESS, ctx1.apply());
+  */
+
+  // ========== Check Test Valid ==========
+  // EXPECT_EQ(true, ctx1.check_status_valid(true/*should commit*/));
+  // EXPECT_EQ(true, ctx2.check_status_valid(true/*should commit*/));
+}
+
+}
+
+namespace transaction
+{
+  void ObTransCtx::after_unlock(CtxLockArg &) 
+  {
+  }
+}
+
+}
+
+int main(int argc, char **argv)
+{
+  system("rm -rf test_simple_tx_ctx.log*");
+  OB_LOGGER.set_file_name("test_simple_tx_ctx.log");
+  OB_LOGGER.set_log_level("INFO");
+  STORAGE_LOG(INFO, "begin unittest: test simple mock ob tx ctx");
+  ::testing::InitGoogleTest(&argc, argv);
+  return RUN_ALL_TESTS();
+}