phenny/modules/test/test_weather.py

72 lines
2.5 KiB
Python

"""
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_locations(self):
def check_places(*args):
def validate(actual_name, actual_lat, actual_lon):
names = [n.strip() for n in actual_name.split(',')]
for arg in args:
self.assertIn(arg, names)
return validate
locations = [
('92121', check_places("San Diego", "California")),
('94110', check_places("SF", "California")),
('94041', check_places("Mountain View", "California")),
('27959', check_places("Dare County", "North Carolina")),
('48067', check_places("Royal Oak", "Michigan")),
('23606', check_places("Newport News", "Virginia")),
('23113', check_places("Chesterfield County", "Virginia")),
('27517', check_places("Chapel Hill", "North Carolina")),
('15213', check_places("Allegheny County", "Pennsylvania")),
('90210', check_places("Los Angeles County", "California")),
('33109', check_places("Miami-Dade County", "Florida")),
('80201', check_places("Denver", "Colorado")),
("Berlin", check_places("Berlin", "Deutschland")),
("Paris", check_places("Paris", "France")),
("Vilnius", check_places("Vilnius", "Lietuva")),
('Blacksburg, VA', check_places("Blacksburg", "Virginia")),
('Granger, IN', check_places("Granger", "Indiana")),
]
for loc, validator in locations:
names, lat, lon = location(loc)
validator(names, lat, lon)
def test_code_94110(self):
icao = code(self.phenny, '94110')
self.assertEqual(icao, 'KSFO')
def test_airport(self):
input = Mock(group=lambda x: 'KIAD')
f_weather(self.phenny, input)
assert self.phenny.say.called is True
def test_place(self):
input = Mock(group=lambda x: 'Blacksburg')
f_weather(self.phenny, input)
assert self.phenny.say.called is True
def test_notfound(self):
input = Mock(group=lambda x: 'Hell')
f_weather(self.phenny, input)
self.phenny.say.called_once_with('#phenny',
"No NOAA data available for that location.")