credentials.TLSInfo: export State

This commit is contained in:
Tamir Duberstein 2015-09-06 17:14:41 -04:00
parent 1a700f0ee9
commit a6fe03ae2e
2 changed files with 6 additions and 2 deletions

View File

@ -119,7 +119,7 @@ type TransportAuthenticator interface {
// TLSInfo contains the auth information for a TLS authenticated connection.
// It implements the AuthInfo interface.
type TLSInfo struct {
state tls.ConnectionState
State tls.ConnectionState
}
func (t TLSInfo) AuthType() string {

View File

@ -117,16 +117,20 @@ func (s *testServer) UnaryCall(ctx context.Context, in *testpb.SimpleRequest) (*
if !ok {
grpclog.Fatalf("Failed to get AuthInfo from ctx.")
}
var authType string
var authType, serverName string
switch info := authInfo.(type) {
case credentials.TLSInfo:
authType = info.AuthType()
serverName = info.State.ServerName
default:
grpclog.Fatalf("Unknown AuthInfo type")
}
if authType != s.security {
grpclog.Fatalf("Wrong auth type: got %q, want %q", authType, s.security)
}
if serverName != "x.test.youtube.com" {
return nil, fmt.Errorf("Unknown server name %q", serverName)
}
}
// Simulate some service delay.