runtimes/google/ipc/stream/vc/cancel_test.go

package vc

import (
	"testing"
	"time"
)

func TestCancelChannelNil(t *testing.T) {
	var zero time.Time
	if cancel := cancelChannel(zero); cancel != nil {
		t.Errorf("Got %v want nil with deadline %v", cancel, zero)
	}
}

func TestCancelChannel(t *testing.T) {
	deadline := time.Now()
	cancel := cancelChannel(deadline)
	if cancel == nil {
		t.Fatalf("Got nil channel for deadline %v", deadline)
	}
	if _, ok := <-cancel; ok {
		t.Errorf("Expected channel to be closed")
	}
}