tests/test_pycapi.py

   1 from __future__ import absolute_import
   2
   3 import sys
   4 import unittest
   5 import ctypes
   6 from ctypes import c_void_p, py_object, c_char_p
   7
   8 import gi
   9 from gi.repository import Gio
  10
  11
  12 def get_capi():
  13
  14     if not hasattr(ctypes, "pythonapi"):
  15         return
  16
  17     class CAPI(ctypes.Structure):
  18         _fields_ = [
  19             ("", c_void_p),
  20             ("", c_void_p),
  21             ("", c_void_p),
  22             ("newgobj", ctypes.PYFUNCTYPE(py_object, c_void_p)),
  23         ]
  24
  25     api_obj = gi._gobject._PyGObject_API
  26     if sys.version_info[0] == 2:
  27         func_type = ctypes.PYFUNCTYPE(c_void_p, py_object)
  28         PyCObject_AsVoidPtr = func_type(
  29             ('PyCObject_AsVoidPtr', ctypes.pythonapi))
  30         ptr = PyCObject_AsVoidPtr(api_obj)
  31     else:
  32         func_type = ctypes.PYFUNCTYPE(c_void_p, py_object, c_char_p)
  33         PyCapsule_GetPointer = func_type(
  34             ('PyCapsule_GetPointer', ctypes.pythonapi))
  35         ptr = PyCapsule_GetPointer(api_obj, b"gobject._PyGObject_API")
  36
  37     ptr = ctypes.cast(ptr, ctypes.POINTER(CAPI))
  38     return ptr.contents
  39
  40
  41 API = get_capi()
  42
  43
  44 @unittest.skipUnless(API, "no pythonapi support")
  45 class TestPythonCAPI(unittest.TestCase):
  46
  47     def test_newgobj(self):
  48         w = Gio.FileInfo()
  49         # XXX: ugh :/
  50         ptr = int(repr(w).split()[-1].split(")")[0], 16)
  51
  52         capi = get_capi()
  53         new_w = capi.newgobj(ptr)
  54         assert w == new_w