github.com/google/cloudprober@v0.11.3/config/runconfig/runconfig_test.go (about)

     1  package runconfig
     2  
     3  import (
     4  	"testing"
     5  
     6  	"google.golang.org/grpc"
     7  )
     8  
     9  func TestRunConfig(t *testing.T) {
    10  	if srv := DefaultGRPCServer(); srv != nil {
    11  		t.Fatalf("RunConfig has server unexpectedly set. Got %v Want nil", srv)
    12  	}
    13  	testSrv := grpc.NewServer()
    14  	if testSrv == nil {
    15  		t.Fatal("Unable to create a test gRPC server")
    16  	}
    17  	if err := SetDefaultGRPCServer(testSrv); err != nil {
    18  		t.Fatalf("Unable to set default gRPC server: %v", err)
    19  	}
    20  	if srv := DefaultGRPCServer(); srv != testSrv {
    21  		t.Fatalf("Error retrieving stored service. Got %v Want %v", srv, testSrv)
    22  	}
    23  	if err := SetDefaultGRPCServer(testSrv); err == nil {
    24  		t.Errorf("RunConfig allowed overriding of an already set variable.")
    25  	}
    26  }