Do not call testdata.Path when defining flags (#1394)

This commit is contained in:
Menghan Li
2017-07-27 14:00:18 -07:00
committed by dfawley
parent 3ddcdc268d
commit e64ff5d75c
3 changed files with 22 additions and 17 deletions

View File

@ -33,8 +33,8 @@ import (
var (
useTLS = flag.Bool("use_tls", false, "Connection uses TLS if true, else plain TCP")
certFile = flag.String("tls_cert_file", testdata.Path("server1.pem"), "The TLS cert file")
keyFile = flag.String("tls_key_file", testdata.Path("server1.key"), "The TLS key file")
certFile = flag.String("tls_cert_file", "", "The TLS cert file")
keyFile = flag.String("tls_key_file", "", "The TLS key file")
port = flag.Int("port", 10000, "The server port")
)
@ -47,6 +47,12 @@ func main() {
}
var opts []grpc.ServerOption
if *useTLS {
if *certFile == "" {
testdata.Path("server1.pem")
}
if *keyFile == "" {
testdata.Path("server1.key")
}
creds, err := credentials.NewServerTLSFromFile(*certFile, *keyFile)
if err != nil {
grpclog.Fatalf("Failed to generate credentials %v", err)