From bb10791cb227d8c4855759eb08c77091a8338dd1 Mon Sep 17 00:00:00 2001
From: Taylor Kimmett <tkimmett@outlook.com>
Date: Wed, 25 Sep 2024 20:44:24 -0400
Subject: [PATCH] add unit tests for config processing

---
 .../AzureQueueStreamProviderBuilderTests.cs   | 106 ++++++++++++++++++
 1 file changed, 106 insertions(+)
 create mode 100644 test/Extensions/TesterAzureUtils/Streaming/AzureQueueStreamProviderBuilderTests.cs

diff --git a/test/Extensions/TesterAzureUtils/Streaming/AzureQueueStreamProviderBuilderTests.cs b/test/Extensions/TesterAzureUtils/Streaming/AzureQueueStreamProviderBuilderTests.cs
new file mode 100644
index 0000000000..94a5dbf8d0
--- /dev/null
+++ b/test/Extensions/TesterAzureUtils/Streaming/AzureQueueStreamProviderBuilderTests.cs
@@ -0,0 +1,106 @@
+using System.Text;
+using Microsoft.Extensions.Configuration;
+using Microsoft.Extensions.DependencyInjection;
+using Orleans.Configuration;
+using Xunit;
+
+namespace Tester.AzureUtils;
+
+public class AzureQueueStreamProviderBuilderTests
+{
+	[Fact]
+	public void Missing_ConnectionString()
+	{
+		string json = """
+		{
+			"Orleans": {
+				"Streaming": {
+					"AzureQueueProvider": {
+						"ProviderType": "AzureQueueStorage",
+						"QueueNames": [
+							"q1"
+						]
+					}
+				}
+			}
+		}
+		""";
+
+		var queueOptions = ConfigureSilo(json).Services.BuildServiceProvider().GetOptionsByName<AzureQueueOptions>(null);
+
+		Assert.Null(queueOptions.QueueServiceClient);
+	}
+
+	[Fact]
+	public void Minimal_Configuration()
+	{
+		string json = """
+		{
+			"Orleans": {
+				"Streaming": {
+					"AzureQueueProvider": {
+						"ProviderType": "AzureQueueStorage",
+						"ConnectionString": "UseDevelopmentStorage=true",
+						"QueueNames": [
+							"q1"
+						]
+					}
+				}
+			}
+		}
+		""";
+
+		var queueOptions = ConfigureSilo(json).Services.BuildServiceProvider().GetOptionsByName<AzureQueueOptions>(null);
+
+		Assert.NotNull(queueOptions.QueueServiceClient);
+		Assert.Equal("devstoreaccount1", queueOptions.QueueServiceClient.AccountName);
+		Assert.Equal(["q1"], queueOptions.QueueNames);
+		Assert.Null(queueOptions.MessageVisibilityTimeout);
+	}
+
+	[Fact]
+	public void Full_Configuration()
+	{
+		string json = """
+		{
+			"Orleans": {
+				"Streaming": {
+					"AzureQueueProvider": {
+						"ProviderType": "AzureQueueStorage",
+						"ConnectionString": "UseDevelopmentStorage=true",
+						"MessageVisibilityTimeout": "00:00:37",
+						"QueueNames": [
+							"q1",
+							"q2"
+						]
+					}
+				}
+			}
+		}
+		""";
+
+		var queueOptions = ConfigureSilo(json).Services.BuildServiceProvider().GetOptionsByName<AzureQueueOptions>(null);
+
+		Assert.NotNull(queueOptions.QueueServiceClient);
+		Assert.Equal("devstoreaccount1", queueOptions.QueueServiceClient.AccountName);
+		Assert.Equal(["q1", "q2"], queueOptions.QueueNames);
+		Assert.Equal(TimeSpan.FromSeconds(37), queueOptions.MessageVisibilityTimeout);
+	}
+
+	static TestSiloBuilder ConfigureSilo(string json)
+	{
+		var siloBuilder = new TestSiloBuilder(json);
+		var aqsBuilder = new AzureQueueStreamProviderBuilder();
+		aqsBuilder.Configure(siloBuilder, null, siloBuilder.Configuration.GetSection("Orleans:Streaming:AzureQueueProvider"));
+		return siloBuilder;
+	}
+
+	class TestSiloBuilder(string json) : ISiloBuilder
+	{
+		public IServiceCollection Services { get; } = new ServiceCollection();
+
+		public IConfiguration Configuration { get; } = GetConfig(json);
+	}
+
+	static IConfigurationRoot GetConfig(string json) => new ConfigurationBuilder().AddJsonStream(new MemoryStream(Encoding.UTF8.GetBytes(json))).Build();
+}