src/libsystemd/sd-bus/test-bus-cleanup.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2
   3 #include <stdio.h>
   4
   5 #include "sd-bus.h"
   6
   7 #include "bus-internal.h"
   8 #include "bus-message.h"
   9 #include "bus-util.h"
  10 #include "tests.h"
  11
  12 static bool use_system_bus = false;
  13
  14 static void test_bus_new(void) {
  15         _cleanup_(sd_bus_unrefp) sd_bus *bus = NULL;
  16
  17         assert_se(sd_bus_new(&bus) == 0);
  18         assert_se(bus->n_ref == 1);
  19 }
  20
  21 static int test_bus_open(void) {
  22         _cleanup_(sd_bus_flush_close_unrefp) sd_bus *bus = NULL;
  23         int r;
  24
  25         r = sd_bus_open_user(&bus);
  26         if (IN_SET(r, -ECONNREFUSED, -ENOENT, -ENOMEDIUM)) {
  27                 r = sd_bus_open_system(&bus);
  28                 if (IN_SET(r, -ECONNREFUSED, -ENOENT))
  29                         return r;
  30                 use_system_bus = true;
  31         }
  32
  33         assert_se(r >= 0);
  34         assert_se(bus->n_ref >= 1); /* we send a hello message when opening, so the count is above 1 */
  35
  36         return 0;
  37 }
  38
  39 static void test_bus_new_method_call(void) {
  40         sd_bus *bus = NULL;
  41         _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL;
  42
  43         assert_se(use_system_bus ? sd_bus_open_system(&bus) >= 0 : sd_bus_open_user(&bus) >= 0);
  44
  45         assert_se(sd_bus_message_new_method_call(bus, &m, "a.service.name", "/an/object/path", "an.interface.name", "AMethodName") >= 0);
  46
  47         assert_se(m->n_ref == 1); /* We hold the only reference to the message */
  48         assert_se(bus->n_ref >= 2);
  49         sd_bus_flush_close_unref(bus);
  50         assert_se(m->n_ref == 1);
  51 }
  52
  53 static void test_bus_new_signal(void) {
  54         sd_bus *bus = NULL;
  55         _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL;
  56
  57         assert_se(use_system_bus ? sd_bus_open_system(&bus) >= 0 : sd_bus_open_user(&bus) >= 0);
  58
  59         assert_se(sd_bus_message_new_signal(bus, &m, "/an/object/path", "an.interface.name", "Name") >= 0);
  60
  61         assert_se(m->n_ref == 1); /* We hold the only reference to the message */
  62         assert_se(bus->n_ref >= 2);
  63         sd_bus_flush_close_unref(bus);
  64         assert_se(m->n_ref == 1);
  65 }
  66
  67 int main(int argc, char **argv) {
  68         test_setup_logging(LOG_INFO);
  69
  70         test_bus_new();
  71
  72         if (test_bus_open() < 0)
  73                 return log_tests_skipped("Failed to connect to bus");
  74
  75         test_bus_new_method_call();
  76         test_bus_new_signal();
  77
  78         return EXIT_SUCCESS;
  79 }