63 lines
1.7 KiB
Go
63 lines
1.7 KiB
Go
package dispatcher
|
|
|
|
import (
|
|
"errors"
|
|
"sync"
|
|
"testing"
|
|
|
|
configuration "git.g3e.fr/syonad/two/internal/config/agent"
|
|
"github.com/dgraph-io/badger/v4"
|
|
)
|
|
|
|
func TestDispatcher_Prepare_Success(t *testing.T) {
|
|
d, _ := newTestDispatcher(t)
|
|
cmd := mockCmd{
|
|
prepareFn: func(*badger.DB, *configuration.Config) error { return nil },
|
|
executeFn: func(*badger.DB, *configuration.Config) error { return nil },
|
|
}
|
|
if err := d.Prepare(cmd); err != nil {
|
|
t.Errorf("Prepare devrait retourner nil, obtenu : %v", err)
|
|
}
|
|
}
|
|
|
|
func TestDispatcher_Prepare_PropagatesError(t *testing.T) {
|
|
d, _ := newTestDispatcher(t)
|
|
want := errors.New("prepare failed")
|
|
cmd := mockCmd{
|
|
prepareFn: func(*badger.DB, *configuration.Config) error { return want },
|
|
executeFn: func(*badger.DB, *configuration.Config) error { return nil },
|
|
}
|
|
if err := d.Prepare(cmd); !errors.Is(err, want) {
|
|
t.Errorf("attendu %v, obtenu %v", want, err)
|
|
}
|
|
}
|
|
|
|
func TestDispatcher_Dispatch_ExecutesCommand(t *testing.T) {
|
|
d, _ := newTestDispatcher(t)
|
|
var wg sync.WaitGroup
|
|
wg.Add(1)
|
|
cmd := mockCmd{
|
|
prepareFn: func(*badger.DB, *configuration.Config) error { return nil },
|
|
executeFn: func(*badger.DB, *configuration.Config) error {
|
|
wg.Done()
|
|
return nil
|
|
},
|
|
}
|
|
d.Dispatch(cmd)
|
|
wg.Wait()
|
|
}
|
|
|
|
func TestDispatcher_Dispatch_ExecuteErrorLogged(t *testing.T) {
|
|
d, _ := newTestDispatcher(t)
|
|
var wg sync.WaitGroup
|
|
wg.Add(1)
|
|
cmd := mockCmd{
|
|
prepareFn: func(*badger.DB, *configuration.Config) error { return nil },
|
|
executeFn: func(*badger.DB, *configuration.Config) error {
|
|
defer wg.Done()
|
|
return errors.New("execute failed")
|
|
},
|
|
}
|
|
d.Dispatch(cmd)
|
|
wg.Wait() // Execute s'est terminé — l'erreur est loggée, pas propagée
|
|
}
|