github.com/tilt-dev/wat@v0.0.2-0.20180626175338-9349b638e250/errors/errors_test.go (about)

     1  package errors
     2  
     3  import (
     4  	"context"
     5  	"testing"
     6  
     7  	"google.golang.org/grpc"
     8  	"google.golang.org/grpc/codes"
     9  )
    10  
    11  func TestPreserveContextCanceled(t *testing.T) {
    12  	ctx, cancel := context.WithCancel(context.Background())
    13  	cancel()
    14  
    15  	err := ctx.Err()
    16  	if !IsCanceled(err) {
    17  		t.Errorf("Expected canceled: %v", err)
    18  	}
    19  
    20  	err = Propagatef(err, "Msg1")
    21  	if !IsCanceled(err) {
    22  		t.Errorf("Expected canceled: %v", err)
    23  	}
    24  
    25  	err = Propagatef(err, "Msg2(%s)", "hi")
    26  	if !IsCanceled(err) {
    27  		t.Errorf("Expected canceled: %v", err)
    28  	}
    29  }
    30  
    31  func TestPreserveGRPCCanceled(t *testing.T) {
    32  	err := grpc.Errorf(codes.Canceled, "Canceled")
    33  	if !IsCanceled(err) {
    34  		t.Errorf("Expected canceled: %v", err)
    35  	}
    36  
    37  	err = Propagatef(err, "Msg1")
    38  	if !IsCanceled(err) {
    39  		t.Errorf("Expected canceled: %v", err)
    40  	}
    41  }