1e2ab4a5e5
Use wecomMaxProcessedMessages in tests and add a concurrent same-message test to lock in race-safety behavior for markMessageProcessed.
73 lines
1.8 KiB
Go
73 lines
1.8 KiB
Go
package channels
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
)
|
|
|
|
func TestMarkMessageProcessed_DuplicateDetection(t *testing.T) {
|
|
var mu sync.RWMutex
|
|
processed := make(map[string]bool)
|
|
|
|
if ok := markMessageProcessed(&mu, &processed, "msg-1", wecomMaxProcessedMessages); !ok {
|
|
t.Fatalf("first message should be accepted")
|
|
}
|
|
|
|
if ok := markMessageProcessed(&mu, &processed, "msg-1", wecomMaxProcessedMessages); ok {
|
|
t.Fatalf("duplicate message should be rejected")
|
|
}
|
|
}
|
|
|
|
func TestMarkMessageProcessed_ConcurrentSameMessage(t *testing.T) {
|
|
var mu sync.RWMutex
|
|
processed := make(map[string]bool)
|
|
|
|
const goroutines = 64
|
|
var wg sync.WaitGroup
|
|
wg.Add(goroutines)
|
|
|
|
results := make(chan bool, goroutines)
|
|
for i := 0; i < goroutines; i++ {
|
|
go func() {
|
|
defer wg.Done()
|
|
results <- markMessageProcessed(&mu, &processed, "msg-concurrent", wecomMaxProcessedMessages)
|
|
}()
|
|
}
|
|
|
|
wg.Wait()
|
|
close(results)
|
|
|
|
successes := 0
|
|
for ok := range results {
|
|
if ok {
|
|
successes++
|
|
}
|
|
}
|
|
|
|
if successes != 1 {
|
|
t.Fatalf("expected exactly 1 successful mark, got %d", successes)
|
|
}
|
|
}
|
|
|
|
func TestMarkMessageProcessed_RotationClearsMapAtBoundary(t *testing.T) {
|
|
var mu sync.RWMutex
|
|
processed := make(map[string]bool)
|
|
|
|
if ok := markMessageProcessed(&mu, &processed, "msg-1", 1); !ok {
|
|
t.Fatalf("first message should be accepted")
|
|
}
|
|
if len(processed) != 1 {
|
|
t.Fatalf("expected map size 1 after first insert, got %d", len(processed))
|
|
}
|
|
|
|
// Inserting second unique message exceeds maxEntries and should reset map.
|
|
if ok := markMessageProcessed(&mu, &processed, "msg-2", 1); !ok {
|
|
t.Fatalf("second unique message should be accepted")
|
|
}
|
|
if len(processed) != 0 {
|
|
t.Fatalf("expected map to be reset after rotation, got size %d", len(processed))
|
|
}
|
|
if processed["msg-2"] {
|
|
t.Fatalf("expected current message marker to be cleared after rotation")
|
|
}
|
|
}
|