Add allow_batch to StreamConfig (#764)

Signed-off-by: Casper Beyer <[email protected]>

Author: caspervonb

nats/src/nats/js/api.py

@@ -311,6 +311,12 @@ class StreamConfig(Base):
     # Allow scheduled/delayed messages. Introduced in nats-server 2.12.0.
     allow_msg_schedules: Optional[bool] = None
 
+    # Allow atomic batch publishing. Introduced in nats-server 2.12.0.
+    allow_atomic: Optional[bool] = None
+
+    # Allow batched publishing. Introduced in nats-server 2.12.0.
+    allow_batched: Optional[bool] = None
+
     # Metadata are user defined string key/value pairs.
     metadata: Optional[Dict[str, str]] = None
 

nats/tests/test_js.py

@@ -4649,6 +4649,42 @@ async def test_stream_allow_msg_schedules(self):
 
         await nc.close()
 
+    @async_test
+    async def test_stream_allow_atomic(self):
+        nc = await nats.connect()
+
+        server_version = nc.connected_server_version
+        if server_version.major == 2 and server_version.minor < 12:
+            pytest.skip("allow_atomic requires nats-server v2.12.0 or later")
+
+        js = nc.jetstream()
+        await js.add_stream(
+            name="ATOMIC",
+            subjects=["test"],
+            allow_atomic=True,
+        )
+        sinfo = await js.stream_info("ATOMIC")
+        assert sinfo.config.allow_atomic is True
+
+        # Test that it can be set to False
+        await js.add_stream(
+            name="NOATOMIC",
+            subjects=["foo"],
+            allow_atomic=False,
+        )
+        sinfo = await js.stream_info("NOATOMIC")
+        assert sinfo.config.allow_atomic is not True
+
+        # Test that it defaults to falsy when not set
+        await js.add_stream(
+            name="DEFAULT2",
+            subjects=["baz"],
+        )
+        sinfo = await js.stream_info("DEFAULT2")
+        assert sinfo.config.allow_atomic is not True
+
+        await nc.close()
+
     @async_test
     async def test_fetch_pull_subscribe_bind(self):
         nc = NATS()