Add unit tests for AdminResourceRepository and AdminResourceService

Author: o-ii

test/AzureOpenAIProxy.ApiApp.Tests/Repositories/AdminResourceRepositoryTests.cs

@@ -1,4 +1,3 @@
-using Azure;
 using Azure.Data.Tables;
 
 using AzureOpenAIProxy.ApiApp.Configurations;
@@ -10,10 +9,85 @@
 using Microsoft.Extensions.DependencyInjection;
 
 using NSubstitute;
-using NSubstitute.ExceptionExtensions;
 
 namespace AzureOpenAIProxy.ApiApp.Tests.Repositories;
 
 public class AdminResourceRepositoryTests
 {
+    [Fact]
+    public void Given_ServiceCollection_When_AddAdminResourceRepository_Invoked_Then_It_Should_Contain_AdminResourceRepository()
+    {
+        // Arrange
+        var services = new ServiceCollection();
+
+        // Act
+        services.AddAdminResourceRepository();
+
+        // Assert
+        services.SingleOrDefault(p => p.ServiceType == typeof(IAdminResourceRepository)).Should().NotBeNull();
+    }
+
+    [Fact]
+    public void Given_Null_TableServiceClient_When_Creating_AdminResourceRepository_Then_It_Should_Throw_Exception()
+    {
+        // Arrange
+        var settings = Substitute.For<StorageAccountSettings>();
+        var tableServiceClient = default(TableServiceClient);
+        
+        // Act
+        Action action = () => new AdminResourceRepository(tableServiceClient!, settings);
+
+        // Assert
+        action.Should().Throw<ArgumentNullException>();
+    }
+
+    [Fact]
+    public void Given_Null_StorageAccountSettings_When_Creating_AdminResourceRepository_Then_It_Should_Throw_Exception()
+    {
+        // Arrange
+        var settings = default(StorageAccountSettings);
+        var tableServiceClient = Substitute.For<TableServiceClient>();
+        
+        // Act
+        Action action = () => new AdminResourceRepository(tableServiceClient, settings!);
+
+        // Assert
+        action.Should().Throw<ArgumentNullException>();
+    }
+
+    [Fact]
+    public async Task Given_Instance_When_CreateResource_Invoked_Then_It_Should_Add_Entity()
+    {
+        // Arrange
+        var settings = Substitute.For<StorageAccountSettings>();
+        var tableServiceClient = Substitute.For<TableServiceClient>();
+        var tableClient = Substitute.For<TableClient>();
+        tableServiceClient.GetTableClient(Arg.Any<string>()).Returns(tableClient);
+
+        var repository = new AdminResourceRepository(tableServiceClient, settings);
+
+        var resourceId = Guid.NewGuid();
+        var resourceDetails = new AdminResourceDetails
+        {
+            ResourceId = resourceId,
+            FriendlyName = "Test Resource",
+            DeploymentName = "Test Deployment",
+            ResourceType = ResourceType.Chat,
+            Endpoint = "https://test.endpoint.com",
+            ApiKey = "test-api-key",
+            Region = "test-region",
+            IsActive = true,
+            PartitionKey = PartitionKeys.ResourceDetails,
+            RowKey = resourceId.ToString()
+        };
+
+        // Act
+        var result = await repository.CreateResource(resourceDetails);
+
+        // Assert
+        await tableClient.Received(1).AddEntityAsync(Arg.Is<AdminResourceDetails>(x =>
+            x.ResourceId == resourceDetails.ResourceId
+        ));
+        result.Should().BeEquivalentTo(resourceDetails);
+    }
 }

test/AzureOpenAIProxy.ApiApp.Tests/Services/AdminResourceServiceTests.cs

@@ -1,5 +1,3 @@
-using Azure;
-
 using AzureOpenAIProxy.ApiApp.Models;
 using AzureOpenAIProxy.ApiApp.Repositories;
 using AzureOpenAIProxy.ApiApp.Services;
@@ -15,4 +13,89 @@ namespace AzureOpenAIProxy.ApiApp.Tests.Services;
 
 public class AdminResourceServiceTests
 {
+    [Fact]
+    public void Given_ServiceCollection_When_AddAdminResourceService_Invoked_Then_It_Should_Contain_AdminResourceService()
+    {
+        // Arrange
+        var services = new ServiceCollection();
+
+        // Act
+        services.AddAdminResourceService();
+
+        // Assert
+        services.SingleOrDefault(p => p.ServiceType == typeof(IAdminResourceService)).Should().NotBeNull();
+    }
+
+    [Fact]
+    public void Given_Null_Repository_When_Creating_AdminResourceService_Then_It_Should_Throw_Exception()
+    {
+        // Arrange
+        IAdminResourceRepository? repository = null;
+
+        // Act
+        Action action = () => new AdminResourceService(repository!);
+
+        // Assert
+        action.Should().Throw<ArgumentNullException>();
+    }
+
+    [Fact]
+    public async Task Given_Instance_When_CreateResource_Invoked_Then_It_Should_Add_Entity()
+    {
+        // Arrange
+        var repository = Substitute.For<IAdminResourceRepository>();
+        var service = new AdminResourceService(repository);
+
+        var resourceDetails = new AdminResourceDetails
+        {
+            ResourceId = Guid.NewGuid(),
+            FriendlyName = "Test Resource",
+            DeploymentName = "Test Deployment",
+            ResourceType = ResourceType.Chat,
+            Endpoint = "https://test.endpoint.com",
+            ApiKey = "test-api-key",
+            Region = "test-region",
+            IsActive = true
+        };
+
+        repository.CreateResource(resourceDetails).Returns(resourceDetails);
+
+        // Act
+        var result = await service.CreateResource(resourceDetails);
+
+        // Assert
+        await repository.Received(1).CreateResource(Arg.Is<AdminResourceDetails>(x =>
+            x.ResourceId == resourceDetails.ResourceId
+        ));
+
+        result.Should().BeEquivalentTo(resourceDetails);
+    }
+
+    [Fact]
+    public async Task Given_RepositoryFails_When_CreateResource_Invoked_Then_It_Should_Throw_Exception()
+    {
+        // Arrange
+        var repository = Substitute.For<IAdminResourceRepository>();
+        var service = new AdminResourceService(repository);
+
+        var resourceDetails = new AdminResourceDetails
+        {
+            ResourceId = Guid.NewGuid(),
+            FriendlyName = "Test Resource",
+            DeploymentName = "Test Deployment",
+            ResourceType = ResourceType.Chat,
+            Endpoint = "https://test.endpoint.com",
+            ApiKey = "test-api-key",
+            Region = "test-region",
+            IsActive = true
+        };
+
+        repository.CreateResource(Arg.Any<AdminResourceDetails>()).ThrowsAsync(new InvalidOperationException());
+
+        // Act
+        Func<Task> act = async () => await service.CreateResource(resourceDetails);
+
+        // Assert
+        await act.Should().ThrowAsync<InvalidOperationException>();
+    }
 }