func TestKeyspaceNodesTabletTypes(t *testing.T) { input := KeyspaceNodes{ ShardNodes: []*ShardNodes{ &ShardNodes{ TabletNodes: []*TabletNodesByType{ &TabletNodesByType{ TabletType: pb.TabletType_REPLICA, }, }, }, &ShardNodes{ TabletNodes: []*TabletNodesByType{ &TabletNodesByType{ TabletType: pb.TabletType_MASTER, }, &TabletNodesByType{ TabletType: pb.TabletType_REPLICA, }, }, }, }, } want := topo.MakeStringTypeList([]pb.TabletType{pb.TabletType_REPLICA, pb.TabletType_MASTER}) got := topo.MakeStringTypeList(input.TabletTypes()) if !reflect.DeepEqual(got, want) { t.Errorf("KeyspaceNodes.TabletTypes() = %v, want %v", got, want) } }
func TestKeyspaceNodesTabletTypes(t *testing.T) { input := KeyspaceNodes{ ShardNodes: []*ShardNodes{ &ShardNodes{TabletNodes: TabletNodesByType{topo.TYPE_REPLICA: nil}}, &ShardNodes{TabletNodes: TabletNodesByType{topo.TYPE_MASTER: nil, topo.TYPE_REPLICA: nil}}, }, } want := topo.MakeStringTypeList([]topo.TabletType{topo.TYPE_REPLICA, topo.TYPE_MASTER}) got := topo.MakeStringTypeList(input.TabletTypes()) if !reflect.DeepEqual(got, want) { t.Errorf("KeyspaceNodes.TabletTypes() = %v, want %v", got, want) } }