added test to showcase the problem

mochi-hpc · Jul 31, 2024 · c919055 · c919055
1 parent 4cdf588
commit c919055
Showing 1 changed file with 64 additions and 0 deletions.
diff --git a/tests/unit-tests/margo-init.c b/tests/unit-tests/margo-init.c
@@ -1,5 +1,7 @@
 
 #include <margo.h>
+#include <stdlib.h>
+#include <unistd.h>
 #include "helper-server.h"
 #include "munit/munit.h"
 
@@ -184,6 +186,67 @@ static MunitResult ref_incr_and_release(const MunitParameter params[], void* dat
     return MUNIT_OK;
 }
 
+static void kill_test(void* args) {
+    volatile int* x = (int*)args;
+    sleep(1);
+    if(*x == 1) return;
+    else {
+        x = NULL;
+        *x = 1; // segfault
+    }
+}
+
+void my_func(void*) {}
+
+static MunitResult multiple_pools_progress_loop(const MunitParameter params[], void* data)
+{
+    const char* config = "{\n"
+        "\"argobots\": {"
+            "\"pools\": ["
+                "{\"name\":\"__primary__\",\"access\":\"mpmc\",\"kind\":\"fifo_wait\"},"
+                "{\"name\":\"p1\",\"access\":\"mpmc\",\"kind\":\"fifo_wait\"},"
+                "{\"name\":\"p2\",\"access\":\"mpmc\",\"kind\":\"fifo_wait\"}"
+            "],"
+            "\"xstreams\": ["
+                "{\"name\":\"__primary__\","
+                 "\"scheduler\":{"
+                     "\"pools\":[\"__primary__\",\"p1\"],"
+                     "\"type\":\"basic_wait\""
+                   "}"
+                "},"
+                "{\"name\":\"es1\","
+                 "\"scheduler\":{"
+                     "\"pools\":[\"p2\"],"
+                     "\"type\":\"basic_wait\""
+                   "}"
+                "}"
+            "]"
+        "},"
+        "\"progress_pool\":\"p1\","
+        "\"rpc_pool\":\"p1\""
+    "}";
+
+    struct margo_init_info info = MARGO_INIT_INFO_INITIALIZER;
+    info.json_config = config;
+    margo_instance_id mid = margo_init_ext("na+sm", MARGO_SERVER_MODE, &info);
+
+    struct margo_pool_info p2 = {0};
+    margo_find_pool_by_name(mid, "p2", &p2);
+
+    ABT_thread ult, killer;
+    volatile int x = 0;
+    ABT_thread_create(p2.pool, kill_test, (void*)&x, ABT_THREAD_ATTR_NULL, &killer);
+    ABT_thread_create(p2.pool, my_func, NULL, ABT_THREAD_ATTR_NULL, &ult);
+    ABT_thread_join(ult);
+    x = 1;
+    ABT_thread_free(&ult);
+    ABT_thread_join(killer);
+    ABT_thread_free(&killer);
+
+    margo_finalize(mid);
+    return MUNIT_OK;
+}
+
 static char* protocol_params[] = {
     "na+sm", NULL
 };
@@ -208,6 +271,7 @@ static MunitTest tests[] = {
     { "/init-cycle-server", init_cycle_server, test_context_setup, test_context_tear_down, MUNIT_TEST_OPTION_NONE, test_params},
     { "/finalize-and-wait", finalize_and_wait, test_context_setup, test_context_tear_down, MUNIT_TEST_OPTION_NONE, test_params},
     { "/ref-incr-and-release", ref_incr_and_release, test_context_setup, test_context_tear_down, MUNIT_TEST_OPTION_NONE, test_params},
+    { "/multiple-pools-progress-loop", multiple_pools_progress_loop, test_context_setup, test_context_tear_down, MUNIT_TEST_OPTION_NONE, NULL},
     { NULL, NULL, NULL, NULL, MUNIT_TEST_OPTION_NONE, NULL }
 };