Creating intitial tests for python (#5)

Author: Yucong Sun

.circleci/config.yml

@@ -11,8 +11,13 @@ jobs:
           command: |
             sudo sh -c 'echo "deb http://deb.debian.org/debian stretch-backports main" > /etc/apt/sources.list.d/backports.list'
             sudo apt-get update
-            sudo apt-get -t stretch-backports install cmake
+            sudo apt-get -t stretch-backports install cmake python3-dev python3-venv
       - run:
           name: Build everything
           command: |
             ./build.sh
+      - run:
+          name: Build Python stuff
+          command: |
+            cd python
+            ./test.sh

CMakeLists.txt

@@ -3,4 +3,4 @@ cmake_minimum_required(VERSION 3.7)
 project(libra-client-dev)
 
 add_subdirectory(c)
-add_subdirectory(cpp)
+add_subdirectory(cpp)

build.sh

@@ -1,4 +1,4 @@
-#!/bin/bash
+#!/bin/bash -xv
 set -euo pipefail
 
 # Build libra-dev first

python/.gitignore

@@ -1 +1,7 @@
-*.pyc
+*.pyc
+*.so
+/*.egg-info
+/_skbuild
+/.eggs
+/dist
+/venv

python/CMakeLists.txt

@@ -0,0 +1,26 @@
+cmake_minimum_required(VERSION 3.10)
+
+project(pylibra)
+
+set(CMAKE_C_STANDARD 11)
+
+find_package(PythonExtensions REQUIRED)
+
+add_library(_pylibra MODULE "pylibra/_pylibra.c")
+python_extension_module(_pylibra)
+install(TARGETS _pylibra LIBRARY DESTINATION pylibra)
+
+
+target_include_directories(_pylibra PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}/../libra-dev/include")
+target_link_directories(_pylibra PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}/../libra-dev/target/debug")
+target_link_libraries(_pylibra "libra_dev")
+
+IF(APPLE)
+    target_link_libraries(_pylibra "-framework Security")
+elseif(WIN32)
+
+else()
+    target_link_libraries(_pylibra "pthread")
+    target_link_libraries(_pylibra "dl")
+    target_link_libraries(_pylibra "m")
+endif()

python/pylibra/__init__.py

@@ -0,0 +1 @@
+from ._pylibra import *

python/pylibra/_pylibra.c

@@ -0,0 +1,61 @@
+#define PY_SSIZE_T_CLEAN
+#include <Python.h>
+
+#include "data.h"
+
+static PyObject *f_account_resource_from_lcs(PyObject *self, PyObject *args) {
+    PyBytesObject* bytes;
+
+    /* Parse arguments */
+    if(!PyArg_ParseTuple(args, "S", &bytes)) {
+        return NULL;
+    }
+
+    struct CDevAccountResource result = account_resource_from_lcs((const uint8_t*) PyBytes_AsString(bytes), PyBytes_Size(bytes));
+
+    PyObject* obj = PyDict_New();
+    PyDict_SetItem(obj, PyUnicode_FromString("balance"), PyLong_FromUnsignedLongLong(result.balance));
+
+    return obj;
+}
+
+static PyMethodDef pyLibraMethods[] = {
+        {"account_resource_from_lcs", f_account_resource_from_lcs, METH_VARARGS, "parse account_state_blob"},
+        {NULL, NULL, 0, NULL}
+};
+
+
+static struct PyModuleDef pyLibraModule = {
+        PyModuleDef_HEAD_INIT,
+        "_pylibra",
+        "Python interface for the libra-dev library",
+        -1,
+        pyLibraMethods
+};
+
+PyMODINIT_FUNC PyInit__pylibra(void) {
+    return PyModule_Create(&pyLibraModule);
+}
+
+/*
+PyMODINIT_FUNC
+PyInit_pyLibra(void)
+{
+    PyObject *m;
+    if (PyType_Ready(&pyLibraAccountResourceType) < 0)
+        return NULL;
+
+    m = PyModule_Create(&pyLibraModule);
+    if (m == NULL)
+        return NULL;
+
+    Py_INCREF(&pyLibraAccountResourceType);
+    if (PyModule_AddObject(m, "Custom", (PyObject *) &pyLibraAccountResourceType) < 0) {
+        Py_DECREF(&pyLibraAccountResourceType);
+        PY_DECREF(m);
+        return NULL;
+    }
+
+    return m;
+}
+*/

python/pyproject.toml

@@ -0,0 +1,2 @@
+[build-system]
+requires = ["setuptools", "wheel", "scikit-build", "cmake", "ninja"]

python/setup.py

@@ -0,0 +1,20 @@
+import sys
+
+from skbuild import setup  # This line replaces 'from setuptools import setup'
+
+# Require pytest-runner only when running tests
+pytest_runner = (['pytest-runner']
+                 if any(arg in sys.argv for arg in ('pytest', 'test'))
+                 else [])
+
+setup_requires = pytest_runner
+
+setup(name="pylibra",
+      version="1.0.0",
+      description="Python interface for the libra-dev library function",
+      author="Yucong Sun",
+      author_email="[email protected]",
+      packages=['pylibra'],
+      tests_require=['pytest', 'pytest-runner'],
+      setup_requires=setup_requires
+)

python/test.sh

@@ -0,0 +1,10 @@
+#!/bin/bash -xv
+set -euo pipefail
+
+python3 -m venv ./venv
+
+./venv/bin/pip3 install --upgrade pip
+./venv/bin/pip3 install --upgrade scikit-build pytest
+
+./venv/bin/python3 setup.py develop
+./venv/bin/python3 setup.py pytest

python/test_pylibra.py

@@ -0,0 +1,7 @@
+import pylibra
+
+BLOB = bytes.fromhex("020000002100000001674deac5e7fca75f00ca92b1ba3697f5f01ef585011beea7b361150f4504638f0800000002000000000000002100000001a208df134fefed8442b1f01fab59071898f5a1af5164e12c594de55a7004a91c8e0000002000000036ccb9ba8b4f0cd1f3e2d99338806893dff7478c69acee9b8e1247c053783a4800e876481700000000000200000000000000200000000b14ed4f5af8f8f077c7ec4313c6d395b9a7eb5f41eab9ec15367215ca9e420a01000000000000002000000032f56f77b09773aa64c78ee39943da7ec73f91cd757e325098e11b3edc4eccb10100000000000000")
+
+def test_account_state_blob():
+    res = pylibra.account_resource_from_lcs(BLOB)
+    assert res["balance"] == 100000000000