Function TestTerminateAndWaitForClose

base/util_test.go:1576–1674

(t *testing.T)

}

func TestTerminateAndWaitForClose(t *testing.T) {
	LongRunningTest(t)

	SetUpTestLogging(t, LevelInfo, KeySGTest)

	tests := []struct {
		name       string
		terminator chan struct{}
		done       chan struct{}
		fn         func(ctx context.Context, terminator chan struct{}, done chan struct{})
		timeout    time.Duration
		wantErr    bool
	}{
		{
			name:       "terminate and done immediate",
			terminator: make(chan struct{}),
			done:       make(chan struct{}),
			fn: func(ctx context.Context, t chan struct{}, d chan struct{}) {
				<-t
				InfofCtx(ctx, KeySGTest, "got t closing d")
				close(d)
			},
			timeout: time.Millisecond * 100,
			wantErr: false,
		},
		{
			name:       "terminate and done within timeout",
			terminator: make(chan struct{}),
			done:       make(chan struct{}),
			fn: func(ctx context.Context, t chan struct{}, d chan struct{}) {
				<-t
				InfofCtx(ctx, KeySGTest, "got t waiting to close d")
				select {
				case <-time.After(time.Millisecond * 100):
					InfofCtx(ctx, KeySGTest, "closing d")
					close(d)
				case <-ctx.Done():
					InfofCtx(ctx, KeySGTest, "test context done")
				}
			},
			timeout: time.Second * 10,
			wantErr: false,
		},
		{
			name:       "terminate and done after timeout",
			terminator: make(chan struct{}),
			done:       make(chan struct{}),
			fn: func(ctx context.Context, t chan struct{}, d chan struct{}) {
				<-t
				InfofCtx(ctx, KeySGTest, "got t waiting to close d")
				select {
				case <-time.After(time.Second * 10):
					InfofCtx(ctx, KeySGTest, "closing d")
					close(d)
				case <-ctx.Done():
					InfofCtx(ctx, KeySGTest, "test context done")
				}
			},