summaryrefslogtreecommitdiff
path: root/src/greenlet/tests/_test_extension_cpp.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/greenlet/tests/_test_extension_cpp.cpp')
-rw-r--r--src/greenlet/tests/_test_extension_cpp.cpp141
1 files changed, 72 insertions, 69 deletions
diff --git a/src/greenlet/tests/_test_extension_cpp.cpp b/src/greenlet/tests/_test_extension_cpp.cpp
index fc3adad..72e3d81 100644
--- a/src/greenlet/tests/_test_extension_cpp.cpp
+++ b/src/greenlet/tests/_test_extension_cpp.cpp
@@ -4,47 +4,51 @@
#include "../greenlet.h"
-struct exception_t
-{
- int depth;
- exception_t(int depth) : depth(depth) { }
+struct exception_t {
+ int depth;
+ exception_t(int depth) : depth(depth) {}
};
/* Functions are called via pointers to prevent inlining */
static void (*p_test_exception_throw)(int depth);
static PyObject* (*p_test_exception_switch_recurse)(int depth, int left);
-static void test_exception_throw(int depth)
+static void
+test_exception_throw(int depth)
{
- throw exception_t(depth);
+ throw exception_t(depth);
}
-static PyObject* test_exception_switch_recurse(int depth, int left)
+static PyObject*
+test_exception_switch_recurse(int depth, int left)
{
- if (left > 0) {
- return p_test_exception_switch_recurse(depth, left - 1);
- }
-
- PyObject* result = NULL;
- PyGreenlet* self = PyGreenlet_GetCurrent();
- if (self == NULL)
- return NULL;
-
- try {
- PyGreenlet_Switch(self->parent, NULL, NULL);
- p_test_exception_throw(depth);
- PyErr_SetString(PyExc_RuntimeError, "throwing C++ exception didn't work");
- } catch(exception_t& e) {
- if (e.depth != depth)
- PyErr_SetString(PyExc_AssertionError, "depth mismatch");
- else
- result = PyLong_FromLong(depth);
- } catch(...) {
- PyErr_SetString(PyExc_RuntimeError, "unexpected C++ exception");
- }
-
- Py_DECREF(self);
- return result;
+ if (left > 0) {
+ return p_test_exception_switch_recurse(depth, left - 1);
+ }
+
+ PyObject* result = NULL;
+ PyGreenlet* self = PyGreenlet_GetCurrent();
+ if (self == NULL)
+ return NULL;
+
+ try {
+ PyGreenlet_Switch(self->parent, NULL, NULL);
+ p_test_exception_throw(depth);
+ PyErr_SetString(PyExc_RuntimeError,
+ "throwing C++ exception didn't work");
+ }
+ catch (exception_t& e) {
+ if (e.depth != depth)
+ PyErr_SetString(PyExc_AssertionError, "depth mismatch");
+ else
+ result = PyLong_FromLong(depth);
+ }
+ catch (...) {
+ PyErr_SetString(PyExc_RuntimeError, "unexpected C++ exception");
+ }
+
+ Py_DECREF(self);
+ return result;
}
/* test_exception_switch(int depth)
@@ -53,66 +57,65 @@ static PyObject* test_exception_switch_recurse(int depth, int left)
* - throws an exception that (expected to be caught in the same function)
* - verifies depth matches (exceptions shouldn't be caught in other greenlets)
*/
-static PyObject *
-test_exception_switch(PyObject *self, PyObject *args)
+static PyObject*
+test_exception_switch(PyObject* self, PyObject* args)
{
- int depth;
- if (!PyArg_ParseTuple(args, "i", &depth))
- return NULL;
- return p_test_exception_switch_recurse(depth, depth);
+ int depth;
+ if (!PyArg_ParseTuple(args, "i", &depth))
+ return NULL;
+ return p_test_exception_switch_recurse(depth, depth);
}
static PyMethodDef test_methods[] = {
- {"test_exception_switch", (PyCFunction)&test_exception_switch, METH_VARARGS,
- "Switches to parent twice, to test exception handling and greenlet switching."},
- {NULL, NULL, 0, NULL}
-};
-
+ {"test_exception_switch",
+ (PyCFunction)&test_exception_switch,
+ METH_VARARGS,
+ "Switches to parent twice, to test exception handling and greenlet "
+ "switching."},
+ {NULL, NULL, 0, NULL}};
#if PY_MAJOR_VERSION >= 3
-#define INITERROR return NULL
-
-static struct PyModuleDef moduledef = {
- PyModuleDef_HEAD_INIT,
- "greenlet.tests._test_extension_cpp",
- NULL,
- 0,
- test_methods,
- NULL,
- NULL,
- NULL,
- NULL
-};
+# define INITERROR return NULL
+
+static struct PyModuleDef moduledef = {PyModuleDef_HEAD_INIT,
+ "greenlet.tests._test_extension_cpp",
+ NULL,
+ 0,
+ test_methods,
+ NULL,
+ NULL,
+ NULL,
+ NULL};
PyMODINIT_FUNC
PyInit__test_extension_cpp(void)
#else
-#define INITERROR return
+# define INITERROR return
PyMODINIT_FUNC
init_test_extension_cpp(void)
#endif
{
- PyObject *module = NULL;
+ PyObject* module = NULL;
#if PY_MAJOR_VERSION >= 3
- module = PyModule_Create(&moduledef);
+ module = PyModule_Create(&moduledef);
#else
- module = Py_InitModule("greenlet.tests._test_extension_cpp", test_methods);
+ module = Py_InitModule("greenlet.tests._test_extension_cpp", test_methods);
#endif
- if (module == NULL) {
- INITERROR;
- }
+ if (module == NULL) {
+ INITERROR;
+ }
- PyGreenlet_Import();
- if (_PyGreenlet_API == NULL) {
- INITERROR;
- }
+ PyGreenlet_Import();
+ if (_PyGreenlet_API == NULL) {
+ INITERROR;
+ }
- p_test_exception_throw = test_exception_throw;
- p_test_exception_switch_recurse = test_exception_switch_recurse;
+ p_test_exception_throw = test_exception_throw;
+ p_test_exception_switch_recurse = test_exception_switch_recurse;
#if PY_MAJOR_VERSION >= 3
- return module;
+ return module;
#endif
}