瀏覽代碼

fix http(s) errors in test_all_text, test_answer_links

eoglethorpe 7 年之前
父節點
當前提交
fad1f1a9bf
共有 1 個文件被更改,包括 3 次插入2 次删除
  1. 3 2
      test_howdoi.py

+ 3 - 2
test_howdoi.py

@@ -3,6 +3,7 @@
 """Tests for Howdoi."""
 import os
 import unittest
+import re
 
 from howdoi import howdoi
 
@@ -52,7 +53,7 @@ class HowdoiTestCase(unittest.TestCase):
 
     def test_answer_links(self):
         for query in self.queries:
-            self.assertTrue('http://' in self.call_howdoi(query + ' -l'))
+            self.assertIsNotNone(re.match('http.?://.*', self.call_howdoi(query + ' -l'), re.DOTALL))
 
     def test_position(self):
         query = self.queries[0]
@@ -65,7 +66,7 @@ class HowdoiTestCase(unittest.TestCase):
         first_answer = self.call_howdoi(query)
         second_answer = self.call_howdoi(query + ' -a')
         self.assertNotEqual(first_answer, second_answer)
-        self.assertTrue("Answer from http://stackoverflow.com" in second_answer)
+        self.assertIsNotNone(re.match('.*Answer from http.?://stackoverflow.com.*', second_answer, re.DOTALL))
 
     def test_multiple_answers(self):
         query = self.queries[0]