test improve

abtink · Sep 16, 2023 · 58f4365 · 58f4365
1 parent 98e4881
commit 58f4365
Showing 1 changed file with 20 additions and 3 deletions.
diff --git a/tests/mdns/main.cpp b/tests/mdns/main.cpp
@@ -325,6 +325,23 @@ void PublishServiceSubTypes(void *aContext)
         [](otbrError aError) { SuccessOrDie(aError, "ServiceWithSubTypes._meshcop._udp"); });
 }
 
+typedef void (*TestRunner)(void *aContext, Mdns::Publisher::State aState);
+
+otbrError Test(TestRunner aTestRunner)
+{
+    otbrError error = OTBR_ERROR_NONE;
+
+    Mdns::Publisher *pub = Mdns::Publisher::Create(
+        [aTestRunner](Mdns::Publisher::State aState) { aTestRunner(&sContext, aState); });
+    sContext.mPublisher = pub;
+    SuccessOrExit(error = pub->Start());
+    RunMainloop();
+
+exit:
+    Mdns::Publisher::Destroy(pub);
+    return error;
+}
+
 otbrError TestSingleServiceWithCustomHost(void)
 {
     otbrError error = OTBR_ERROR_NONE;
@@ -547,13 +564,13 @@ int main(int argc, char *argv[])
         switch (argv[1][1])
         {
         case 'c':
-            ret = TestSingleServiceWithCustomHost();
+            ret = Test(PublishSingleServiceWithCustomHost);
             break;
         case 'e':
-            ret = TestSingleServiceWithEmptyName();
+            ret = Test(PublishSingleServiceWithEmptyName);
             break;
         default:
-            ret = TestSingleService();
+            ret = Test(PublishSingleService);
             break;
         }
         break;