test/dbus_py_test.c

   1 /* Test fixtures for dbus-python, based on _dbus_glib_bindings. */
   2
   3 #include <Python.h>
   4 #include "dbus-python.h"
   5
   6 #if defined(__GNUC__)
   7 #   if __GNUC__ >= 3
   8 #       define UNUSED __attribute__((__unused__))
   9 #   else
  10 #       define UNUSED /*nothing*/
  11 #   endif
  12 #else
  13 #   define UNUSED /*nothing*/
  14 #endif
  15
  16 static dbus_bool_t
  17 dbus_py_test_set_up_conn(DBusConnection *conn UNUSED, void *data UNUSED)
  18 {
  19     PyErr_SetString(PyExc_ValueError, "Dummy error from UnusableMainLoop");
  20     return 0;
  21 }
  22
  23 static dbus_bool_t
  24 dbus_py_test_set_up_srv(DBusServer *srv UNUSED, void *data UNUSED)
  25 {
  26     PyErr_SetString(PyExc_ValueError, "Dummy error from UnusableMainLoop");
  27     return 0;
  28 }
  29
  30 static void
  31 dbus_py_test_free(void *data UNUSED)
  32 {
  33 }
  34
  35 static PyObject *
  36 dbus_test_native_mainloop(void)
  37 {
  38     PyObject *loop = DBusPyNativeMainLoop_New4(dbus_py_test_set_up_conn,
  39                                                dbus_py_test_set_up_srv,
  40                                                dbus_py_test_free,
  41                                                NULL);
  42     return loop;
  43 }
  44
  45 static PyObject *
  46 UnusableMainLoop (PyObject *always_null UNUSED, PyObject *args, PyObject *kwargs)
  47 {
  48     PyObject *mainloop, *function, *result;
  49     int set_as_default = 0;
  50     static char *argnames[] = {"set_as_default", NULL};
  51
  52     if (PyTuple_Size(args) != 0) {
  53         PyErr_SetString(PyExc_TypeError, "UnusableMainLoop() takes no "
  54                                          "positional arguments");
  55         return NULL;
  56     }
  57     if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|i", argnames,
  58                                      &set_as_default)) {
  59         return NULL;
  60     }
  61
  62     mainloop = dbus_test_native_mainloop();
  63     if (mainloop && set_as_default) {
  64         if (!_dbus_bindings_module) {
  65             PyErr_SetString(PyExc_ImportError, "_dbus_bindings not imported");
  66             Py_DECREF(mainloop);
  67             return NULL;
  68         }
  69         function = PyObject_GetAttrString(_dbus_bindings_module,
  70                                           "set_default_main_loop");
  71         if (!function) {
  72             Py_DECREF(mainloop);
  73             return NULL;
  74         }
  75         result = PyObject_CallFunctionObjArgs(function, mainloop, NULL);
  76         Py_DECREF(function);
  77         if (!result) {
  78             Py_DECREF(mainloop);
  79             return NULL;
  80         }
  81     }
  82     return mainloop;
  83 }
  84
  85 static PyMethodDef module_functions[] = {
  86     {"UnusableMainLoop", (PyCFunction)UnusableMainLoop,
  87      METH_VARARGS|METH_KEYWORDS, "Return a main loop that fails to attach"},
  88     {NULL, NULL, 0, NULL}
  89 };
  90
  91 PyMODINIT_FUNC
  92 initdbus_py_test(void)
  93 {
  94     PyObject *this_module;
  95
  96     if (import_dbus_bindings("dbus_py_test") < 0) return;
  97     this_module = Py_InitModule3 ("dbus_py_test", module_functions, "");
  98     if (!this_module) return;
  99 }
 100
 101 /* vim:set ft=c cino< sw=4 sts=4 et: */