github.com/annwntech/go-micro/v2@v2.9.5/util/registry/util_test.go (about)

     1  package registry
     2  
     3  import (
     4  	"os"
     5  	"testing"
     6  
     7  	"github.com/annwntech/go-micro/v2/registry"
     8  )
     9  
    10  func TestRemove(t *testing.T) {
    11  	services := []*registry.Service{
    12  		{
    13  			Name:    "foo",
    14  			Version: "1.0.0",
    15  			Nodes: []*registry.Node{
    16  				{
    17  					Id:      "foo-123",
    18  					Address: "localhost:9999",
    19  				},
    20  			},
    21  		},
    22  		{
    23  			Name:    "foo",
    24  			Version: "1.0.0",
    25  			Nodes: []*registry.Node{
    26  				{
    27  					Id:      "foo-123",
    28  					Address: "localhost:6666",
    29  				},
    30  			},
    31  		},
    32  	}
    33  
    34  	servs := Remove([]*registry.Service{services[0]}, []*registry.Service{services[1]})
    35  	if i := len(servs); i > 0 {
    36  		t.Errorf("Expected 0 nodes, got %d: %+v", i, servs)
    37  	}
    38  	if len(os.Getenv("IN_TRAVIS_CI")) == 0 {
    39  		t.Logf("Services %+v", servs)
    40  	}
    41  }
    42  
    43  func TestRemoveNodes(t *testing.T) {
    44  	services := []*registry.Service{
    45  		{
    46  			Name:    "foo",
    47  			Version: "1.0.0",
    48  			Nodes: []*registry.Node{
    49  				{
    50  					Id:      "foo-123",
    51  					Address: "localhost:9999",
    52  				},
    53  				{
    54  					Id:      "foo-321",
    55  					Address: "localhost:6666",
    56  				},
    57  			},
    58  		},
    59  		{
    60  			Name:    "foo",
    61  			Version: "1.0.0",
    62  			Nodes: []*registry.Node{
    63  				{
    64  					Id:      "foo-123",
    65  					Address: "localhost:6666",
    66  				},
    67  			},
    68  		},
    69  	}
    70  
    71  	nodes := delNodes(services[0].Nodes, services[1].Nodes)
    72  	if i := len(nodes); i != 1 {
    73  		t.Errorf("Expected only 1 node, got %d: %+v", i, nodes)
    74  	}
    75  	if len(os.Getenv("IN_TRAVIS_CI")) == 0 {
    76  		t.Logf("Nodes %+v", nodes)
    77  	}
    78  }