Test CreateEdgeModelShouldCreateEdgeModel

CGI-FR · Dec 21, 2022 · c9a47de · c9a47de
1 parent 23e986f
commit c9a47de
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/AzureIoTHub.Portal.Server/Services/EdgeModelService.cs b/src/AzureIoTHub.Portal.Server/Services/EdgeModelService.cs
@@ -99,7 +99,7 @@ public async Task CreateEdgeModel(IoTEdgeModel edgeModel)
                 throw new ResourceAlreadyExistsException($"The edge model with id {edgeModel?.ModelId} already exists");
             }
 
-            _ = this.deviceModelImageManager.SetDefaultImageToModel(edgeModel.ModelId);
+            _ = await this.deviceModelImageManager.SetDefaultImageToModel(edgeModel?.ModelId);
 
             await SaveModuleCommands(edgeModel);
             await this.configService.RollOutEdgeModelConfiguration(edgeModel);

diff --git a/src/AzureIoTHub.Portal.Tests.Unit/Server/Services/EdgeModelServiceTest.cs b/src/AzureIoTHub.Portal.Tests.Unit/Server/Services/EdgeModelServiceTest.cs
@@ -180,6 +180,7 @@ public async Task CreateEdgeModelShouldCreateEdgeModel()
         {
             // Arrange
             var edgeDeviceModel = Fixture.Create<IoTEdgeModel>();
+            var expectedImageUri = Fixture.Create<Uri>();
 
             _ = this.mockEdgeDeviceModelRepository.Setup(x => x.GetByIdAsync(It.IsAny<string>()))
                 .ReturnsAsync((EdgeDeviceModel)null);
@@ -202,6 +203,10 @@ public async Task CreateEdgeModelShouldCreateEdgeModel()
             _ = this.mockConfigService.Setup(x => x.RollOutEdgeModelConfiguration(It.IsAny<IoTEdgeModel>()))
                 .Returns(Task.CompletedTask);
 
+            _ = this.mockDeviceModelImageManager.Setup(manager =>
+                    manager.SetDefaultImageToModel(It.IsAny<string>()))
+                .ReturnsAsync(expectedImageUri.ToString());
+
             // Act
             await this.edgeDeviceModelService.CreateEdgeModel(edgeDeviceModel);