| 14 | ) |
| 15 | |
| 16 | func TestRPCConcurrentCall(t *testing.T) { |
| 17 | if !testutils.IsSingleTesting() { |
| 18 | return |
| 19 | } |
| 20 | |
| 21 | rpcClient, err := rpc.SharedRPC() |
| 22 | if err != nil { |
| 23 | t.Fatal(err) |
| 24 | } |
| 25 | |
| 26 | var before = time.Now() |
| 27 | defer func() { |
| 28 | t.Log("cost:", time.Since(before).Seconds()*1000, "ms") |
| 29 | }() |
| 30 | |
| 31 | var concurrent = 3 |
| 32 | |
| 33 | var wg = sync.WaitGroup{} |
| 34 | wg.Add(concurrent) |
| 35 | |
| 36 | for i := 0; i < concurrent; i++ { |
| 37 | go func() { |
| 38 | defer wg.Done() |
| 39 | |
| 40 | _, err = rpcClient.NodeRPC.FindCurrentNodeConfig(rpcClient.Context(), &pb.FindCurrentNodeConfigRequest{}) |
| 41 | if err != nil { |
| 42 | t.Log(err) |
| 43 | } |
| 44 | }() |
| 45 | } |
| 46 | |
| 47 | wg.Wait() |
| 48 | } |
| 49 | |
| 50 | func TestRPC_Retry(t *testing.T) { |
| 51 | if !testutils.IsSingleTesting() { |