phenny/modules/test/test_weather.py

48 lines
1.3 KiB
Python
Raw Normal View History

2012-06-02 06:08:29 -04:00
"""
test_weather.py - tests for the weather module
author: mutantmonkey <mutantmonkey@mutantmonkey.in>
"""
import re
import unittest
from mock import MagicMock, Mock, patch
from modules.weather import location, local, code, f_weather
class TestWeather(unittest.TestCase):
def setUp(self):
self.phenny = MagicMock()
def test_location(self):
name, countryName, lat, lng = location('24060')
self.assertEqual(name, "Blacksburg")
self.assertEqual(countryName, "United States")
self.assertEqual(lat, 37.2295733)
self.assertEqual(lng, -80.4139393)
def test_code(self):
icao = code(self.phenny, '20164')
self.assertEqual(icao, 'KIAD')
2012-06-29 23:52:07 -04:00
def test_airport(self):
input = Mock(group=lambda x: 'KIAD')
2012-06-02 06:08:29 -04:00
f_weather(self.phenny, input)
assert self.phenny.say.called is True
2012-06-29 23:52:07 -04:00
def test_place(self):
input = Mock(group=lambda x: 'Blacksburg')
2012-06-29 23:52:07 -04:00
f_weather(self.phenny, input)
assert self.phenny.say.called is True
2012-06-29 23:52:07 -04:00
def test_notfound(self):
input = Mock(group=lambda x: 'Hell')
2012-06-29 23:52:07 -04:00
f_weather(self.phenny, input)
self.phenny.say.called_once_with('#phenny',
2012-06-29 23:52:07 -04:00
"No NOAA data available for that location.")