github.com/keysonZZZ/kmg@v0.0.0-20151121023212-05317bfd7d39/kmgRpc/testPackage/generated_test.go (about)

     1  package testPackage
     2  
     3  import (
     4  	"testing"
     5  	"time"
     6  
     7  	"github.com/bronze1man/kmg/kmgCrypto"
     8  	"github.com/bronze1man/kmg/kmgTest"
     9  	"github.com/bronze1man/kmg/kmgTime"
    10  )
    11  
    12  func TestGenerated(ot *testing.T) {
    13  	psk := kmgCrypto.Get32PskFromString("abc")
    14  	closer := ListenAndServe_Demo(":34895", &Demo{}, psk)
    15  	defer closer()
    16  	client := NewClient_Demo("http://127.0.0.1:34895/f", psk)
    17  	info, err := client.PostScoreInt("LbId", 1)
    18  	kmgTest.Equal(err, nil)
    19  	kmgTest.Equal(info, "LbId")
    20  
    21  	info, err = client.PostScoreInt("LbId", 2)
    22  	kmgTest.Equal(info, "")
    23  	kmgTest.Ok(err != nil, err)
    24  	kmgTest.Equal(err.Error(), "Score!=1")
    25  
    26  	info, err = client.DemoFunc8(DemoRequest{}, &DemoRequest{}, 1)
    27  	kmgTest.Equal(info, "info1")
    28  
    29  	info, err = client.DemoFunc8(DemoRequest{}, &DemoRequest{}, 2)
    30  	kmgTest.Equal(info, "info")
    31  
    32  	inT := kmgTime.MustParseAutoInDefault("2001-01-01 01:01:01")
    33  	outT, err := client.DemoTime(inT)
    34  	kmgTest.Equal(err, nil)
    35  	kmgTest.Ok(outT.Equal(kmgTime.MustParseAutoInDefault("2001-01-01T02:01:01.001000001+08:00")), outT)
    36  
    37  	ip, err := client.DemoClientIp()
    38  	kmgTest.Equal(err, nil)
    39  	kmgTest.Equal(ip, "127.0.0.1")
    40  }
    41  
    42  func BenchmarkGenerated(ot *testing.B) {
    43  	psk := kmgCrypto.Get32PskFromString("abcd")
    44  	closer := ListenAndServe_Demo(":34896", &Demo{}, psk)
    45  	defer func() {
    46  		closer()
    47  		time.Sleep(10 * time.Millisecond)
    48  	}()
    49  	client := NewClient_Demo("http://127.0.0.1:34896/f", psk)
    50  	ot.ResetTimer()
    51  	for i := 0; i < ot.N; i++ {
    52  		info, err := client.PostScoreInt("LbId", 1)
    53  		kmgTest.Equal(err, nil)
    54  		kmgTest.Equal(info, "LbId")
    55  	}
    56  	ot.StopTimer()
    57  	// 结果 770 qps BenchmarkGenerated	    1000	   1298511 ns/op
    58  }