update router test

antlinker · Jan 3, 2018 · e96999b · e96999b
1 parent 8a21f38
commit e96999b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/router_test.go b/router_test.go
@@ -24,7 +24,7 @@ import (
 )
 
 func TestTextRouter_Dispatch(t *testing.T) {
-	r := &TextRouter{}
+	r := NewTextRouter()
 	count := 0
 	testTopics := make(map[string]string)
 	const topicCount = 1000
@@ -61,7 +61,7 @@ func TestTextRouter_Dispatch(t *testing.T) {
 }
 
 func TestRegexRouter_Dispatch(t *testing.T) {
-	r := &RegexRouter{}
+	r := NewRegexRouter()
 	allCount, prefixCount, numCount := 0, 0, 0
 
 	r.Handle(".*", func(topic string, code SubAckCode, msg []byte) {