You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
742 B
43 lines
742 B
package service
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type testService struct {
|
|
BaseService
|
|
}
|
|
|
|
func (testService) OnStop() {}
|
|
func (testService) OnStart(context.Context) error {
|
|
return nil
|
|
}
|
|
|
|
func TestBaseServiceWait(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
ts := &testService{}
|
|
ts.BaseService = *NewBaseService(nil, "TestService", ts)
|
|
err := ts.Start(ctx)
|
|
require.NoError(t, err)
|
|
|
|
waitFinished := make(chan struct{})
|
|
go func() {
|
|
ts.Wait()
|
|
waitFinished <- struct{}{}
|
|
}()
|
|
|
|
go cancel()
|
|
|
|
select {
|
|
case <-waitFinished:
|
|
// all good
|
|
case <-time.After(100 * time.Millisecond):
|
|
t.Fatal("expected Wait() to finish within 100 ms.")
|
|
}
|
|
}
|