From 34cd75f77d0ca94dbdba4e6cc9451fe4d33e78b3 Mon Sep 17 00:00:00 2001
From: lichao <lichao@aiotlink.com>
Date: 星期三, 19 五月 2021 19:14:13 +0800
Subject: [PATCH] add api BHQueryProcs.

---
 utest/utest.cpp |   39 +++++++++++++++++++++++++++------------
 1 files changed, 27 insertions(+), 12 deletions(-)

diff --git a/utest/utest.cpp b/utest/utest.cpp
index b2de97f..7cb9587 100644
--- a/utest/utest.cpp
+++ b/utest/utest.cpp
@@ -1,16 +1,33 @@
 #include "center.h"
 #include "defs.h"
+#include "log.h"
 #include "util.h"
 #include <atomic>
-#include <boost/uuid/uuid_generators.hpp>
-#include <boost/uuid/uuid_io.hpp>
 #include <condition_variable>
 #include <stdio.h>
 #include <string>
 #include <thread>
 #include <vector>
 
+namespace
+{
+bool InitLog()
+{
+	ns_log::AddLog("/tmp/bhshmq_test.log", true, true);
+	ns_log::ResetLogLevel(ns_log::LogLevel::debug);
+	return true;
+}
+static bool g_test_init_log = InitLog();
+} // namespace
+
+using namespace bhome_shm;
 using namespace bhome_msg;
+
+SharedMemory &TestShm()
+{
+	static SharedMemory shm("utest_0", 1024 * 1024 * 512);
+	return shm;
+}
 
 template <class A, class B>
 struct IsSameType {
@@ -84,13 +101,12 @@
 
 BOOST_AUTO_TEST_CASE(PubSubTest)
 {
-	const std::string shm_name("ShmPubSub");
-	ShmRemover auto_remove(shm_name); //remove twice? in case of killed?
-	SharedMemory shm(shm_name, 1024 * 1024 * 50);
-	DEFER1(shm.Remove());
+	SharedMemory &shm = TestShm();
+	GlobalInit(shm);
+
 	auto Avail = [&]() { return shm.get_free_memory(); };
 	auto init_avail = Avail();
-	int *flag = shm.find_or_construct<int>("flag")(123);
+	int *flag = shm.FindOrCreate<int>("flag", 123);
 	printf("flag = %d\n", *flag);
 	++*flag;
 	const std::string sub_proc_id = "subscriber";
@@ -196,13 +212,12 @@
 
 BOOST_AUTO_TEST_CASE(ReqRepTest)
 {
-	const std::string shm_name("ShmReqRep");
-	ShmRemover auto_remove(shm_name);
-	SharedMemory shm(shm_name, 1024 * 1024 * 512);
+	SharedMemory &shm = TestShm();
+	GlobalInit(shm);
 
 	auto Avail = [&]() { return shm.get_free_memory(); };
 	auto init_avail = Avail();
-	int *flag = shm.find_or_construct<int>("flag")(123);
+	int *flag = shm.FindOrCreate<int>("flag", 123);
 	printf("flag = %d\n", *flag);
 	++*flag;
 
@@ -233,7 +248,7 @@
 		boost::timer::auto_cpu_timer timer;
 		for (int i = 0; i < nreq; ++i) {
 			std::string msg_id;
-			if (!client.ClientAsyncRequest(req, msg_id)) {
+			if (!client.ClientAsyncRequest(BHAddress(), req, msg_id)) {
 				printf("client request failed\n");
 				++count;
 			}

--
Gitblit v1.8.0