add tests

Author: Bakytgerey Ashirbekov

.github/workflows/python.yml

@@ -29,6 +29,8 @@ jobs:
       - uses: actions/setup-python@v5
         with:
           python-version: '3.10'
+      - name: Install dependencies
+        run: pip install ".[testing]"
       - name: Build wheels
         uses: PyO3/maturin-action@v1
         with:
@@ -37,6 +39,8 @@ jobs:
           manylinux: '2_28'
           args: --release --out dist --find-interpreter
           sccache: 'true'
+      - name: Run tests
+        run: pytest
       - name: Upload wheels
         uses: actions/upload-artifact@v4
         with:
@@ -54,6 +58,8 @@ jobs:
         with:
           python-version: '3.10'
           architecture: ${{ matrix.target }}
+      - name: Install dependencies
+        run: pip install ".[testing]"
       - name: Set up NASM
         uses: ilammy/setup-nasm@v1
       - name: Build wheels
@@ -62,14 +68,16 @@ jobs:
           target: ${{ matrix.target }}
           args: --release --out dist --find-interpreter
           sccache: 'true'
+      - name: Run tests
+        run: pytest
       - name: Upload wheels
         uses: actions/upload-artifact@v4
         with:
           name: wheels-windows-${{ matrix.target }}
           path: dist
 
   macos:
-    runs-on: macos-latest
+    runs-on: macos-13
     strategy:
       matrix:
         target: [x86_64, aarch64]
@@ -78,12 +86,38 @@ jobs:
       - uses: actions/setup-python@v5
         with:
           python-version: '3.10'
+      - name: Install dependencies
+        run: pip install ".[testing]"
+      - name: Install and start Colima
+        run: |
+          brew install colima docker
+          colima start --network-address --network-host-addresses
+          echo "DOCKER_HOST=unix://$HOME/.colima/default/docker.sock" >> $GITHUB_ENV
+          docker network ls
+      - name: Ensure Colima is running and Docker is accessible
+        run: |
+          colima status || (echo "Colima is not running!" && exit 1)
+          docker ps || (echo "Docker is not running inside Colima!" && exit 1)
+      - name: Debug Running Containers
+        run: |
+          docker ps -a
+          docker port $(docker ps -q) || echo "No running containers yet"
+      - name: Manually Forward Port 8090
+        run: |
+          limactl shell colima sudo iptables -t nat -A PREROUTING -p tcp --dport 8090 -j REDIRECT --to-port 8090
+          limactl shell colima sudo iptables -L -t nat
       - name: Build wheels
         uses: PyO3/maturin-action@v1
         with:
           target: ${{ matrix.target }}
           args: --release --out dist --find-interpreter
           sccache: 'true'
+      - name: Wait for Container to Start & Check Port
+        run: |
+          sleep 5
+          netstat -an | grep 8090 || echo "Port 8090 is not open"
+      - name: Run tests
+        run: pytest
       - name: Upload wheels
         uses: actions/upload-artifact@v4
         with:
@@ -94,11 +128,15 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
+      - name: Install dependencies
+        run: pip install ".[testing]"
       - name: Build sdist
         uses: PyO3/maturin-action@v1
         with:
           command: sdist
           args: --out dist
+      - name: Run tests
+        run: pytest
       - name: Upload sdist
         uses: actions/upload-artifact@v4
         with:

Cargo.lock

@@ -14,12 +14,12 @@ name = "iggy_py"
 crate-type = ["cdylib", "rlib"]
 
 [dependencies]
-pyo3 = "0.23.0"
+pyo3 = "0.22.0"
 iggy = "0.6.201"
 tokio = { version = "1.43.0", features = ["full"] }
 bytes = "1.10.0"
 openssl = { version = "0.10.*", features = ["vendored"] }
-pyo3-async-runtimes = { version = "0.23.0", features = ["attributes", "tokio-runtime"] }
+pyo3-async-runtimes = { version = "0.22.0", features = ["attributes", "tokio-runtime"] }
 pyo3-stub-gen = "0.7.0"
 
 [[bin]]

Cargo.toml

@@ -11,7 +11,15 @@ class IggyClient:
     It wraps the RustIggyClient and provides asynchronous functionality
     through the contained runtime.
     """
-    def __new__(cls,conn:typing.Optional[builtins.str]=None): ...
+    def new(self, conn:typing.Optional[builtins.str]) -> IggyClient:
+        r"""
+        Constructs a new IggyClient.
+        
+        This initializes a new runtime for asynchronous operations.
+        Future versions might utilize asyncio for more Pythonic async.
+        """
+        ...
+
     def login_user(self, username:builtins.str, password:builtins.str) -> typing.Any:
         r"""
         Logs in the user with the given credentials.
@@ -28,7 +36,7 @@ class IggyClient:
         """
         ...
 
-    def create_stream(self, name:builtins.str, stream_id:typing.Optional[builtins.int]=None) -> typing.Any:
+    def create_stream(self, name:builtins.str, stream_id:typing.Optional[builtins.int]) -> typing.Any:
         r"""
         Creates a new stream with the provided ID and name.
         
