Fixed stubs, added stubtest.

Author: s3rius

Cargo.toml

@@ -18,7 +18,7 @@ async-nats = "0.46"
 bytes = "1.11.1"
 futures-util = "0.3.32"
 log = "0.4.29"
-pyo3 = { version = "0.28", features = ["abi3"] }
+pyo3 = { version = "0.28", features = ["abi3", "experimental-inspect"] }
 pyo3-async-runtimes = { version = "0.28", features = ["tokio-runtime"] }
 pyo3-log = "0.13.3"
 thiserror = "2.0.18"

python/natsrpy/_natsrpy_rs/__init__.pyi

@@ -50,16 +50,16 @@ class Nats:
     def __new__(
         cls,
         /,
-        addrs: list[str] = ["nats://localhost:4222"],
+        addrs: list[str] | None = None,
         user_and_pass: tuple[str, str] | None = None,
         nkey: str | None = None,
         token: str | None = None,
         custom_inbox_prefix: str | None = None,
-        read_buffer_capacity: int = 65535,
-        sender_capacity: int = 128,
+        read_buffer_capacity: int = ...,  # 65535 bytes
+        sender_capacity: int = ...,  # 128 bytes
         max_reconnects: int | None = None,
-        connection_timeout: float | timedelta = ...,
-        request_timeout: float | timedelta = ...,
+        connection_timeout: float | timedelta = ...,  # 5 sec
+        request_timeout: float | timedelta = ...,  # 10 sec
     ) -> Self: ...
     async def startup(self) -> None: ...
     async def shutdown(self) -> None: ...
@@ -78,8 +78,8 @@ class Nats:
         payload: bytes | str | bytearray | memoryview,
         *,
         headers: dict[str, Any] | None = None,
-        reply: str | None = None,
-        err_on_disconnect: bool = False,
+        inbox: str | None = None,
+        timeout: float | timedelta | None = None,
     ) -> None: ...
     async def drain(self) -> None: ...
     async def flush(self) -> None: ...
@@ -95,6 +95,16 @@ class Nats:
         subject: str,
         callback: None = None,
     ) -> IteratorSubscription: ...
-    async def jetstream(self) -> js.JetStream: ...
+    async def jetstream(
+        self,
+        *,
+        domain: str | None = None,
+        api_prefix: str | None = None,
+        timeout: timedelta | None = None,
+        ack_timeout: timedelta | None = None,
+        concurrency_limit: int | None = None,
+        max_ack_inflight: int | None = None,
+        backpressure_on_inflight: bool | None = None,
+    ) -> js.JetStream: ...
 
 __all__ = ["CallbackSubscription", "IteratorSubscription", "Message", "Nats", "js"]

python/natsrpy/_natsrpy_rs/js/object_store.pyi

@@ -39,13 +39,13 @@ class ObjectStore:
         self,
         name: str,
         writer: Writer[bytes],
-        chunk_size: int | None = 24576,
+        chunk_size: int | None = ...,  # 24MB
     ) -> None: ...
     async def put(
         self,
         name: str,
         value: bytes | str,
-        chunk_size: int = 24576,
+        chunk_size: int = ...,  # 24MB
         description: str | None = None,
         headers: dict[str, str | list[str]] | None = None,
         metadata: dict[str, str] | None = None,

src/js/consumers/push/consumer.rs

@@ -60,6 +60,7 @@ impl MessagesIterator {
         slf
     }
 
+    #[pyo3(signature=(timeout=None))]
     pub fn next<'py>(
         &self,
         py: Python<'py>,

src/js/mod.rs

@@ -10,7 +10,8 @@ pub mod stream;
 pub mod pymod {
     // Classes
     #[pymodule_export]
-    pub use super::jetstream::JetStream;
+    pub use super::jetstream::{JetStream, Publication};
+
     #[pymodule_export]
     pub use super::message::JetStreamMessage;
 

src/js/stream.rs

@@ -353,7 +353,7 @@ impl Source {
         start_sequence = None,
         start_time=None,
         domain=None,
-        subject_transforms = vec![]
+        subject_transforms = None
     ))]
     pub fn __new__(
         name: String,
@@ -362,7 +362,7 @@ impl Source {
         start_sequence: Option<u64>,
         start_time: Option<i64>,
         domain: Option<String>,
-        subject_transforms: Vec<Bound<'_, SubjectTransform>>,
+        subject_transforms: Option<Vec<Bound<'_, SubjectTransform>>>,
     ) -> NatsrpyResult<Self> {
         Ok(Self {
             name,
@@ -371,6 +371,7 @@ impl Source {
             start_sequence,
             filter_subject,
             subject_transforms: subject_transforms
+                .unwrap_or_default()
                 .into_iter()
                 .map(|val| val.borrow().deref().clone())
                 .collect(),

src/nats_cls.rs

@@ -37,7 +37,7 @@ impl NatsCls {
     #[new]
     #[pyo3(signature = (
         /,
-        addrs=vec![String::from("nats://localhost:4222")],
+        addrs=None,
         user_and_pass=None,
         nkey=None,
         token=None,
@@ -49,7 +49,7 @@ impl NatsCls {
         request_timeout=TimeValue::FloatSecs(10.0),
     ))]
     fn __new__(
-        addrs: Vec<String>,
+        addrs: Option<Vec<String>>,
         user_and_pass: Option<(String, String)>,
         nkey: Option<String>,
         token: Option<String>,
@@ -71,7 +71,7 @@ impl NatsCls {
             max_reconnects,
             connection_timeout,
             request_timeout,
-            addr: addrs,
+            addr: addrs.unwrap_or_else(|| vec![String::from("nats://localhost:4222")]),
         }
     }