@@ -259,7 +259,7 @@ func (c *Cluster) ProtoMembers() []*pb.Member {
259
259
return ms
260
260
}
261
261
262
- func (c * Cluster ) mustNewMember (t testutil.TB ) * Member {
262
+ func (c * Cluster ) MustNewMember (t testutil.TB ) * Member {
263
263
memberNumber := c .LastMemberNum
264
264
c .LastMemberNum ++
265
265
@@ -299,7 +299,7 @@ func (c *Cluster) mustNewMember(t testutil.TB) *Member {
299
299
300
300
// addMember return PeerURLs of the added member.
301
301
func (c * Cluster ) addMember (t testutil.TB ) types.URLs {
302
- m := c .mustNewMember (t )
302
+ m := c .MustNewMember (t )
303
303
304
304
scheme := SchemeFromTLSInfo (c .Cfg .PeerTLS )
305
305
@@ -1394,7 +1394,7 @@ func NewCluster(t testutil.TB, cfg *ClusterConfig) *Cluster {
1394
1394
c := & Cluster {Cfg : cfg }
1395
1395
ms := make ([]* Member , cfg .Size )
1396
1396
for i := 0 ; i < cfg .Size ; i ++ {
1397
- ms [i ] = c .mustNewMember (t )
1397
+ ms [i ] = c .MustNewMember (t )
1398
1398
}
1399
1399
c .Members = ms
1400
1400
if err := c .fillClusterForMembers (); err != nil {
@@ -1580,7 +1580,7 @@ func (c *Cluster) GetLearnerMembers() ([]*pb.Member, error) {
1580
1580
// AddAndLaunchLearnerMember creates a learner member, adds it to Cluster
1581
1581
// via v3 MemberAdd API, and then launches the new member.
1582
1582
func (c * Cluster ) AddAndLaunchLearnerMember (t testutil.TB ) {
1583
- m := c .mustNewMember (t )
1583
+ m := c .MustNewMember (t )
1584
1584
m .IsLearner = true
1585
1585
1586
1586
scheme := SchemeFromTLSInfo (c .Cfg .PeerTLS )
@@ -1679,9 +1679,8 @@ func (p SortableProtoMemberSliceByPeerURLs) Less(i, j int) bool {
1679
1679
}
1680
1680
func (p SortableProtoMemberSliceByPeerURLs ) Swap (i , j int ) { p [i ], p [j ] = p [j ], p [i ] }
1681
1681
1682
- // MustNewMember creates a new member instance based on the response of V3 Member Add API.
1683
- func (c * Cluster ) MustNewMember (t testutil.TB , resp * clientv3.MemberAddResponse ) * Member {
1684
- m := c .mustNewMember (t )
1682
+ // InitializeMemberWithResponse initializes a member with the response
1683
+ func (c * Cluster ) InitializeMemberWithResponse (t testutil.TB , m * Member , resp * clientv3.MemberAddResponse ) {
1685
1684
m .IsLearner = resp .Member .IsLearner
1686
1685
m .NewCluster = false
1687
1686
@@ -1691,5 +1690,4 @@ func (c *Cluster) MustNewMember(t testutil.TB, resp *clientv3.MemberAddResponse)
1691
1690
}
1692
1691
m .InitialPeerURLsMap [m .Name ] = types .MustNewURLs (resp .Member .PeerURLs )
1693
1692
c .Members = append (c .Members , m )
1694
- return m
1695
1693
}
0 commit comments