added exceptions tests

mvenditto · mvenditto · commit 70f757b59485 · 2021-12-17T10:56:16.000Z
diff --git a/errors.v b/errors.v
@@ -22,6 +22,7 @@ pub fn C.PyErr_Format(&C.PyObject, &char, ) &C.PyObject
 pub fn C.PyErr_FormatV(&C.PyObject, &char, &C.va_list) &C.PyObject
 pub fn C.PyErr_SetImportErrorSubclass(&C.PyObject, &C.PyObject, &C.PyObject, &C.PyObject) &C.PyObject
 pub fn C.PyErr_SetImportError(&C.PyObject, &C.PyObject, &C.PyObject) &C.PyObject
+pub fn C.PyErr_NewException(&char, &C.PyObject, &C.PyObject) &C.PyObject
 pub fn C.PyErr_NewExceptionWithDoc(&char, &char, &C.PyObject, &C.PyObject) &C.PyObject
 pub fn C.PyErr_WriteUnraisable(&C.PyObject) 
 pub fn C.PyErr_SetInterrupt() 
diff --git a/exceptions_test.v b/exceptions_test.v
@@ -6,4 +6,25 @@ fn testsuite_begin() {
 }
 
 fn testsuite_end() {
+}
+
+fn test_exc_new_text() {
+	exc := C.PyErr_NewException(c'mymodule.TestException', C.NULL, C.NULL)
+	assert exc.ptr() != C.NULL
+	C.Py_XDECREF(exc)
+}
+
+fn test_exc_new_with_doc() {
+	exc := C.PyErr_NewExceptionWithDoc(c'mymodule.TestException', c'docstring', C.NULL, C.NULL)
+	assert exc.ptr() != C.NULL
+	C.Py_XDECREF(exc)
+}
+
+fn test_exc_set_context() {
+	exc := C.PyErr_NewException(c'mymodule.TestException', C.NULL, C.NULL)
+	assert exc.ptr() != C.NULL
+	defer { C.Py_XDECREF(exc) }
+	C.PyException_SetContext(exc, C.PyExc_BufferError)
+	ctx := C.PyException_GetContext(exc)
+	assert ctx.ptr() == C.PyExc_BufferError
 }