@@ -44,7 +52,7 @@ class IggyClient:
         """
         ...
 
-    def create_topic(self, stream:PyIdentifier, name:builtins.str, partitions_count:builtins.int, compression_algorithm:typing.Optional[builtins.str]=None, topic_id:typing.Optional[builtins.int]=None, replication_factor:typing.Optional[builtins.int]=None) -> typing.Any:
+    def create_topic(self, stream:PyIdentifier, name:builtins.str, partitions_count:builtins.int, compression_algorithm:typing.Optional[builtins.str], topic_id:typing.Optional[builtins.int], replication_factor:typing.Optional[builtins.int]) -> typing.Any:
         r"""
         Creates a new topic with the given parameters.
         
@@ -140,6 +148,28 @@ class ReceiveMessage:
         ...
 
 
+class SendMessage:
+    r"""
+    A Python class representing a message to be sent.
+    
+    This class wraps a Rust message meant for sending, facilitating
+    the creation of such messages from Python and their subsequent use in Rust.
+    """
+    def __new__(cls,data:builtins.str): ...
+    ...
+
+class StreamDetails:
+    id: builtins.int
+    name: builtins.str
+    messages_count: builtins.int
+    topics_count: builtins.int
+
+class TopicDetails:
+    id: builtins.int
+    name: builtins.str
+    messages_count: builtins.int
+    topics_count: builtins.int
+
 class MessageState(Enum):
     Available = auto()
     Unavailable = auto()

iggy_py.pyi

@@ -15,3 +15,14 @@ description= "Apache Iggy is the persistent message streaming platform written i
 
 [tool.maturin]
 features = ["pyo3/extension-module"]
+
+[project.optional-dependencies]
+testing = [
+    "pytest",
+    "pytest-asyncio",
+    "testcontainers[docker]",
+    "maturin"
+]
+
+[tool.pytest.ini_options]
+asyncio_mode = "auto"

pyproject.toml

@@ -50,8 +50,8 @@ impl From<PyIdentifier> for Identifier {
     }
 }
 
-#[gen_stub_pymethods]
 #[pymethods]
+#[gen_stub_pymethods]
 impl IggyClient {
     /// Constructs a new IggyClient.
     ///

src/client.rs

@@ -40,7 +40,7 @@ impl ReceiveMessage {
     ///
     /// The payload is returned as a Python bytes object.
     pub fn payload(&self, py: Python) -> PyObject {
-        PyBytes::new(py, &self.inner.payload).into()
+        PyBytes::new_bound(py, &self.inner.payload).into()
     }
 
     /// Retrieves the offset of the received message.

src/receive_message.rs

@@ -1,12 +1,14 @@
 use iggy::messages::send_messages::Message as RustSendMessage;
 use pyo3::prelude::*;
+use pyo3_stub_gen::derive::{gen_stub_pyclass, gen_stub_pymethods};
 use std::str::FromStr;
 
 /// A Python class representing a message to be sent.
 ///
 /// This class wraps a Rust message meant for sending, facilitating
 /// the creation of such messages from Python and their subsequent use in Rust.
 #[pyclass]
+#[gen_stub_pyclass]
 pub struct SendMessage {
     pub(crate) inner: RustSendMessage,
 }
@@ -23,6 +25,7 @@ impl Clone for SendMessage {
     }
 }
 
+#[gen_stub_pymethods]
 #[pymethods]
 impl SendMessage {
     /// Constructs a new `SendMessage` instance from a string.

src/send_message.rs

@@ -1,7 +1,9 @@
 use iggy::models::stream::StreamDetails as RustStreamDetails;
 use pyo3::prelude::*;
+use pyo3_stub_gen::derive::{gen_stub_pyclass, gen_stub_pymethods};
 
 #[pyclass]
+#[gen_stub_pyclass]
 pub struct StreamDetails {
     pub(crate) inner: RustStreamDetails,
 }
@@ -14,6 +16,7 @@ impl From<RustStreamDetails> for StreamDetails {
     }
 }
 
+#[gen_stub_pymethods]
 #[pymethods]
 impl StreamDetails {
     #[getter]

src/stream.rs

@@ -1,6 +1,8 @@
 use iggy::models::topic::TopicDetails as RustTopicDetails;
 use pyo3::prelude::*;
+use pyo3_stub_gen::derive::{gen_stub_pyclass, gen_stub_pymethods};
 
+#[gen_stub_pyclass]
 #[pyclass]
 pub struct TopicDetails {
     pub(crate) inner: RustTopicDetails,
@@ -14,6 +16,7 @@ impl From<RustTopicDetails> for TopicDetails {
     }
 }
 
+#[gen_stub_pymethods]
 #[pymethods]
 impl TopicDetails {
     #[getter]