mirror of
https://github.com/MeshCore-Beacon/beacon-server.git
synced 2026-09-01 16:48:19 +00:00
tests(background): test bg scheduler
This commit is contained in:
@@ -0,0 +1,104 @@
|
||||
// Copyright 2026 Beacon Contributors
|
||||
// SPDX-License-Identifier: AGPL-3.0-or-later
|
||||
|
||||
package background
|
||||
|
||||
import (
|
||||
"context"
|
||||
"sync/atomic"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestScheduler_RunsTask(t *testing.T) {
|
||||
var calls atomic.Int32
|
||||
|
||||
tasks := []Task{
|
||||
{
|
||||
Name: "test-task",
|
||||
Interval: 10 * time.Millisecond,
|
||||
Run: func(ctx context.Context) error {
|
||||
calls.Add(1)
|
||||
return nil
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
s := New(tasks)
|
||||
go s.Start(ctx)
|
||||
|
||||
time.Sleep(50 * time.Millisecond)
|
||||
if calls.Load() == 0 {
|
||||
t.Error("expected task to have run at least once")
|
||||
}
|
||||
}
|
||||
|
||||
func TestScheduler_StopsOnContextCancel(t *testing.T) {
|
||||
var calls atomic.Int32
|
||||
|
||||
tasks := []Task{
|
||||
{
|
||||
Name: "test-task",
|
||||
Interval: 10 * time.Millisecond,
|
||||
Run: func(ctx context.Context) error {
|
||||
calls.Add(1)
|
||||
return nil
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
s := New(tasks)
|
||||
go s.Start(ctx)
|
||||
|
||||
time.Sleep(30 * time.Millisecond)
|
||||
cancel()
|
||||
time.Sleep(10 * time.Millisecond)
|
||||
|
||||
snapshot := calls.Load()
|
||||
time.Sleep(30 * time.Millisecond)
|
||||
|
||||
if calls.Load() != snapshot {
|
||||
t.Errorf("expected task to stop after cancel, got %d more calls", calls.Load()-snapshot)
|
||||
}
|
||||
}
|
||||
|
||||
func TestScheduler_MultipleTasksRunIndependently(t *testing.T) {
|
||||
var calls1, calls2 atomic.Int32
|
||||
|
||||
tasks := []Task{
|
||||
{
|
||||
Name: "task1",
|
||||
Interval: 10 * time.Millisecond,
|
||||
Run: func(ctx context.Context) error {
|
||||
calls1.Add(1)
|
||||
return nil
|
||||
},
|
||||
},
|
||||
{
|
||||
Name: "task2",
|
||||
Interval: 10 * time.Millisecond,
|
||||
Run: func(ctx context.Context) error {
|
||||
calls2.Add(1)
|
||||
return nil
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
s := New(tasks)
|
||||
go s.Start(ctx)
|
||||
|
||||
time.Sleep(50 * time.Millisecond)
|
||||
if calls1.Load() == 0 {
|
||||
t.Error("expected task1 to have run")
|
||||
}
|
||||
if calls2.Load() == 0 {
|
||||
t.Error("expected task2 to have run")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user