diff --git a/protocol/rest/rest_invoker_test.go b/protocol/rest/rest_invoker_test.go index 103877036739307349f1d3b449ddf263e466ede5..80bd83ecadd94b8010952749b687cbb9e91dc73e 100644 --- a/protocol/rest/rest_invoker_test.go +++ b/protocol/rest/rest_invoker_test.go @@ -92,4 +92,6 @@ func TestRestInvoker_Invoke(t *testing.T) { assert.Equal(t, now.Unix(), res.Result().(*User).Time.Unix()) assert.Equal(t, int32(23), res.Result().(*User).Age) assert.Equal(t, "username", res.Result().(*User).Name) + err = common.ServiceMap.UnRegister(url.Protocol, "com.ikurento.user.UserProvider") + assert.NoError(t, err) } diff --git a/protocol/rest/rest_protocol_test.go b/protocol/rest/rest_protocol_test.go index 520fa82a63f4fe3594874de2e068ed7be736d637..4ed72b76c688df08e27d79dc609383592507d91b 100644 --- a/protocol/rest/rest_protocol_test.go +++ b/protocol/rest/rest_protocol_test.go @@ -104,6 +104,8 @@ func TestJsonrpcProtocol_Export(t *testing.T) { proto.Destroy() _, ok = proto.(*RestProtocol).serverMap[url.Location] assert.False(t, ok) + err = common.ServiceMap.UnRegister(url.Protocol, "com.ikurento.user.UserProvider") + assert.NoError(t, err) } type UserProvider struct {