fix: apply category-based WS URLs to demo and test environments too

Author: pcriadoperez

src/node-binance-api.ts

@@ -313,16 +313,24 @@ export default class Binance {
     }
 
     getFStreamSingleUrl(category?: 'public' | 'market' | 'private') {
+        if (category) {
+            if (this.Options.demo) return `wss://fstream.binancefuture.com/${category}/ws/`;
+            if (this.Options.test) return `wss://stream.binancefuture.${this.domain}/${category}/ws/`;
+            return `wss://fstream.binance.${this.domain}/${category}/ws/`;
+        }
         if (this.Options.demo) return this.fstreamSingleDemo;
         if (this.Options.test) return this.fstreamSingleTest;
-        if (category) return `wss://fstream.binance.${this.domain}/${category}/ws/`;
         return this.fstreamSingle;
     }
 
     getFStreamUrl(category?: 'public' | 'market' | 'private') {
+        if (category) {
+            if (this.Options.demo) return `wss://fstream.binancefuture.com/${category}/stream?streams=`;
+            if (this.Options.test) return `wss://stream.binancefuture.${this.domain}/${category}/stream?streams=`;
+            return `wss://fstream.binance.${this.domain}/${category}/stream?streams=`;
+        }
         if (this.Options.demo) return this.fstreamDemo;
         if (this.Options.test) return this.fstreamTest;
-        if (category) return `wss://fstream.binance.${this.domain}/${category}/stream?streams=`;
         return this.fstream;
     }
 

tests/ws-endpoints-migration.test.ts

@@ -114,12 +114,22 @@ describe('getFStreamUrl with category', function () {
     });
 });
 
-describe('Demo mode ignores category (uses legacy URLs)', function () {
-    it('getFStreamSingleUrl returns demo URL regardless of category', function () {
-        assert.equal(demoBinance.getFStreamSingleUrl('market'), 'wss://fstream.binancefuture.com/ws/');
-    });
-    it('getFStreamUrl returns demo URL regardless of category', function () {
-        assert.equal(demoBinance.getFStreamUrl('market'), 'wss://fstream.binancefuture.com/stream?streams=');
+describe('Demo mode uses category-based URLs', function () {
+    it('getFStreamSingleUrl returns demo URL with category', function () {
+        assert.equal(demoBinance.getFStreamSingleUrl('public'), 'wss://fstream.binancefuture.com/public/ws/');
+        assert.equal(demoBinance.getFStreamSingleUrl('market'), 'wss://fstream.binancefuture.com/market/ws/');
+        assert.equal(demoBinance.getFStreamSingleUrl('private'), 'wss://fstream.binancefuture.com/private/ws/');
+    });
+    it('getFStreamUrl returns demo URL with category', function () {
+        assert.equal(demoBinance.getFStreamUrl('public'), 'wss://fstream.binancefuture.com/public/stream?streams=');
+        assert.equal(demoBinance.getFStreamUrl('market'), 'wss://fstream.binancefuture.com/market/stream?streams=');
+        assert.equal(demoBinance.getFStreamUrl('private'), 'wss://fstream.binancefuture.com/private/stream?streams=');
+    });
+    it('getFStreamSingleUrl returns legacy demo URL without category', function () {
+        assert.equal(demoBinance.getFStreamSingleUrl(), 'wss://fstream.binancefuture.com/ws/');
+    });
+    it('getFStreamUrl returns legacy demo URL without category', function () {
+        assert.equal(demoBinance.getFStreamUrl(), 'wss://fstream.binancefuture.com/stream?streams=');
     });
 });