github.com/Myriad-Dreamin/tarus@v0.0.0-20220422082640-5379b6998284/cmd/tests/containerd-judge/single_test.go (about)

     1  package main
     2  
     3  import (
     4  	"context"
     5  	oci_judge "github.com/Myriad-Dreamin/tarus/pkg/tarus-judge/oci"
     6  	"sync/atomic"
     7  	"testing"
     8  )
     9  
    10  var client *oci_judge.ContainerdJudgeServiceServer
    11  
    12  func init() {
    13  	var err error
    14  	// _ = os.Chdir("../../../")
    15  
    16  	client, err = oci_judge.NewContainerdServer()
    17  	if err != nil {
    18  		panic(err)
    19  	}
    20  }
    21  
    22  func BenchmarkEcho(b *testing.B) {
    23  	var n int32
    24  	b.RunParallel(func(pb *testing.PB) {
    25  		ctx := context.WithValue(context.Background(), "No", atomic.AddInt32(&n, 1))
    26  		for pb.Next() {
    27  			echoTest(client, ctx)
    28  		}
    29  	})
    30  }