From 982c3f155085d95fd825aa7b65902fc23813e533 Mon Sep 17 00:00:00 2001
From: Joe Zou <yixian.zou@gmail.com>
Date: Mon, 13 Jul 2020 00:08:54 +0800
Subject: [PATCH] Fix router bug and revert test case
---
config/router_config.go | 2 +-
config/router_config_test.go | 12 ++++++++++++
2 files changed, 13 insertions(+), 1 deletion(-)
diff --git a/config/router_config.go b/config/router_config.go
index 16943d96b..16a2bec91 100644
--- a/config/router_config.go
+++ b/config/router_config.go
@@ -44,7 +44,7 @@ func RouterInit(confRouterFile string) error {
r, e := factory.NewFileRouter(bytes)
if e == nil {
url := r.URL()
- routerURLSet.Add(url)
+ routerURLSet.Add(&url)
return nil
}
logger.Warnf("router config type %s create fail {%v}\n", k, e)
diff --git a/config/router_config_test.go b/config/router_config_test.go
index bf189b600..72e51c1c8 100644
--- a/config/router_config_test.go
+++ b/config/router_config_test.go
@@ -52,3 +52,15 @@ func TestString(t *testing.T) {
assert.Equal(t, n2[0], "a1")
assert.Equal(t, n2[1], "")
}
+
+func TestRouterInit(t *testing.T) {
+ errPro := RouterInit(errorTestYML)
+ assert.Error(t, errPro)
+
+ assert.Equal(t, 0, routerURLSet.Size())
+
+ errPro = RouterInit(testYML)
+ assert.NoError(t, errPro)
+
+ assert.Equal(t, 1, routerURLSet.Size())
+}
--
GitLab