f-21: test: add test for worker pkg
Signed-off-by: GnomeZworc <nicolas.boufidjeline@g3e.fr>
This commit is contained in:
parent
9950e0e24a
commit
712692d414
1 changed files with 103 additions and 0 deletions
103
pkg/worker/queue_test.go
Normal file
103
pkg/worker/queue_test.go
Normal file
|
|
@ -0,0 +1,103 @@
|
||||||
|
package worker
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync"
|
||||||
|
"sync/atomic"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestNew_ReturnsQueue(t *testing.T) {
|
||||||
|
q := New(10)
|
||||||
|
if q == nil {
|
||||||
|
t.Fatal("New devrait retourner une queue non-nil")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestQueue_SingleTaskExecuted(t *testing.T) {
|
||||||
|
q := New(1)
|
||||||
|
q.Start(1)
|
||||||
|
|
||||||
|
var done atomic.Bool
|
||||||
|
var wg sync.WaitGroup
|
||||||
|
wg.Add(1)
|
||||||
|
q.Submit(func() {
|
||||||
|
done.Store(true)
|
||||||
|
wg.Done()
|
||||||
|
})
|
||||||
|
|
||||||
|
wg.Wait()
|
||||||
|
if !done.Load() {
|
||||||
|
t.Error("la tâche n'a pas été exécutée")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestQueue_AllTasksExecuted(t *testing.T) {
|
||||||
|
const n = 50
|
||||||
|
q := New(n)
|
||||||
|
q.Start(1)
|
||||||
|
|
||||||
|
var count atomic.Int32
|
||||||
|
var wg sync.WaitGroup
|
||||||
|
wg.Add(n)
|
||||||
|
for range n {
|
||||||
|
q.Submit(func() {
|
||||||
|
count.Add(1)
|
||||||
|
wg.Done()
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
wg.Wait()
|
||||||
|
if count.Load() != n {
|
||||||
|
t.Errorf("attendu %d exécutions, obtenu %d", n, count.Load())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestQueue_MultipleWorkers(t *testing.T) {
|
||||||
|
const n = 100
|
||||||
|
q := New(n)
|
||||||
|
q.Start(4)
|
||||||
|
|
||||||
|
var count atomic.Int32
|
||||||
|
var wg sync.WaitGroup
|
||||||
|
wg.Add(n)
|
||||||
|
for range n {
|
||||||
|
q.Submit(func() {
|
||||||
|
count.Add(1)
|
||||||
|
wg.Done()
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
wg.Wait()
|
||||||
|
if count.Load() != n {
|
||||||
|
t.Errorf("attendu %d exécutions, obtenu %d", n, count.Load())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestQueue_SubmitBlocksWhenFull(t *testing.T) {
|
||||||
|
q := New(1)
|
||||||
|
// Remplit le buffer sans worker
|
||||||
|
q.Submit(func() {})
|
||||||
|
|
||||||
|
submitted := make(chan struct{})
|
||||||
|
go func() {
|
||||||
|
q.Submit(func() {}) // doit bloquer jusqu'à ce qu'un worker consomme
|
||||||
|
close(submitted)
|
||||||
|
}()
|
||||||
|
|
||||||
|
select {
|
||||||
|
case <-submitted:
|
||||||
|
t.Error("Submit aurait dû bloquer sur une queue pleine")
|
||||||
|
case <-time.After(50 * time.Millisecond):
|
||||||
|
// comportement attendu : goroutine bloquée
|
||||||
|
}
|
||||||
|
|
||||||
|
// Démarre un worker pour débloquer
|
||||||
|
q.Start(1)
|
||||||
|
select {
|
||||||
|
case <-submitted:
|
||||||
|
// Submit a pu avancer
|
||||||
|
case <-time.After(time.Second):
|
||||||
|
t.Error("Submit aurait dû se débloquer après démarrage d'un worker")
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Add table
Add a link
Reference in a new issue