headers test

Author: doghappy

src/SocketIOClient.Test/SocketIOTests/HeadersTest.cs

@@ -0,0 +1,43 @@
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using System.Collections.Generic;
+using System.Threading.Tasks;
+
+namespace SocketIOClient.Test.SocketIOTests
+{
+    public abstract class HeadersTest
+    {
+        protected abstract ISocketIOCreateable SocketIOCreator { get; }
+
+        public async virtual Task CustomHeader()
+        {
+            string result = null;
+
+            var client = new SocketIO(SocketIOCreator.Url, new SocketIOOptions
+            {
+                Reconnection = false,
+                Query = new List<KeyValuePair<string, string>>
+                {
+                    new KeyValuePair<string, string>("token", SocketIOCreator.Token)
+                },
+                ExtraHeaders = new Dictionary<string, string>
+                {
+                    { "CustomHeader", "CustomHeader-Value" }
+                }
+            });
+
+            client.OnConnected += async (sender, e) =>
+            {
+                var sio = sender as SocketIO;
+                await sio.EmitAsync("headers", response =>
+                {
+                    result = response.GetValue().GetProperty("customheader").GetString();
+                });
+            };
+
+            await client.ConnectAsync();
+            await Task.Delay(400);
+
+            Assert.AreEqual(result, "CustomHeader-Value");
+        }
+    }
+}

src/SocketIOClient.Test/SocketIOTests/V3/HeadersV3Test.cs

@@ -0,0 +1,22 @@
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using System.Threading.Tasks;
+
+namespace SocketIOClient.Test.SocketIOTests.V3
+{
+    [TestClass]
+    public class HeadersV3Test : HeadersTest
+    {
+        public HeadersV3Test()
+        {
+            SocketIOCreator = new SocketIOV3Creator();
+        }
+
+        protected override ISocketIOCreateable SocketIOCreator { get; }
+
+        [TestMethod]
+        public override async Task CustomHeader()
+        {
+            await base.CustomHeader();
+        }
+    }
+}

src/socket.io-server-v3/app.js

@@ -32,6 +32,10 @@ io.on('connection', socket => {
         socket.emit('hi', prefix + msg);
     });
 
+    socket.on('headers', cb => {
+        cb(socket.request.headers)
+    });
+
     socket.on('welcome', () => {
         //socket.emit('welcome', Buffer.from("welcome " + socket.id, 'utf8'));
         socket.emit('welcome', Buffer.from("a", 'utf8'));