check condition

pmem · Jul 18, 2024 · bb93bba · bb93bba
1 parent d287953
commit bb93bba
Show file tree

Hide file tree

Showing 5 changed files with 31 additions and 11 deletions.
diff --git a/src/tests/ras/unsafe_shutdown/local_move_tests.cc b/src/tests/ras/unsafe_shutdown/local_move_tests.cc
@@ -37,14 +37,7 @@ std::ostream& operator<<(std::ostream& stream, move_param const& m) {
  return stream;
 }
 
-/* Change the status of the shutdown state */
-void set_sds_at_create_func(bool state) {
- int ret = pmemobj_ctl_set(NULL, "sds.at_create", &state);
- if (ret) {
- std::cerr << "Failed to set sds.at_create: " << pmemobj_errormsg() << std::endl;
- exit(1);
- }
-}
+bool pmem;
 
 std::vector<move_param> GetMoveParams() {
  std::vector<move_param> ret_vec;
@@ -60,6 +53,7 @@ std::vector<move_param> GetMoveParams() {
  tc.enough_dimms = true;
  tc.src_pool_dir = unsafe_dn[0].GetTestDir();
  tc.dest_pool_dir = test_phase.GetTestDir();
+ tc.pmem = true;
  } else {
  tc.enough_dimms = false;
  }
@@ -74,6 +68,7 @@ std::vector<move_param> GetMoveParams() {
  tc.enough_dimms = true;
  tc.src_pool_dir = test_phase.GetTestDir();
  tc.dest_pool_dir = unsafe_dn[0].GetTestDir();
+ tc.pmem = false;
  } else {
  tc.enough_dimms = false;
  }
@@ -84,6 +79,7 @@ std::vector<move_param> GetMoveParams() {
  {
  move_param tc;
  tc.description = "from unsafely shutdown DIMM to safely shutdown DIMM";
+ tc.pmem = true;
  if (unsafe_dn.size() >= 1 && safe_dn.size() >= 1) {
  tc.enough_dimms = true;
  tc.src_pool_dir = unsafe_dn[0].GetTestDir();
@@ -98,6 +94,7 @@ std::vector<move_param> GetMoveParams() {
  {
  move_param tc;
  tc.description = "from unsafely shutdown DIMM to unsafely shutdown DIMM";
+ tc.pmem = true;
  if (unsafe_dn.size() >= 2) {
  tc.enough_dimms = true;
  tc.src_pool_dir = unsafe_dn[1].GetTestDir();
@@ -112,6 +109,7 @@ std::vector<move_param> GetMoveParams() {
  {
  move_param tc;
  tc.description = "From safely shutdown DIMM to non-pmem";
+ tc.pmem = true;
  if (safe_dn.size() >= 1) {
  tc.enough_dimms = true;
  tc.src_pool_dir = safe_dn[0].GetTestDir();
@@ -126,6 +124,7 @@ std::vector<move_param> GetMoveParams() {
  {
  move_param tc;
  tc.description = "From non-pmem to safely shutdown DIMM";
+ tc.pmem = false;
  if (safe_dn.size() >= 1) {
  tc.enough_dimms = true;
  tc.src_pool_dir = test_phase.GetTestDir();
@@ -145,6 +144,7 @@ void MovePoolClean::SetUp() {
  << "Insufficient number of DIMMs to run this test";
  src_pool_path_ = param.src_pool_dir + GetNormalizedTestName() + "_pool";
  dest_pool_path_ = param.dest_pool_dir + GetNormalizedTestName() + "_pool";
+ pmem = param.pmem;
 }
 
 /**
@@ -165,12 +165,14 @@ void MovePoolClean::SetUp() {
  */
 TEST_P(MovePoolClean, TC_MOVE_POOL_CLEAN_phase_1) {
  /* Step1 */
- set_sds_at_create_func(false);
+ if (pmem == false)
+ set_sds_at_create_func(false);
  pop_ = pmemobj_create(src_pool_path_.c_str(), nullptr, PMEMOBJ_MIN_POOL, 0644);
  ASSERT_TRUE(pop_ != nullptr) << "Pool creating failed. Errno: " << errno
  << std::endl
  << pmemobj_errormsg();
- set_sds_at_create_func(true);
+ if (pmem == false)
+ set_sds_at_create_func(true);
 
  /* Step2 */
  ObjData<int> pd{pop_};
@@ -227,9 +229,13 @@ void MovePoolDirty::SetUp() {
  */
 TEST_P(MovePoolDirty, TC_MOVE_POOL_DIRTY_phase_1) {
  /* Step1 */
+ if (pmem == false)
+ set_sds_at_create_func(false);
  pop_ = pmemobj_create(src_pool_path_.c_str(), nullptr, PMEMOBJ_MIN_POOL, 0644);
  ASSERT_TRUE(pop_ != nullptr) << "Pool creating failed" << std::endl
  << pmemobj_errormsg();
+ if (pmem == false)
+ set_sds_at_create_func(true);
 
  /* Step2 */
  ObjData<int> pd{pop_};

diff --git a/src/tests/ras/unsafe_shutdown/local_move_tests.h b/src/tests/ras/unsafe_shutdown/local_move_tests.h
@@ -1,5 +1,5 @@
 /*
- * Copyright 2018, Intel Corporation
+ * Copyright 2018-2024, Intel Corporation
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
@@ -40,6 +40,7 @@ struct move_param {
  std::string src_pool_dir;
  std::string dest_pool_dir;
  bool enough_dimms;
+ bool pmem;
 };
 
 std::ostream& operator<<(std::ostream& stream, move_param const& m);

diff --git a/src/tests/ras/unsafe_shutdown/local_replicas_tests.cc b/src/tests/ras/unsafe_shutdown/local_replicas_tests.cc
@@ -71,10 +71,12 @@ TEST_P(SyncLocalReplica, TC_SYNC_LOCAL_REPLICA_phase_1) {
  ASSERT_TRUE(p_mgmt.PoolsetFileExists(ps))
  << "Poolset file " << ps.GetFullPath() << " does not exist";
 
+ set_sds_at_create_func(false);
  pop_ = pmemobj_create(ps.GetFullPath().c_str(), nullptr, 0, 0644);
  ASSERT_TRUE(pop_ != nullptr)
  << "Error while creating the pool. Errno:" << errno << std::endl
  << pmemobj_errormsg();
+ set_sds_at_create_func(true);
 
  /* Step2 */
  ObjData<int> pd{pop_};

diff --git a/src/tests/ras/unsafe_shutdown/unsafe_shutdown.cc b/src/tests/ras/unsafe_shutdown/unsafe_shutdown.cc
@@ -86,3 +86,12 @@ int UnsafeShutdown::PmempoolRepair(std::string pool_file_path) const {
 
  return pmempool_check_end(ppc);
 }
+
+/* Change the status of the shutdown state */
+void set_sds_at_create_func(bool state) {
+ int ret = pmemobj_ctl_set(NULL, "sds.at_create", &state);
+ if (ret) {
+ std::cerr << "Failed to set sds.at_create: " << pmemobj_errormsg() << std::endl;
+ exit(1);
+ }
+}
diff --git a/src/tests/ras/unsafe_shutdown/unsafe_shutdown.h b/src/tests/ras/unsafe_shutdown/unsafe_shutdown.h
@@ -78,6 +78,8 @@ class UnsafeShutdown : public ::testing::Test {
  return test_phase_.GetTestDir() + GetNormalizedTestName() + "_passed";
  }
  void StampPassedResult() const;
+
+ void set_sds_at_create_func(bool state)
 };
 
 #endif // UNSAFE_SHUTDOWN_H