Skip to content
Snippets Groups Projects
Commit 863dd0fb authored by vito.he's avatar vito.he
Browse files

Mod:make ut run

parent b3dddc88
No related branches found
No related tags found
No related merge requests found
...@@ -328,6 +328,7 @@ func MockInitProviderWithSingleRegistry() { ...@@ -328,6 +328,7 @@ func MockInitProviderWithSingleRegistry() {
Weight: 200, Weight: 200,
}, },
}, },
exported: new(atomic.Bool),
}, },
}, },
Protocols: map[string]*ProtocolConfig{ Protocols: map[string]*ProtocolConfig{
......
...@@ -18,6 +18,7 @@ ...@@ -18,6 +18,7 @@
package config package config
import ( import (
"go.uber.org/atomic"
"testing" "testing"
) )
...@@ -92,6 +93,7 @@ func doInitProvider() { ...@@ -92,6 +93,7 @@ func doInitProvider() {
Weight: 200, Weight: 200,
}, },
}, },
exported: new(atomic.Bool),
}, },
"MockServiceNoRightProtocol": { "MockServiceNoRightProtocol": {
InterfaceName: "com.MockService", InterfaceName: "com.MockService",
...@@ -116,6 +118,7 @@ func doInitProvider() { ...@@ -116,6 +118,7 @@ func doInitProvider() {
Weight: 200, Weight: 200,
}, },
}, },
exported: new(atomic.Bool),
}, },
}, },
Protocols: map[string]*ProtocolConfig{ Protocols: map[string]*ProtocolConfig{
......
...@@ -69,27 +69,29 @@ func TestMetadataService(t *testing.T) { ...@@ -69,27 +69,29 @@ func TestMetadataService(t *testing.T) {
"module=dubbogo+user-info+server&org=ikurento.com&owner=ZX&pid=1447&revision=0.0.1&"+ "module=dubbogo+user-info+server&org=ikurento.com&owner=ZX&pid=1447&revision=0.0.1&"+
"side=provider&timeout=3000&timestamp=1556509797245&group=%v&version=%v&bean.name=%v", protocol, serviceName, group, version, beanName)) "side=provider&timeout=3000&timestamp=1556509797245&group=%v&version=%v&bean.name=%v", protocol, serviceName, group, version, beanName))
mts.ExportURL(u) mts.ExportURL(u)
sets := mts.GetExportedURLs(serviceName, group, version, protocol) sets, _ := mts.GetExportedURLs(serviceName, group, version, protocol)
assert.Equal(t, 1, sets.Size()) assert.Equal(t, 1, sets.Size())
mts.SubscribeURL(u) mts.SubscribeURL(u)
mts.SubscribeURL(u) mts.SubscribeURL(u)
sets2 := mts.GetSubscribedURLs() sets2, _ := mts.GetSubscribedURLs()
assert.Equal(t, 1, sets2.Size()) assert.Equal(t, 1, sets2.Size())
mts.UnexportURL(u) mts.UnexportURL(u)
sets11 := mts.GetExportedURLs(serviceName, group, version, protocol) sets11, _ := mts.GetExportedURLs(serviceName, group, version, protocol)
assert.Equal(t, 0, sets11.Size()) assert.Equal(t, 0, sets11.Size())
mts.UnsubscribeURL(u) mts.UnsubscribeURL(u)
sets22 := mts.GetSubscribedURLs() sets22, _ := mts.GetSubscribedURLs()
assert.Equal(t, 0, sets22.Size()) assert.Equal(t, 0, sets22.Size())
userProvider := &UserProvider{} userProvider := &UserProvider{}
common.ServiceMap.Register(protocol, userProvider) common.ServiceMap.Register(serviceName, protocol, userProvider)
mts.PublishServiceDefinition(u) mts.PublishServiceDefinition(u)
expected := `{"CanonicalName":"com.ikurento.user.UserProvider","CodeSource":"","Methods":[{"Name":"GetUser","ParameterTypes":["slice"],"ReturnType":"ptr","Parameters":null}],"Types":null}` expected := `{"CanonicalName":"com.ikurento.user.UserProvider","CodeSource":"","Methods":[{"Name":"GetUser","ParameterTypes":["slice"],"ReturnType":"ptr","Parameters":null}],"Types":null}`
assert.Equal(t, mts.GetServiceDefinition(serviceName, group, version), expected) def1, _ := mts.GetServiceDefinition(serviceName, group, version)
assert.Equal(t, def1, expected)
serviceKey := definition.ServiceDescriperBuild(serviceName, group, version) serviceKey := definition.ServiceDescriperBuild(serviceName, group, version)
assert.Equal(t, mts.GetServiceDefinitionByServiceKey(serviceKey), expected) def2, _ := mts.GetServiceDefinitionByServiceKey(serviceKey)
assert.Equal(t, def2, expected)
} }
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment