add system_test.

Author: kafka1991

system_test/questdb_line_sender.py

@@ -98,9 +98,16 @@ class CertificateAuthority(Enum):
 c_line_protocol_version = ctypes.c_int
 
 
-class ProtocolVersion(Enum):
-    V1 = (c_line_protocol_version(0), 'v1')
-    V2 = (c_line_protocol_version(1), 'v2')
+class LineProtocolVersion(Enum):
+    V1 = (c_line_protocol_version(1), 'v1')
+    V2 = (c_line_protocol_version(2), 'v2')
+
+    @classmethod
+    def from_int(cls, value: c_line_protocol_version):
+        for member in cls:
+            if member.value[0].value == value:
+                return member
+        raise ValueError(f"invalid protocol version: {value}")
 
 
 class c_line_sender_opts(ctypes.Structure):
@@ -213,6 +220,8 @@ def set_sig(fn, restype, *argtypes):
         c_size_t)
     set_sig(
         dll.line_sender_buffer_set_line_protocol_version,
+        c_bool,
+        c_line_sender_buffer_p,
         c_line_protocol_version,
         c_line_sender_error_p_p)
     set_sig(
@@ -594,12 +603,14 @@ def __init__(self, micros: int):
 
 
 class Buffer:
-    def __init__(self, init_buf_size=65536, max_name_len=127, line_protocol_version=ProtocolVersion.V2):
+    def __init__(self, init_buf_size=65536, max_name_len=127, line_protocol_version=LineProtocolVersion.V2):
         self._impl = _DLL.line_sender_buffer_with_max_name_len(
             c_size_t(max_name_len))
         _DLL.line_sender_buffer_reserve(self._impl, c_size_t(init_buf_size))
         _error_wrapped_call(
-            _DLL.line_sender_buffer_set_line_protocol_version(self._impl, line_protocol_version.value[0]))
+            _DLL.line_sender_buffer_set_line_protocol_version,
+            self._impl,
+            line_protocol_version.value[0])
 
     def __len__(self):
         return _DLL.line_sender_buffer_size(self._impl)
@@ -617,8 +628,11 @@ def peek(self) -> str:
         else:
             return ''
 
-    def set_line_protocol_version(self, version: ProtocolVersion):
-        _error_wrapped_call(_DLL.line_sender_buffer_set_line_protocol_version(self._impl, version))
+    def set_line_protocol_version(self, version: LineProtocolVersion):
+        _error_wrapped_call(
+            _DLL.line_sender_buffer_set_line_protocol_version,
+            self._impl,
+            version.value[0])
 
     def reserve(self, additional):
         _DLL.line_sender_buffer_reserve(self._impl, c_size_t(additional))
@@ -809,7 +823,8 @@ def connect(self):
 
     def __enter__(self):
         self.connect()
-        self._buffer = Buffer(line_protocol_version=self.line_sender_default_line_protocol_version())
+        self._buffer = Buffer(
+            line_protocol_version=LineProtocolVersion.from_int(self.line_sender_default_line_protocol_version()))
         return self
 
     def _check_connected(self):
@@ -818,9 +833,7 @@ def _check_connected(self):
 
     def line_sender_default_line_protocol_version(self):
         self._check_connected()
-        return _error_wrapped_call(
-            _DLL.line_sender_default_line_protocol_version,
-            self._impl)
+        return _DLL.line_sender_default_line_protocol_version(self._impl)
 
     def table(self, table: str):
         self._buffer.table(table)

system_test/test.py

@@ -523,7 +523,7 @@ def test_line_protocol_version_v1(self):
         table_name = uuid.uuid4().hex
         pending = None
         with self._mk_linesender() as sender:
-            sender.buffer.set_line_protocol_version(qls.ProtocolVersion.V1)
+            sender.buffer.set_line_protocol_version(qls.LineProtocolVersion.V1)
             for num in numbers:
                 sender.table(table_name)
                 sender.column('n', num)