From 1d9a500b727e8879cdaeb9c378ac717001c56ccc Mon Sep 17 00:00:00 2001
From: 苏向夜 <fu050409@163.com>
Date: Sat, 27 Jan 2024 22:27:41 +0800
Subject: :sparkles: feat(router): supports alias router

---
 tests/test_core.py        |  8 ++++----
 tests/test_handlers.py    | 12 ++++++------
 tests/test_interceptor.py |  4 ++--
 3 files changed, 12 insertions(+), 12 deletions(-)

(limited to 'tests')

diff --git a/tests/test_core.py b/tests/test_core.py
index 01d7ecb4..11a8c57d 100644
--- a/tests/test_core.py
+++ b/tests/test_core.py
@@ -4,7 +4,7 @@ from infini.handler import Handler
 from infini.input import Input
 from infini.interceptor import Interceptor
 from infini.output import Output
-from infini.router import ContainsRouter, StartswithRouter
+from infini.router import Contains, Startswith
 
 
 def test_core():
@@ -20,7 +20,7 @@ def test_core():
     interceptor.interceptors = [
         {
             "priority": 1,
-            "router": ContainsRouter("简律纯"),
+            "router": Contains("简律纯"),
             "handler": intercept_jianlvchun,
         }
     ]
@@ -38,12 +38,12 @@ def test_core():
     handler.handlers = [
         {
             "priority": 2,
-            "router": StartswithRouter(".add"),
+            "router": Startswith(".add"),
             "handler": add,
         },
         {
             "priority": 1,
-            "router": StartswithRouter("."),
+            "router": Startswith("."),
             "handler": cmd,
         },
     ]
diff --git a/tests/test_handlers.py b/tests/test_handlers.py
index 00bb6f84..ee96cdab 100644
--- a/tests/test_handlers.py
+++ b/tests/test_handlers.py
@@ -1,7 +1,7 @@
 from infini.handler import Handler
 from infini.input import Input
 from infini.output import Output
-from infini.router import StartswithRouter
+from infini.router import Startswith
 
 
 def test_handler():
@@ -22,12 +22,12 @@ def test_handler():
     handler.handlers = [
         {
             "priority": 2,
-            "router": StartswithRouter(".add"),
+            "router": Startswith(".add"),
             "handler": add,
         },
         {
             "priority": 1,
-            "router": StartswithRouter("."),
+            "router": Startswith("."),
             "handler": cmd,
         },
     ]
@@ -56,12 +56,12 @@ def test_handler_block():
     handler.handlers = [
         {
             "priority": 2,
-            "router": StartswithRouter(".add"),
+            "router": Startswith(".add"),
             "handler": add,
         },
         {
             "priority": 1,
-            "router": StartswithRouter("."),
+            "router": Startswith("."),
             "handler": cmd,
         },
     ]
@@ -88,7 +88,7 @@ def test_handler_interator():
     handler.handlers = [
         {
             "priority": 1,
-            "router": StartswithRouter(".add"),
+            "router": Startswith(".add"),
             "handler": add,
         },
     ]
diff --git a/tests/test_interceptor.py b/tests/test_interceptor.py
index afb135e9..ffb03d56 100644
--- a/tests/test_interceptor.py
+++ b/tests/test_interceptor.py
@@ -1,7 +1,7 @@
 from infini.input import Input
 from infini.interceptor import Interceptor
 from infini.output import Output
-from infini.router import ContainsRouter
+from infini.router import Contains
 
 
 def test_interceptor():
@@ -15,7 +15,7 @@ def test_interceptor():
     interceptor.interceptors = [
         {
             "priority": 1,
-            "router": ContainsRouter("简律纯"),
+            "router": Contains("简律纯"),
             "handler": intercept,
         }
     ]
-- 
cgit v1.2.3-70-g09d2