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

   1 // +build 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         "context"
  25         "fmt"
  26         "net"
  27 )
  28
  29 var errForInvalidTarget = fmt.Errorf("invalid target address [2001:db8:a0b:12f0::1, error info: address [2001:db8:a0b:12f0::1:443: missing ']' in address")
  30
  31 func replaceNetFunc() func() {
  32         oldLookupHost := lookupHost
  33         oldLookupSRV := lookupSRV
  34         oldLookupTXT := lookupTXT
  35         lookupHost = func(ctx context.Context, host string) ([]string, error) {
  36                 return hostLookup(host)
  37         }
  38         lookupSRV = func(ctx context.Context, service, proto, name string) (string, []*net.SRV, error) {
  39                 return srvLookup(service, proto, name)
  40         }
  41         lookupTXT = func(ctx context.Context, host string) ([]string, error) {
  42                 return txtLookup(host)
  43         }
  44         return func() {
  45                 lookupHost = oldLookupHost
  46                 lookupSRV = oldLookupSRV
  47                 lookupTXT = oldLookupTXT
  48         }
  49 }