src/test/test-bus-util.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2
   3 #include "bus-util.h"
   4 #include "log.h"
   5
   6 static void test_name_async(unsigned n_messages) {
   7         _cleanup_(sd_bus_flush_close_unrefp) sd_bus *bus = NULL;
   8         int r;
   9         unsigned i;
  10
  11         log_info("/* %s (%u) */", __func__, n_messages);
  12
  13         r = bus_open_system_watch_bind_with_description(&bus, "test-bus");
  14         if (r < 0) {
  15                 log_error_errno(r, "Failed to connect to bus: %m");
  16                 return;
  17         }
  18
  19         r = bus_request_name_async_may_reload_dbus(bus, NULL, "org.freedesktop.elogind.test-bus-util", 0, NULL);
  20         if (r < 0) {
  21                 log_error_errno(r, "Failed to request name: %m");
  22                 return;
  23         }
  24
  25         for (i = 0; i < n_messages; i++) {
  26                 r = sd_bus_process(bus, NULL);
  27                 log_debug("stage %u: sd_bus_process returned %d", i, r);
  28                 if (r < 0) {
  29                         log_notice_errno(r, "Processing failed: %m");
  30                         return;
  31                 }
  32
  33                 if (r > 0 && i + 1 < n_messages)
  34                         (void) sd_bus_wait(bus, USEC_PER_SEC / 3);
  35         }
  36 }
  37
  38 static int callback(sd_bus_message *m, void *userdata, sd_bus_error *ret_error) {
  39         return 1;
  40 }
  41
  42 static void destroy_callback(void *userdata) {
  43         int *n_called = userdata;
  44
  45         (*n_called) ++;
  46 }
  47
  48 static void test_destroy_callback(void) {
  49         _cleanup_(sd_bus_flush_close_unrefp) sd_bus *bus = NULL;
  50         sd_bus_slot *slot = NULL;
  51         sd_bus_destroy_t t;
  52
  53         int r, n_called = 0;
  54
  55         log_info("/* %s */", __func__);
  56
  57         r = bus_open_system_watch_bind_with_description(&bus, "test-bus");
  58         if (r < 0) {
  59                 log_error_errno(r, "Failed to connect to bus: %m");
  60                 return;
  61         }
  62
  63         r = sd_bus_request_name_async(bus, &slot, "org.freedesktop.elogind.test-bus-util", 0, callback, &n_called);
  64         assert(r == 1);
  65
  66         assert_se(sd_bus_slot_get_destroy_callback(slot, NULL) == 0);
  67         assert_se(sd_bus_slot_get_destroy_callback(slot, &t) == 0);
  68
  69         assert_se(sd_bus_slot_set_destroy_callback(slot, destroy_callback) == 0);
  70         assert_se(sd_bus_slot_get_destroy_callback(slot, NULL) == 1);
  71         assert_se(sd_bus_slot_get_destroy_callback(slot, &t) == 1);
  72         assert_se(t == destroy_callback);
  73
  74         /* Force cleanup so we can look at n_called */
  75         assert(n_called == 0);
  76         sd_bus_slot_unref(slot);
  77         assert(n_called == 1);
  78 }
  79
  80 int main(int argc, char **argv) {
  81         log_set_max_level(LOG_DEBUG);
  82         log_parse_environment();
  83         log_open();
  84
  85         test_name_async(0);
  86         test_name_async(20);
  87         test_destroy_callback();
  88
  89         return 0;
  90 }