Add support for multiline C definitions in NK_C_API + test

Signed-off-by: Szczepan Zalega <szczepan@nitrokey.com>

1 files changed, 5 insertions, 5 deletions

diff --git a/unittest/conftest.py b/unittest/conftest.py
index 253e1d8..669dc23 100644
--- a/unittest/conftest.py
+++ b/unittest/conftest.py
@@ -44,13 +44,13 @@ def C(request=None):
     cnt = 0
     a = iter(declarations)
     for declaration in a:
-        if declaration.strip().startswith('NK_C_API'):
+        if declaration.strip().startswith('NK_C_API') \
+                or declaration.strip().startswith('struct'):
             declaration = declaration.replace('NK_C_API', '').strip()
-            while ';' not in declaration:
-                declaration += (next(a)).strip()
-            # print(declaration)
+            while ');' not in declaration and '};' not in declaration:
+                declaration += (next(a)).strip()+'\n'
             ffi.cdef(declaration, override=True)
-            cnt +=1
+            cnt += 1
     print('Imported {} declarations'.format(cnt))
 
     C = None