@@ -197,7 +197,7 @@ func TestCreateInstance(t *testing.T) {
197
197
time .Sleep (5 * time .Second )
198
198
instanceId := fmt .Sprintf ("%s_%s_%d" , appId , host , startPort )
199
199
code := eurekaSrv .deregisterInstance (context .Background (), namespace , appId , instanceId , false )
200
- assert .Equal (t , api .ExecuteSuccess , code )
200
+ assert .Equal (t , api .ExecuteSuccess , code , fmt . Sprintf ( "%d" , code ) )
201
201
time .Sleep (20 * time .Second )
202
202
203
203
deltaReq := restful .NewRequest (httpRequest )
@@ -244,18 +244,17 @@ func Test_EurekaWrite(t *testing.T) {
244
244
injectRestfulReqPathParameters (t , restfulReq , map [string ]string {
245
245
ParamAppId : mockIns .AppName ,
246
246
})
247
- // 这里是异步注册
248
247
eurekaSrv .RegisterApplication (restfulReq , restful .NewResponse (mockRsp ))
249
248
assert .Equal (t , http .StatusNoContent , mockRsp .statusCode )
250
249
assert .Equal (t , restfulReq .Attribute (statusCodeHeader ), uint32 (apimodel .Code_ExecuteSuccess ))
251
250
252
- time . Sleep ( 5 * time . Second )
253
- saveIns , err := eurekaSrv . originDiscoverSvr . Cache (). GetStore () .GetInstance (mockIns .InstanceId )
251
+ _ = discoverSuit . CacheMgr (). TestUpdate ( )
252
+ saveIns , err := discoverSuit . Storage .GetInstance (mockIns .InstanceId )
254
253
assert .NoError (t , err )
255
254
assert .NotNil (t , saveIns )
256
255
257
256
t .Run ("UpdateStatus" , func (t * testing.T ) {
258
- t .Run ("StatusUnknown " , func (t * testing.T ) {
257
+ t .Run ("01_StatusUnknown " , func (t * testing.T ) {
259
258
mockReq := httptest .NewRequest ("" , fmt .Sprintf ("http://127.0.0.1:8761/eureka/v2/apps/%s/%s/status" ,
260
259
mockIns .AppName , mockIns .InstanceId ), nil )
261
260
mockReq .PostForm = url.Values {}
@@ -278,7 +277,7 @@ func Test_EurekaWrite(t *testing.T) {
278
277
assert .False (t , saveIns .Isolate ())
279
278
})
280
279
281
- t .Run ("StatusDown " , func (t * testing.T ) {
280
+ t .Run ("02_StatusDown " , func (t * testing.T ) {
282
281
mockReq := httptest .NewRequest ("" , fmt .Sprintf ("http://127.0.0.1:8761/eureka/v2/apps/%s/%s/status" ,
283
282
mockIns .AppName , mockIns .InstanceId ), nil )
284
283
mockReq .PostForm = url.Values {}
@@ -301,7 +300,7 @@ func Test_EurekaWrite(t *testing.T) {
301
300
assert .Equal (t , StatusDown , saveIns .Proto .Metadata [InternalMetadataStatus ])
302
301
})
303
302
304
- t .Run ("StatusUp " , func (t * testing.T ) {
303
+ t .Run ("03_StatusUp " , func (t * testing.T ) {
305
304
mockReq := httptest .NewRequest ("" , fmt .Sprintf ("http://127.0.0.1:8761/eureka/v2/apps/%s/%s/status" ,
306
305
mockIns .AppName , mockIns .InstanceId ), nil )
307
306
mockReq .PostForm = url.Values {}
@@ -324,17 +323,17 @@ func Test_EurekaWrite(t *testing.T) {
324
323
assert .Equal (t , StatusUp , saveIns .Proto .Metadata [InternalMetadataStatus ])
325
324
})
326
325
327
- t .Run ("Polaris_UpdateInstances " , func (t * testing.T ) {
326
+ t .Run ("04_Polaris_UpdateInstances " , func (t * testing.T ) {
328
327
defer func () {
329
- rsp := discoverSuit .OriginDiscoverServer ().UpdateInstances (discoverSuit .DefaultCtx , []* service_manage.Instance {
328
+ rsp := discoverSuit .DiscoverServer ().UpdateInstances (discoverSuit .DefaultCtx , []* service_manage.Instance {
330
329
{
331
330
Id : wrapperspb .String (mockIns .InstanceId ),
332
331
Isolate : wrapperspb .Bool (false ),
333
332
},
334
333
})
335
334
assert .Equal (t , apimodel .Code_ExecuteSuccess , apimodel .Code (rsp .GetCode ().GetValue ()))
336
335
}()
337
- rsp := discoverSuit .OriginDiscoverServer ().UpdateInstances (discoverSuit .DefaultCtx , []* service_manage.Instance {
336
+ rsp := discoverSuit .DiscoverServer ().UpdateInstances (discoverSuit .DefaultCtx , []* service_manage.Instance {
338
337
{
339
338
Id : wrapperspb .String (mockIns .InstanceId ),
340
339
Isolate : wrapperspb .Bool (true ),
@@ -349,8 +348,8 @@ func Test_EurekaWrite(t *testing.T) {
349
348
assert .Equal (t , StatusOutOfService , saveIns .Proto .Metadata [InternalMetadataStatus ])
350
349
})
351
350
352
- t .Run ("Polaris_UpdateInstancesIsolate " , func (t * testing.T ) {
353
- rsp := discoverSuit .OriginDiscoverServer ().UpdateInstances (discoverSuit .DefaultCtx , []* service_manage.Instance {
351
+ t .Run ("05_Polaris_UpdateInstancesIsolate " , func (t * testing.T ) {
352
+ rsp := discoverSuit .DiscoverServer ().UpdateInstances (discoverSuit .DefaultCtx , []* service_manage.Instance {
354
353
{
355
354
Id : wrapperspb .String (mockIns .InstanceId ),
356
355
Isolate : wrapperspb .Bool (true ),
0 commit comments