Skip to content

Commit 693c900

Browse files
committed
Remove check make tests beter
1 parent 4ab3240 commit 693c900

File tree

3 files changed

+22
-19
lines changed

3 files changed

+22
-19
lines changed

shared/Microsoft.Extensions.Buffers.MemoryPool.Sources/MemoryPoolBlock.Release.cs

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,6 @@ internal MemoryPoolBlock(SlabMemoryPool pool, MemoryPoolSlab slab, int offset, i
5353

5454
public void Dispose()
5555
{
56-
if (!Slab.IsActive) MemoryPoolThrowHelper.ThrowObjectDisposedException(MemoryPoolThrowHelper.ExceptionArgument.MemoryPoolBlock);
5756
Pool.Return(this);
5857
}
5958

test/Microsoft.Extensions.Internal.Test/MemoryPoolTests.Debug.cs

Lines changed: 21 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -26,10 +26,9 @@ public void DisposeWithActiveBlocksThrows()
2626
{
2727
var memoryPool = new SlabMemoryPool();
2828
var block = memoryPool.Rent();
29-
var exception = Assert.Throws<InvalidOperationException>(() => memoryPool.Dispose());
30-
Assert.Equal("Memory pool with active blocks is being disposed, 30 of 31 returned", exception.Message);
29+
ExpectDisposeException(memoryPool);
3130

32-
exception = Assert.Throws<InvalidOperationException>(() => block.Dispose());
31+
var exception = Assert.Throws<InvalidOperationException>(() => block.Dispose());
3332
Assert.Equal("Block is being returned to disposed pool", exception.Message);
3433
}
3534

@@ -39,11 +38,10 @@ public void GetMemoryOfDisposedThrows()
3938
var memoryPool = new SlabMemoryPool();
4039
var block = memoryPool.Rent();
4140

42-
var exception = Assert.Throws<InvalidOperationException>(() => memoryPool.Dispose());
43-
Assert.Equal("Memory pool with active blocks is being disposed, 30 of 31 returned", exception.Message);
41+
ExpectDisposeException(memoryPool);
4442

45-
exception = Assert.Throws<ObjectDisposedException>(() => block.Memory);
46-
Assert.Equal("Cannot access a disposed object.\r\nObject name: 'MemoryPoolBlock'.", exception.Message);
43+
var exception = Assert.Throws<ObjectDisposedException>(() => block.Memory);
44+
Assert.Equal($"Cannot access a disposed object.{Environment.NewLine}Object name: 'MemoryPoolBlock'.", exception.Message);
4745
}
4846

4947
[Fact]
@@ -52,11 +50,11 @@ public void GetMemoryPinOfDisposedThrows()
5250
var memoryPool = new SlabMemoryPool();
5351
var block = memoryPool.Rent();
5452
var memory = block.Memory;
55-
var exception = Assert.Throws<InvalidOperationException>(() => memoryPool.Dispose());
56-
Assert.Equal("Memory pool with active blocks is being disposed, 30 of 31 returned", exception.Message);
5753

58-
exception = Assert.Throws<ObjectDisposedException>(() => memory.Pin());
59-
Assert.Equal("Cannot access a disposed object.\r\nObject name: 'MemoryPoolBlock'.", exception.Message);
54+
ExpectDisposeException(memoryPool);
55+
56+
var exception = Assert.Throws<ObjectDisposedException>(() => memory.Pin());
57+
Assert.Equal($"Cannot access a disposed object.{Environment.NewLine}Object name: 'MemoryPoolBlock'.", exception.Message);
6058
}
6159

6260
[Fact]
@@ -65,8 +63,8 @@ public void GetMemorySpanOfDisposedThrows()
6563
var memoryPool = new SlabMemoryPool();
6664
var block = memoryPool.Rent();
6765
var memory = block.Memory;
68-
var exception = Assert.Throws<InvalidOperationException>(() => memoryPool.Dispose());
69-
Assert.Equal("Memory pool with active blocks is being disposed, 30 of 31 returned", exception.Message);
66+
67+
ExpectDisposeException(memoryPool);
7068

7169
var threw = false;
7270
try
@@ -76,7 +74,7 @@ public void GetMemorySpanOfDisposedThrows()
7674
catch (ObjectDisposedException ode)
7775
{
7876
threw = true;
79-
Assert.Equal("Cannot access a disposed object.\r\nObject name: 'MemoryPoolBlock'.", ode.Message);
77+
Assert.Equal($"Cannot access a disposed object.{Environment.NewLine}Object name: 'MemoryPoolBlock'.", ode.Message);
8078
}
8179
Assert.True(threw);
8280
}
@@ -87,11 +85,17 @@ public void GetMemoryTryGetArrayOfDisposedThrows()
8785
var memoryPool = new SlabMemoryPool();
8886
var block = memoryPool.Rent();
8987
var memory = block.Memory;
88+
89+
ExpectDisposeException(memoryPool);
90+
91+
var exception = Assert.Throws<ObjectDisposedException>(() => MemoryMarshal.TryGetArray<byte>(memory, out _));
92+
Assert.Equal($"Cannot access a disposed object.{Environment.NewLine}Object name: 'MemoryPoolBlock'.", exception.Message);
93+
}
94+
95+
private static void ExpectDisposeException(SlabMemoryPool memoryPool)
96+
{
9097
var exception = Assert.Throws<InvalidOperationException>(() => memoryPool.Dispose());
9198
Assert.Equal("Memory pool with active blocks is being disposed, 30 of 31 returned", exception.Message);
92-
93-
exception = Assert.Throws<ObjectDisposedException>(() => MemoryMarshal.TryGetArray<byte>(memory, out _));
94-
Assert.Equal("Cannot access a disposed object.\r\nObject name: 'MemoryPoolBlock'.", exception.Message);
9599
}
96100
}
97101
}

test/Microsoft.Extensions.Internal.Test/MemoryPoolTests.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ public void LeasingFromDisposedPoolThrows()
3232
memoryPool.Dispose();
3333

3434
var exception = Assert.Throws<ObjectDisposedException>(() => memoryPool.Rent());
35-
Assert.Equal("Cannot access a disposed object.\r\nObject name: 'MemoryPool'.", exception.Message);
35+
Assert.Equal($"Cannot access a disposed object.{Environment.NewLine}Object name: 'MemoryPool'.", exception.Message);
3636
}
3737
}
3838
}

0 commit comments

Comments
 (0)