test/context/context_test.go

package context_test

import (
	"context"
	"fmt"
	"testing"
	"time"

	"gitee.com/wecisecode/util/logger"
)

func TestContext(t *testing.T) {
	ctx, cancel := context.WithCancelCause(context.Background())
	cancel(fmt.Errorf("%s", "cancel"))

	go func() {
		for {
			select {
			case <-ctx.Done():
				err := ctx.Err()
				if err == context.Canceled {
					err = context.Cause(ctx)
				}
				logger.Info("context done", err)
			default:
			}
			time.Sleep(5 * time.Second)
		}
	}()

	time.Sleep(1 * time.Hour)
}

func TestContextDone(t *testing.T) {
	ctx, cancel := context.WithCancelCause(context.Background())

	go func() {
		for {
			select {
			case <-ctx.Done():
				err := ctx.Err()
				if err == context.Canceled {
					err = context.Cause(ctx)
				}
				logger.Info("context done 1", err)
			default:
			}
			time.Sleep(5 * time.Second)
		}
	}()

	go func() {
		for {
			select {
			case <-ctx.Done():
				err := ctx.Err()
				if err == context.Canceled {
					err = context.Cause(ctx)
				}
				logger.Info("context done 2", err)
			default:
			}
			time.Sleep(5 * time.Second)
		}
	}()

	go func() {
		for {
			select {
			case <-ctx.Done():
				err := ctx.Err()
				if err == context.Canceled {
					err = context.Cause(ctx)
				}
				logger.Info("context done 3", err)
			default:
			}
			time.Sleep(5 * time.Second)
		}
	}()

	cancel(fmt.Errorf("%s", "cancel"))

	time.Sleep(1 * time.Hour)
}