Updated tests.

khanjal · Aug 18, 2024 · 788f8a2 · 788f8a2
1 parent 2323006
commit 788f8a2
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 6 deletions.
diff --git a/GigRaptorLib.Tests/Utilities/Google/GoogleSheetManagerTests.cs b/GigRaptorLib.Tests/Utilities/Google/GoogleSheetManagerTests.cs
@@ -89,7 +89,7 @@ public async Task GivenGetSpreadsheetName_WithInvalidSpreadsheetId_ReturnNull()
     public async Task GivenAddSheetData_WithValidSheetId_ThenReturnEmpty()
     {
         var googleSheetManager = new Mock<IGoogleSheetManager>();
-        googleSheetManager.Setup(x => x.AddSheetData(It.IsAny<List<SheetEnum>>(), It.IsAny<SheetEntity>())).ReturnsAsync(It.IsAny<SheetEntity>());
+        googleSheetManager.Setup(x => x.AddSheetData(It.IsAny<List<SheetEnum>>(), It.IsAny<SheetEntity>())).ReturnsAsync(new SheetEntity());
         var result = await googleSheetManager.Object.AddSheetData([new SheetEnum()], new SheetEntity());
         result.Should().NotBeNull();
     }
@@ -122,7 +122,7 @@ public async Task GivenAddSheetData_WithData_ThenReturnData()
     public async Task GivenCreateSheet_WithValidSheetId_ThenReturnEmpty()
     {
         var googleSheetManager = new Mock<IGoogleSheetManager>();
-        googleSheetManager.Setup(x => x.CreateSheets(It.IsAny<List<SheetEnum>>())).ReturnsAsync(It.IsAny<SheetEntity>());
+        googleSheetManager.Setup(x => x.CreateSheets(It.IsAny<List<SheetEnum>>())).ReturnsAsync(new SheetEntity());
         var result = await googleSheetManager.Object.CreateSheets([new SheetEnum()]);
         result.Should().NotBeNull();
     }

diff --git a/GigRaptorLib/Utilities/Google/GoogleSheetManager.cs b/GigRaptorLib/Utilities/Google/GoogleSheetManager.cs
@@ -46,13 +46,16 @@ public async Task<SheetEntity> AddSheetData(List<SheetEnum> sheets, SheetEntity
             switch (sheet) {
                 case SheetEnum.SHIFTS:
                     values = ShiftMapper.MapToRangeData(sheetEntity.Shifts, headers);
+                    sheetEntity.Messages.Add(MessageHelper.CreateInfoMessage($"Adding data to {sheet.UpperName()}"));
                     break;
 
                 case SheetEnum.TRIPS:
                     values = TripMapper.MapToRangeData(sheetEntity.Trips, headers);
+                    sheetEntity.Messages.Add(MessageHelper.CreateInfoMessage($"Adding data to {sheet.UpperName()}"));
                     break;
                 default:
                     // Unsupported sheet.
+                    sheetEntity.Messages.Add(MessageHelper.CreateErrorMessage($"Adding data to {sheet.UpperName()} not supported"));
                     break;
             }
 
@@ -62,13 +65,17 @@ public async Task<SheetEntity> AddSheetData(List<SheetEnum> sheets, SheetEntity
                 var result = await _googleSheetService.AppendData(valueRange, $"{sheet.DisplayName()}!{GoogleConfig.Range}");
 
                 if (result == null)
-                {
-                    //success = false;
-                }
+                    sheetEntity.Messages.Add(MessageHelper.CreateErrorMessage($"Unable to add data to {sheet.UpperName()}"));
+                else
+                    sheetEntity.Messages.Add(MessageHelper.CreateInfoMessage($"Added data to {sheet.UpperName()}"));
+            }
+            else
+            {
+                sheetEntity.Messages.Add(MessageHelper.CreateWarningMessage($"No data to add to {sheet.UpperName()}"));
             }
         }
 
-        return new SheetEntity();
+        return sheetEntity;
     }
 
     public async Task<SheetEntity> CreateSheets()