add test for query port message

ryzom · Jun 26, 2024 · cfb0880 · cfb0880
1 parent 8fe1b01
commit cfb0880
Showing 1 changed file with 24 additions and 1 deletion.
diff --git a/nelns/naming_service/tests/naming_service.it.cpp b/nelns/naming_service/tests/naming_service.it.cpp
@@ -127,12 +127,13 @@ class CNamingServiceIT : public testing::Test
 	CCallbackClient client;
 	CNamingService namingService;
 	int port = 50000;
+	int minPort = 51000;
 
 	void SetUp() override
 	{
 		CVar basePort;
 		basePort.Type = NLMISC::CConfigFile::CVar::T_INT;
-		basePort.setAsInt(51000);
+		basePort.setAsInt(minPort);
 		namingService.ConfigFile.insertVar("BasePort", basePort);
 
 		CVar uniqueOnShardServices;
@@ -213,6 +214,7 @@ TEST_F(CNamingServiceIT, shouldUpdateServiceRegistry)
 
 	client.send(msgout);
 	client.flush();
+	client.update2(-1, 200);
 	namingService.update();
 	client.update2(-1, 200);
 
@@ -225,3 +227,24 @@ TEST_F(CNamingServiceIT, shouldUpdateServiceRegistry)
 				Field(&CServiceEntry::Addr, ElementsAre(Property(&CInetAddress::asString, "[::1]:12345")))
 	            )));
 }
+
+TEST_F(CNamingServiceIT, shouldAnswerToQueryPort)
+{
+	CMessage msgout("QP");
+
+	uint16 response(0);
+	TCallbackItem callbackArray[] = {
+		{ "QP", [&response](CMessage &msgin, TSockId from, CCallbackNetBase &netbase) {
+		     msgin.serial(response);
+		 } }
+	};
+	client.addCallbackArray(callbackArray, sizeof(callbackArray) / sizeof(callbackArray[0]));
+
+	client.send(msgout);
+	client.flush();
+	client.update2(-1, 200);
+	namingService.update();
+	client.update2(-1, 200);
+
+	EXPECT_THAT(response, Eq(minPort));
+}