40 lines
805 B
Go
40 lines
805 B
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package lib
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestStopChannelContext(t *testing.T) {
|
|
ch := make(chan struct{})
|
|
|
|
ctx := StopChannelContext{StopCh: ch}
|
|
|
|
select {
|
|
case <-ctx.Done():
|
|
require.FailNow(t, "StopChannelContext should not be done yet")
|
|
default:
|
|
// do nothing things are good
|
|
}
|
|
|
|
close(ch)
|
|
|
|
select {
|
|
case <-ctx.Done():
|
|
// things are good, as we are done
|
|
default:
|
|
require.FailNow(t, "StopChannelContext should be done")
|
|
}
|
|
|
|
// issue it twice to ensure that we indefinitely return the
|
|
// same value - this is what the context interface says is
|
|
// the correct behavior.
|
|
require.Equal(t, context.Canceled, ctx.Err())
|
|
require.Equal(t, context.Canceled, ctx.Err())
|
|
}
|