Skip to content

Commit 05dae25

Browse files
committed
Fix tests
1 parent badbda5 commit 05dae25

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

werobot/robot.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,7 @@ def _check_content(message):
159159

160160
def wraps(f):
161161
if content_is_list:
162-
for x in target_content:
162+
for x in args:
163163
self.filter(x)(f)
164164
return f
165165
argc = len(inspect.getargspec(f).args)

werobot/tests/test_robot.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
from nose.tools import raises
88
from werobot import WeRoBot
9-
from werobot.utils import generate_token
9+
from werobot.utils import generate_token, to_text
1010

1111

1212
def test_signature_checker():
@@ -108,7 +108,7 @@ def _():
108108

109109
assert len(robot._handlers["text"]) == 1
110110

111-
@robot.filter(re.compile(".*?呵呵.*?"))
111+
@robot.filter(re.compile(to_text(".*?呵呵.*?")))
112112
def _():
113113
return "哼"
114114

@@ -142,12 +142,12 @@ def _make_xml(content):
142142
def _():
143143
return "就不帮"
144144

145+
assert len(robot._handlers["text"]) == 3
146+
145147
@robot.text
146148
def _():
147149
return "哦"
148150

149-
assert len(robot._handlers["text"]) == 3
150-
151151
tester = werobot.testing.WeTest(robot)
152152

153153
assert tester.send_xml(_make_xml("啊")) == "哦"

0 commit comments

Comments
 (0)