@@ -163,15 +163,15 @@ func testUpdateVPCNetworkconfigNetworkInfo(t *testing.T) {
163163	networkInfo  :=  assureNetworkInfo (t , ns , ns )
164164	require .NotNil (t , networkInfo )
165165
166- 	networkInfoNew  :=  getVPCPathFromNetworkInfo (t , ns , networkInfo .Name )
166+ 	networkInfoNew  :=  getNetworkInfoWithPrivateIPs (t , ns , networkInfo .Name )
167167	privateIPs  :=  networkInfoNew .VPCs [0 ].PrivateIPs 
168168	assert .Contains (t , privateIPs , customizedPrivateCIDR1 , "privateIPs %s should contain %s" , privateIPs , customizedPrivateCIDR1 )
169169	assert .Contains (t , privateIPs , customizedPrivateCIDR2 , "privateIPs %s should contain %s" , privateIPs , customizedPrivateCIDR1 )
170170
171171	vncPath , _  :=  filepath .Abs ("./manifest/testVPC/customize_networkconfig_updated.yaml" )
172172	require .NoError (t , applyYAML (vncPath , "" ))
173173
174- 	networkInfoNew  =  getVPCPathFromNetworkInfo (t , ns , networkInfo .Name )
174+ 	networkInfoNew  =  getNetworkInfoWithPrivateIPs (t , ns , networkInfo .Name )
175175	privateIPs  =  networkInfoNew .VPCs [0 ].PrivateIPs 
176176	assert .Contains (t , privateIPs , customizedPrivateCIDR3 , "privateIPs %s should contain %s" , privateIPs , customizedPrivateCIDR3 )
177177}
@@ -262,16 +262,16 @@ func assureNamespaceDeleted(t *testing.T, ns string) {
262262	return 
263263}
264264
265- func  getVPCPathFromNetworkInfo (t  * testing.T , ns , networkInfoName  string ) (networkInfo  * v1alpha1.NetworkInfo ) {
265+ func  getNetworkInfoWithPrivateIPs (t  * testing.T , ns , networkInfoName  string ) (networkInfo  * v1alpha1.NetworkInfo ) {
266266	deadlineCtx , deadlineCancel  :=  context .WithTimeout (context .Background (), defaultTimeout )
267267	defer  deadlineCancel ()
268268	err  :=  wait .PollUntilContextTimeout (deadlineCtx , 1 * time .Second , defaultTimeout , false , func (ctx  context.Context ) (done  bool , err  error ) {
269269		networkInfo , err  =  testData .crdClientset .CrdV1alpha1 ().NetworkInfos (ns ).Get (ctx , networkInfoName , v1.GetOptions {})
270270		if  err  !=  nil  {
271- 			t .Logf ("Check VPC path  of vpcnetworkconfigurations : %v, error: %+v" , networkInfo , err )
272- 			return  false , fmt .Errorf ("error when waiting for vpcnetworkconfigurations VPC path : %s" , networkInfoName )
271+ 			t .Logf ("Check private ips  of networkinfo : %v, error: %+v" , networkInfo , err )
272+ 			return  false , fmt .Errorf ("error when waiting for vpcnetworkinfo private ips : %s" , networkInfoName )
273273		}
274- 		if  len (networkInfo .VPCs ) >  0  &&  networkInfo .VPCs [0 ].VPCPath   !=   ""  {
274+ 		if  len (networkInfo .VPCs ) >  0  &&  len ( networkInfo .VPCs [0 ].PrivateIPs )  >   0  {
275275			return  true , nil 
276276		}
277277		return  false , nil 
0 commit comments