Add more tests for session injection

Anton Smolkov · Anton Smolkov · commit 8d5092d694de · 2024-12-01T16:43:00.000+03:00
diff --git a/Source/MQTTnet.Server/Options/MqttServerOptionsBuilder.cs b/Source/MQTTnet.Server/Options/MqttServerOptionsBuilder.cs
@@ -115,6 +115,12 @@ public MqttServerOptionsBuilder WithMaxPendingMessagesPerClient(int value)
             return this;
         }
 
+        public MqttServerOptionsBuilder WithPendingMessagesOverflowStrategy(MqttPendingMessagesOverflowStrategy value)
+        {
+            _options.PendingMessagesOverflowStrategy = value;
+            return this;
+        }
+
         public MqttServerOptionsBuilder WithoutDefaultEndpoint()
         {
             _options.DefaultEndpointOptions.IsEnabled = false;
diff --git a/Source/MQTTnet.Tests/BaseTestClass.cs b/Source/MQTTnet.Tests/BaseTestClass.cs
@@ -13,10 +13,11 @@ namespace MQTTnet.Tests
     public abstract class BaseTestClass
     {
         public TestContext TestContext { get; set; }
-        
-        protected TestEnvironment CreateTestEnvironment(MqttProtocolVersion protocolVersion = MqttProtocolVersion.V311)
+
+        protected TestEnvironment CreateTestEnvironment(
+            MqttProtocolVersion protocolVersion = MqttProtocolVersion.V311, bool trackUnobservedTaskException = true)
         {
-            return new TestEnvironment(TestContext, protocolVersion);
+            return new TestEnvironment(TestContext, protocolVersion, trackUnobservedTaskException);
         }
 
         protected Task LongTestDelay()
diff --git a/Source/MQTTnet.Tests/Mockups/TestEnvironment.cs b/Source/MQTTnet.Tests/Mockups/TestEnvironment.cs
@@ -32,12 +32,16 @@ public TestEnvironment() : this(null)
         {
         }
 
-        public TestEnvironment(TestContext testContext, MqttProtocolVersion protocolVersion = MqttProtocolVersion.V311)
+        public TestEnvironment(
+            TestContext testContext, MqttProtocolVersion protocolVersion = MqttProtocolVersion.V311, bool trackUnobservedTaskException = true)
         {
             _protocolVersion = protocolVersion;
             TestContext = testContext;
 
-            TaskScheduler.UnobservedTaskException += TrackUnobservedTaskException;
+            if (trackUnobservedTaskException)
+            {
+                TaskScheduler.UnobservedTaskException += TrackUnobservedTaskException;
+            }
 
             ServerLogger.LogMessagePublished += (s, e) =>
             {
diff --git a/Source/MQTTnet.Tests/Server/Injection_Tests.cs b/Source/MQTTnet.Tests/Server/Injection_Tests.cs

Original file line number	Diff line number	Diff line change
`@@ -115,6 +115,12 @@ public MqttServerOptionsBuilder WithMaxPendingMessagesPerClient(int value)`
`115`	`115`	`return this;`
`116`	`116`	`}`
`117`	`117`
	`118`	`+ public MqttServerOptionsBuilder WithPendingMessagesOverflowStrategy(MqttPendingMessagesOverflowStrategy value)`
	`119`	`+ {`
	`120`	`+ _options.PendingMessagesOverflowStrategy = value;`
	`121`	`+ return this;`
	`122`	`+ }`
	`123`	`+`
`118`	`124`	`public MqttServerOptionsBuilder WithoutDefaultEndpoint()`
`119`	`125`	`{`
`120`	`126`	`_options.DefaultEndpointOptions.IsEnabled = false;`
Original file line number	Diff line number	Diff line change
`@@ -13,10 +13,11 @@ namespace MQTTnet.Tests`
`13`	`13`	`public abstract class BaseTestClass`
`14`	`14`	`{`
`15`	`15`	`public TestContext TestContext { get; set; }`
`16`		`-`
`17`		`- protected TestEnvironment CreateTestEnvironment(MqttProtocolVersion protocolVersion = MqttProtocolVersion.V311)`
	`16`	`+`
	`17`	`+ protected TestEnvironment CreateTestEnvironment(`
	`18`	`+ MqttProtocolVersion protocolVersion = MqttProtocolVersion.V311, bool trackUnobservedTaskException = true)`
`18`	`19`	`{`
`19`		`- return new TestEnvironment(TestContext, protocolVersion);`
	`20`	`+ return new TestEnvironment(TestContext, protocolVersion, trackUnobservedTaskException);`
`20`	`21`	`}`
`21`	`22`
`22`	`23`	`protected Task LongTestDelay()`
Original file line number	Diff line number	Diff line change
`@@ -32,12 +32,16 @@ public TestEnvironment() : this(null)`
`32`	`32`	`{`
`33`	`33`	`}`
`34`	`34`
`35`		`- public TestEnvironment(TestContext testContext, MqttProtocolVersion protocolVersion = MqttProtocolVersion.V311)`
	`35`	`+ public TestEnvironment(`
	`36`	`+ TestContext testContext, MqttProtocolVersion protocolVersion = MqttProtocolVersion.V311, bool trackUnobservedTaskException = true)`
`36`	`37`	`{`
`37`	`38`	`_protocolVersion = protocolVersion;`
`38`	`39`	`TestContext = testContext;`
`39`	`40`
`40`		`- TaskScheduler.UnobservedTaskException += TrackUnobservedTaskException;`
	`41`	`+ if (trackUnobservedTaskException)`
	`42`	`+ {`
	`43`	`+ TaskScheduler.UnobservedTaskException += TrackUnobservedTaskException;`
	`44`	`+ }`
`41`	`45`
`42`	`46`	`ServerLogger.LogMessagePublished += (s, e) =>`
`43`	`47`	`{`