test: Fix nullability warnings

VMelnalksnis · Jan 14, 2024 · 330ae67 · 330ae67
1 parent cb20944
commit 330ae67
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/tests/Gnomeshade.WebApi.Tests.Integration/AccessControl/DeleteAccessTests.cs b/tests/Gnomeshade.WebApi.Tests.Integration/AccessControl/DeleteAccessTests.cs
@@ -248,11 +248,11 @@ private async Task ShouldBeDeletedWithClient<TEntity>(Guid id, IGnomeshadeClient
 		var deletedEntity = await Fixture.GetEntityRepository(scope).FindByIdAsync<TEntity>(id);
 		var userId = (await client.GetMyCounterpartyAsync()).CreatedByUserId;
 
-		using (new AssertionScope())
-		{
-			deletedEntity.DeletedAt.Should().NotBeNull();
-			deletedEntity.DeletedByUserId.Should().Be(userId);
-		}
+		using var assertionScope = new AssertionScope();
+
+		deletedEntity.Should().NotBeNull();
+		deletedEntity?.DeletedAt.Should().NotBeNull();
+		deletedEntity?.DeletedByUserId.Should().Be(userId);
 	}
 
 	private Task ShouldBeNotFoundForOthers(Func<IGnomeshadeClient, Task> func, bool inverted = false)

diff --git a/tests/Gnomeshade.WebApi.Tests.Integration/EntityRepository.cs b/tests/Gnomeshade.WebApi.Tests.Integration/EntityRepository.cs
@@ -43,7 +43,7 @@ public EntityRepository(DbConnection dbConnection)
 		_dbConnection = dbConnection;
 	}
 
-	internal async Task<IEntity> FindByIdAsync<TEntity>(Guid id)
+	internal async Task<IEntity?> FindByIdAsync<TEntity>(Guid id)
 		where TEntity : Entity
 	{
 		var tableName = _tableNames[typeof(TEntity)];