accurate tests

This commit is contained in:
chrislu
2025-08-23 13:56:08 -07:00
parent 11baf008ff
commit 530036df27
2 changed files with 14 additions and 7 deletions
+4 -2
View File
@@ -85,8 +85,10 @@ func TestRaceConditionStress(t *testing.T) {
// Simulate successful volume creation
for _, server := range servers {
disk := server.children[NodeId(types.HardDriveType.String())].(*Disk)
diskUsage := disk.diskUsages.getOrCreateDisk(types.HardDriveType)
atomic.AddInt64(&diskUsage.volumeCount, 1)
deltaDiskUsage := &DiskUsageCounts{
volumeCount: 1,
}
disk.UpAdjustDiskUsageDelta(types.HardDriveType, deltaDiskUsage)
atomic.AddInt64(&totalVolumesCreated, 1)
}
@@ -2,7 +2,6 @@ package topology
import (
"sync"
"sync/atomic"
"testing"
"time"
@@ -81,8 +80,11 @@ func TestVolumeGrowth_ReservationBasedAllocation(t *testing.T) {
}
// Simulate successful volume creation
diskUsage := dn.diskUsages.getOrCreateDisk(types.HardDriveType)
atomic.AddInt64(&diskUsage.volumeCount, 1)
disk := dn.children[NodeId(types.HardDriveType.String())].(*Disk)
deltaDiskUsage := &DiskUsageCounts{
volumeCount: 1,
}
disk.UpAdjustDiskUsageDelta(types.HardDriveType, deltaDiskUsage)
// Release reservation after successful creation
reservation.releaseAllReservations()
@@ -154,8 +156,11 @@ func TestVolumeGrowth_ConcurrentAllocationPreventsRaceCondition(t *testing.T) {
if reservation != nil {
reservation.releaseAllReservations()
// Simulate volume creation by incrementing count
diskUsage := dn.diskUsages.getOrCreateDisk(types.HardDriveType)
atomic.AddInt64(&diskUsage.volumeCount, 1)
disk := dn.children[NodeId(types.HardDriveType.String())].(*Disk)
deltaDiskUsage := &DiskUsageCounts{
volumeCount: 1,
}
disk.UpAdjustDiskUsageDelta(types.HardDriveType, deltaDiskUsage)
}
}
mu.Unlock()