Add GetAllAsync on generic repository

CGI-FR · Oct 14, 2022 · a60afe1 · a60afe1
1 parent cf4dd5b
commit a60afe1
Show file tree

Hide file tree

Showing 3 changed files with 84 additions and 0 deletions.
diff --git a/src/AzureIoTHub.Portal.Infrastructure/Repositories/GenericRepository.cs b/src/AzureIoTHub.Portal.Infrastructure/Repositories/GenericRepository.cs
@@ -27,6 +27,14 @@ public IEnumerable<T> GetAll()
                                 .ToList<T>();
         }
 
+        public async Task<IEnumerable<T>> GetAllAsync(Expression<Func<T, bool>>? expression = null, CancellationToken cancellationToken = default)
+        {
+            IQueryable<T> query = this.context.Set<T>();
+            if (expression != null) query = query.Where(expression);
+
+            return await query.ToDynamicListAsync<T>(cancellationToken: cancellationToken);
+        }
+
         public virtual async Task<T?> GetByIdAsync(object id)
         {
             var t = await this.context.Set<T>().FindAsync(id);

diff --git a/src/AzureIoTHub.Portal.Tests.Unit/Infrastructure/Repositories/GenericRepositoryTests.cs b/src/AzureIoTHub.Portal.Tests.Unit/Infrastructure/Repositories/GenericRepositoryTests.cs
@@ -355,5 +355,79 @@ await context.AddRangeAsync(new List<DeviceTag>
             // Assert
             _ = result.Should().BeFalse();
         }
+
+        [Test]
+        public async Task GetAllAsync_WithFilter_AllItemsReturned()
+        {
+            // Arrange
+            var context = SetupDbContext();
+
+            await context.AddRangeAsync(new List<DeviceTag>
+            {
+                new()
+                {
+                    Id = "device_tag_01",
+                    Label = "Location 1",
+                    Required = false,
+                    Searchable = true
+                },
+                new()
+                {
+                    Id = "device_tag_02",
+                    Label = "Location 2",
+                    Required = true,
+                    Searchable = true
+                }
+            });
+
+            _ = await context.SaveChangesAsync();
+
+            var instance = new GenericRepository<DeviceTag>(context);
+
+            // Act
+            var result = await instance.GetAllAsync();
+
+            // Assert
+            _ = result.Count().Should().Be(2);
+        }
+
+        [Test]
+        public async Task GetAllAsync_CustomFilter_ExpectedItemsReturned()
+        {
+            // Arrange
+            var context = SetupDbContext();
+
+            await context.AddRangeAsync(new List<DeviceTag>
+            {
+                new()
+                {
+                    Id = "device_tag_01",
+                    Label = "Location 1",
+                    Required = false,
+                    Searchable = true
+                },
+                new()
+                {
+                    Id = "device_tag_02",
+                    Label = "Location 2",
+                    Required = true,
+                    Searchable = true
+                }
+            });
+
+            _ = await context.SaveChangesAsync();
+
+            var instance = new GenericRepository<DeviceTag>(context);
+
+            // Filter only tags with labels required
+            var deviceTagPredicate = PredicateBuilder.True<DeviceTag>()
+                .And(tag => tag.Required);
+
+            // Act
+            var result = await instance.GetAllAsync(deviceTagPredicate);
+
+            // Assert
+            _ = result.Count().Should().Be(1);
+        }
     }
 }
diff --git a/src/AzureIoTHubPortal.Domain/IRepository.cs b/src/AzureIoTHubPortal.Domain/IRepository.cs
@@ -13,6 +13,8 @@ public interface IRepository<T> where T : EntityBase
     {
         IEnumerable<T> GetAll();
 
+        Task<IEnumerable<T>> GetAllAsync(Expression<Func<T, bool>>? expression = null, CancellationToken cancellationToken = default);
+
         Task<T?> GetByIdAsync(object id);
 
         Task InsertAsync(T obj);