tfw: tests and retry on first failure

modules/test/test_tfw.py

@@ -0,0 +1,47 @@
+"""
+test_tfw.py - tests for the fucking weather module
+author: mutantmonkey <mutantmonkey@mutantmonkey.in>
+"""
+
+# add current working directory to path
+import sys
+sys.path.append('.')
+
+import re
+import unittest
+from mock import MagicMock, Mock
+from modules.tfw import tfw
+
+class TestTfw(unittest.TestCase):
+    def setUp(self):
+        self.phenny = MagicMock()
+
+    #def test_badloc(self):
+    #    input = Mock(group=lambda x: 'tu3jgoajgoahghqog')
+    #    tfw(self.phenny, input)
+    #
+    #    self.phenny.say.assert_called_once_with("UNKNOWN FUCKING LOCATION. Try another?")
+
+    def test_celsius(self):
+        input = Mock(group=lambda x: '24060')
+        tfw(self.phenny, input, celsius=True)
+
+        out = self.phenny.say.call_args[0][0]
+        m = re.match('^\d+°C‽ .* \- .*$', out, flags=re.UNICODE)
+        self.assertTrue(m)
+
+    def test_fahrenheit(self):
+        input = Mock(group=lambda x: '24060')
+        tfw(self.phenny, input, fahrenheit=True)
+
+        out = self.phenny.say.call_args[0][0]
+        m = re.match('^\d+°F‽ .* \- .*$', out, flags=re.UNICODE)
+        self.assertTrue(m)
+
+    def test_mev(self):
+        input = Mock(group=lambda x: '24060')
+        tfw(self.phenny, input)
+
+        out = self.phenny.say.call_args[0][0]
+        m = re.match('^[\d\.]+ meV‽ .* \- .*$', out, flags=re.UNICODE)
+        self.assertTrue(m)

modules/tfw.py

@@ -24,6 +24,10 @@ def tfw(phenny, input, fahrenheit=False, celsius=False):
     else:
         celsius_param = "&CELSIUS=yes"
 
+    try:
+        req = web.get("http://thefuckingweather.com/?where={0}{1}".format(urlquote(where), celsius_param))
+    except (HTTPError, IOError):
+        # the fucking weather is fucking unstable, try again
         try:
             req = web.get("http://thefuckingweather.com/?where={0}{1}".format(urlquote(where), celsius_param))
         except (HTTPError, IOError):