Fixed interfaces and tests.

Author: s3rius

python/natsrpy/_natsrpy_rs/js/managers.pyi

@@ -91,7 +91,7 @@ class CountersManager:
         :return: True if the stream was deleted.
         """
 
-    async def update(self, config: StreamConfig) -> Counters:
+    async def update(self, config: CountersConfig) -> Counters:
         """Update an existing counters stream configuration.
 
         :param config: new stream configuration.

python/tests/test_counters.py

@@ -1,7 +1,7 @@
 import uuid
 
 import pytest
-from natsrpy.js import CounterEntry, Counters, CountersConfig, JetStream, StreamConfig
+from natsrpy.js import CounterEntry, Counters, CountersConfig, JetStream
 
 
 async def test_counters_create(js: JetStream) -> None:
@@ -51,11 +51,9 @@ async def test_counters_update(js: JetStream) -> None:
     config = CountersConfig(name=name, subjects=[f"{name}.>"])
     await js.counters.create(config)
     try:
-        update_cfg = StreamConfig(
+        update_cfg = CountersConfig(
             name=name,
             subjects=[f"{name}.>"],
-            allow_direct=True,
-            allow_message_counter=True,
             description="updated description",
         )
         counters = await js.counters.update(update_cfg)

src/js/managers/counters.rs

@@ -7,7 +7,7 @@ use crate::{
 use pyo3::{Bound, PyAny, Python};
 use tokio::sync::RwLock;
 
-use crate::{exceptions::rust_err::NatsrpyResult, js::stream::StreamConfig, utils::natsrpy_future};
+use crate::{exceptions::rust_err::NatsrpyResult, utils::natsrpy_future};
 
 #[pyo3::pyclass]
 pub struct CountersManager {
@@ -87,7 +87,7 @@ impl CountersManager {
     pub fn update<'py>(
         &self,
         py: Python<'py>,
-        config: StreamConfig,
+        config: CountersConfig,
     ) -> NatsrpyResult<Bound<'py, PyAny>> {
         let ctx = self.ctx.clone();
         natsrpy_future(py, async move {