Test GRPC

grpc_test.go

// TestRMsvrSuccess test success case, this can't fail
func (s *FailTestSuite) TestRMsvrSuccess() {
    err := StartResourceMgr(s.chErr, config.Config.QueueSize, config.Config.RMServerAddr, config.Config.QueuedTaskTTL)
    assert.NoError(s.T(), err)

    var conn *grpc.ClientConn

    for ts := 250; ts <= 1000; ts += 250 {
        time.Sleep(time.Duration(ts) * time.Millisecond)
        conn, err = grpc.Dial(config.Config.RMServerAddr, grpc.WithInsecure(), grpc.WithTimeout(5*time.Second))
        if err == nil {
            break
        }
    }
    assert.NoError(s.T(), err, "should have no error")

    cli := rmpb.NewServiceClient(conn)
    assert.NoError(s.T(), err, "should have no error")

    stopreq := &rmpb.Empty{}
    _, err = cli.StopService(context.Background(), stopreq)
    assert.NoError(s.T(), err, "should have no error")
}