psqlpy-python
diff --git a/‎Cargo.lock
Lines changed: 35 additions & 41 deletions b/‎Cargo.lock
Lines changed: 35 additions & 41 deletions
diff --git a/‎Cargo.toml
Lines changed: 5 additions & 1 deletion b/‎Cargo.toml
Lines changed: 5 additions & 1 deletion
diff --git a/‎python/tests/test_binary_copy.py
Lines changed: 3 additions & 0 deletions b/‎python/tests/test_binary_copy.py
Lines changed: 3 additions & 0 deletions
diff --git a/‎python/tests/test_connection.py
Lines changed: 10 additions & 9 deletions b/‎python/tests/test_connection.py
Lines changed: 10 additions & 9 deletions
diff --git a/‎src/driver/common_options.rs
Lines changed: 10 additions & 9 deletions b/‎src/driver/common_options.rs
Lines changed: 10 additions & 9 deletions
@@ -14,10 +14,14 @@ pyo3 = { version = "*", features = [
     "chrono",
     "experimental-async",
     "rust_decimal",
+    "py-clone",
+    "gil-refs",
+    "macros",
 ] }
-pyo3-asyncio = { git = "https://github.com/chandr-andr/pyo3-asyncio.git", version = "0.20.0", features = [
+pyo3-async-runtimes = { git = "https://github.com/chandr-andr/pyo3-async-runtimes.git", branch = "main", features = [
     "tokio-runtime",
 ] }
+
 tokio = { version = "1.35.1", features = ["full"] }
 thiserror = "1.0.56"
 bytes = "1.5.0"
 
@@ -2,11 +2,14 @@
 import typing
 from io import BytesIO
 
+import pytest
 from pgpq import ArrowToPostgresBinaryEncoder
 from pyarrow import parquet
 
 from psqlpy import ConnectionPool
 
+pytestmark = pytest.mark.anyio
+
 
 async def test_binary_copy_to_table_in_connection(
     psql_pool: ConnectionPool,
 
@@ -142,15 +142,16 @@ async def test_connection_cursor(
     """Test cursor from Connection."""
     connection = await psql_pool.connection()
     cursor: Cursor
-    all_results: list[dict[typing.Any, typing.Any]] = []
-
-    async with connection.transaction(), connection.cursor(
-        querystring=f"SELECT * FROM {table_name}",
-    ) as cursor:
-        async for cur_res in cursor:
-            all_results.extend(cur_res.result())
-
-    assert len(all_results) == number_database_records
+    transaction = connection.transaction()
+    await transaction.begin()
+    cursor = connection.cursor(querystring=f"SELECT * FROM {table_name}")
+    await cursor.start()
+    await cursor.close()
+    await transaction.commit()
+
+    # async with connection.transaction(), connection.cursor(
+    # ) as cursor:
+    #     async for cur_res in cursor:
 
 
 async def test_connection_async_context_manager(
 
@@ -3,8 +3,8 @@ use std::time::Duration;
 use deadpool_postgres::RecyclingMethod;
 use pyo3::{pyclass, pymethods};
 
-#[pyclass]
-#[derive(Clone, Copy)]
+#[pyclass(eq, eq_int)]
+#[derive(Clone, Copy, PartialEq)]
 pub enum ConnRecyclingMethod {
     Fast,
     Verified,
@@ -22,8 +22,8 @@ impl ConnRecyclingMethod {
     }
 }
 
-#[pyclass]
-#[derive(Clone, Copy)]
+#[pyclass(eq, eq_int)]
+#[derive(Clone, Copy, PartialEq)]
 pub enum LoadBalanceHosts {
     /// Make connection attempts to hosts in the order provided.
     Disable,
@@ -41,8 +41,8 @@ impl LoadBalanceHosts {
     }
 }
 
-#[pyclass]
-#[derive(Clone, Copy)]
+#[pyclass(eq, eq_int)]
+#[derive(Clone, Copy, PartialEq)]
 pub enum TargetSessionAttrs {
     /// No special properties are required.
     Any,
@@ -63,7 +63,7 @@ impl TargetSessionAttrs {
     }
 }
 
-#[pyclass]
+#[pyclass(eq, eq_int)]
 #[derive(Clone, Copy, PartialEq)]
 pub enum SslMode {
     /// Do not use TLS.
@@ -110,6 +110,7 @@ pub struct KeepaliveConfig {
 #[pymethods]
 impl KeepaliveConfig {
     #[new]
+    #[pyo3(signature = (idle, interval=None, retries=None))]
     fn build_config(idle: u64, interval: Option<u64>, retries: Option<u32>) -> Self {
         let interval_internal = interval.map(Duration::from_secs);
         KeepaliveConfig {
@@ -120,8 +121,8 @@ impl KeepaliveConfig {
     }
 }
 
-#[pyclass]
-#[derive(Clone, Copy)]
+#[pyclass(eq, eq_int)]
+#[derive(Clone, Copy, PartialEq)]
 pub enum CopyCommandFormat {
     TEXT,
     CSV,