vendor/google.golang.org/grpc/resolver/dns/go17_test.go

   1 // +build go1.6, !go1.8
   2
   3 /*
   4  *
   5  * Copyright 2017 gRPC authors.
   6  *
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *     http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  *
  19  */
  20
  21 package dns
  22
  23 import (
  24         "fmt"
  25         "net"
  26
  27         "golang.org/x/net/context"
  28 )
  29
  30 var errForInvalidTarget = fmt.Errorf("invalid target address [2001:db8:a0b:12f0::1, error info: missing ']' in address [2001:db8:a0b:12f0::1:443")
  31
  32 func replaceNetFunc() func() {
  33         oldLookupHost := lookupHost
  34         oldLookupSRV := lookupSRV
  35         oldLookupTXT := lookupTXT
  36         lookupHost = func(ctx context.Context, host string) ([]string, error) {
  37                 return hostLookup(host)
  38         }
  39         lookupSRV = func(ctx context.Context, service, proto, name string) (string, []*net.SRV, error) {
  40                 return srvLookup(service, proto, name)
  41         }
  42         lookupTXT = func(ctx context.Context, host string) ([]string, error) {
  43                 return txtLookup(host)
  44         }
  45         return func() {
  46                 lookupHost = oldLookupHost
  47                 lookupSRV = oldLookupSRV
  48                 lookupTXT = oldLookupTXT
  49         }
  50 }