open-consul/lib/stop_context_test.go

40 lines
805 B
Go
Raw Permalink Normal View History

// 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())
}