summaryrefslogtreecommitdiff
blob: 33811abdc7acfd379fe0233ce300ff03410e3897 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
diff --git a/test_login.py b/test_login.py
index 3f110e0..0c060f5 100644
--- a/test_login.py
+++ b/test_login.py
@@ -39,6 +39,7 @@ from flask.ext.login import (LoginManager, UserMixin, AnonymousUserMixin,
 if str is not bytes:
     unicode = str
 
+werkzeug_version = tuple(int(i) for i in werkzeug_version.split('.'))
 
 @contextmanager
 def listen_to(signal):
@@ -1073,14 +1074,14 @@ class LoginTestCase(unittest.TestCase):
     #
     # Misc
     #
-    @unittest.skipIf(werkzeug_version.startswith("0.9"),
+    @unittest.skipIf(werkzeug_version >= (0, 9),
                      "wait for upstream implementing RFC 5987")
     def test_chinese_user_agent(self):
         with self.app.test_client() as c:
             result = c.get('/', headers=[('User-Agent', u'中文')])
             self.assertEqual(u'Welcome!', result.data.decode('utf-8'))
 
-    @unittest.skipIf(werkzeug_version.startswith("0.9"),
+    @unittest.skipIf(werkzeug_version >= (0, 9),
                      "wait for upstream implementing RFC 5987")
     def test_russian_cp1251_user_agent(self):
         with self.app.test_client() as c: