test: add unit tests on session buffer size

amqp · Mar 10, 2023 · f902e4f · f902e4f
1 parent ee8c088
commit f902e4f
Show file tree

Hide file tree

Showing 2 changed files with 77 additions and 1 deletion.
diff --git a/test/sessions.ts b/test/sessions.ts
@@ -65,6 +65,82 @@ describe('session error handling', function () {
             context.connection.close();
         });
     });
+    it('session buffer size [default]', function(done) {
+        container.on('session_open', function (context: rhea.EventContext) {
+            assert.equal((s as any).incoming.deliveries.capacity, 2048);
+            assert.equal((s as any).outgoing.deliveries.capacity, 2048);
+            context.session!.close({ condition: 'amqp:internal-error', description: 'testing error on close' });
+        });
+        container.on('session_close', function (context) {
+            assert.equal(context.session.is_closed(), true);
+        });
+        var c: rhea.Connection = container.connect(listener.address());
+        c.on('connection_close', function () {
+            done();
+        });
+        var s = c.create_session();
+        s.begin();
+        s.on('session_close', function (context: rhea.EventContext) {
+            context.connection.close();
+        });
+    });
+    it('session buffer size [number]', function(done) {
+        container.on('session_open', function (context: rhea.EventContext) {
+            assert.equal((s as any).incoming.deliveries.capacity, 4096);
+            assert.equal((s as any).outgoing.deliveries.capacity, 4096);
+            context.session!.close({ condition: 'amqp:internal-error', description: 'testing error on close' });
+        });
+        container.on('session_close', function (context) {
+            assert.equal(context.session.is_closed(), true);
+        });
+        var c: rhea.Connection = container.connect(listener.address());
+        c.on('connection_close', function () {
+            done();
+        });
+        var s = c.create_session(4096);
+        s.begin();
+        s.on('session_close', function (context: rhea.EventContext) {
+            context.connection.close();
+        });
+    });
+    it('session buffer size [object]', function(done) {
+        container.on('session_open', function (context: rhea.EventContext) {
+            assert.equal((s as any).incoming.deliveries.capacity, 2048);
+            assert.equal((s as any).outgoing.deliveries.capacity, 4096);
+            context.session!.close({ condition: 'amqp:internal-error', description: 'testing error on close' });
+        });
+        container.on('session_close', function (context) {
+            assert.equal(context.session.is_closed(), true);
+        });
+        var c: rhea.Connection = container.connect(listener.address());
+        c.on('connection_close', function () {
+            done();
+        });
+        var s = c.create_session({ incoming: 2048, outgoing: 4096 });
+        s.begin();
+        s.on('session_close', function (context: rhea.EventContext) {
+            context.connection.close();
+        });
+    });
+    it('session buffer size [invalid]', function(done) {
+        container.on('session_open', function (context: rhea.EventContext) {
+            assert.equal((s as any).incoming.deliveries.capacity, 2048);
+            assert.equal((s as any).outgoing.deliveries.capacity, 2048);
+            context.session!.close({ condition: 'amqp:internal-error', description: 'testing error on close' });
+        });
+        container.on('session_close', function (context) {
+            assert.equal(context.session.is_closed(), true);
+        });
+        var c: rhea.Connection = container.connect(listener.address());
+        c.on('connection_close', function () {
+            done();
+        });
+        var s = c.create_session('invalid' as unknown as number);
+        s.begin();
+        s.on('session_close', function (context: rhea.EventContext) {
+            context.connection.close();
+        });
+    });
     it('error handled', function (done: Function) {
         var error_handler_called: boolean;
         container.on('session_open', function (context: rhea.EventContext) {

diff --git a/typings/connection.d.ts b/typings/connection.d.ts
@@ -696,7 +696,7 @@ export declare interface Connection extends EventEmitter {
    * @returns {boolean} `true` - closed, `false` otherwise.
    */
   is_closed(): boolean;
-  create_session(): Session;
+  create_session(session_buffer_size?: number | { incoming?: number, outgoing?: number }): Session;
   find_sender(filter: Function): Sender | undefined;
   find_receiver(filter: Function): Receiver | undefined;
   find_link(filter: Function): link | undefined;