add timestamp test,fix Random range error

OpenAtomFoundation · Nov 19, 2023 · 6889cf7 · 6889cf7
1 parent 28bd45b
commit 6889cf7
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 1 deletion.
diff --git a/src/pstd/pstd_util.cc b/src/pstd/pstd_util.cc
@@ -16,7 +16,7 @@ void InitRandom() {
   gen.seed(rd());
 }
 
-int RandomInt(int max) { return RandomInt(0, max + 1); }
+int RandomInt(int max) { return RandomInt(0, max); }
 
 int RandomInt(int min, int max) {
   std::uniform_int_distribution<> dist(min, max);

diff --git a/src/pstd/tests/pstd_util_test.cc b/src/pstd/tests/pstd_util_test.cc
@@ -48,6 +48,30 @@ TEST(UtilTest, RandomPermNegative) {
   ASSERT_EQ(perm.size(), 0);
 }
 
+TEST(UtilTest, UnixTimestamp) {
+  int64_t timestamp = pstd::UnixTimestamp();
+  std::cout << timestamp << std::endl;
+  ASSERT_GT(timestamp, 0);
+}
+
+TEST(UtilTest, UnixMilliTimestamp) {
+  int64_t timestamp = pstd::UnixMilliTimestamp();
+  std::cout << timestamp << std::endl;
+  ASSERT_GT(timestamp, 0);
+}
+
+TEST(UtilTest, UnixMicroTimestamp) {
+  int64_t timestamp = pstd::UnixMicroTimestamp();
+  std::cout << timestamp << std::endl;
+  ASSERT_GT(timestamp, 0);
+}
+
+TEST(UtilTest, UnixNanoTimestamp) {
+  int64_t timestamp = pstd::UnixNanoTimestamp();
+  std::cout << timestamp << std::endl;
+  ASSERT_GT(timestamp, 0);
+}
+
 int main(int argc, char** argv) {
   pstd::InitRandom();
   ::testing::InitGoogleTest(&argc, argv);