def test_us_adapter(self): f = open('tests/data/us_congress_representative.yaml', 'r') data = yaml.load(f.read())[0] f.close() data['bioguide_id'] = data['id']['bioguide'] data['first_name'] = data['name']['first'] data['last_name'] = data['name']['last'] data['phone'] = data['terms'][0]['phone'] data['title'] = "Senator" if data['terms'][0][ 'type'] == "sen" else "Representative" data_provider = USDataProvider({}) key = data_provider.KEY_BIOGUIDE.format(**data) adapter = adapt_by_key(key) target = adapter.target(data) target['offices'] = adapter.offices(data) self.assertEqual(target['uid'], data['id']['bioguide']) self.assertEqual(target['name'], data['name']['official_full']) self.assertEqual(target['title'], 'Representative') self.assertEqual(target['number'], data['phone']) self.assertEqual(target['offices'][0]['number'], data['offices'][0]['phone'])
def setUpClass(cls): # quiet logging logging.getLogger('cache').setLevel(logging.WARNING) logging.getLogger(__name__).setLevel(logging.WARNING) cls.mock_cache = {} # mock flask-cache outside of application context cls.us_data = USDataProvider(cls.mock_cache, 'localmem') cls.us_data.load_data()
def test_usstate_adapter(self): f = open('tests/data/openstates_representative.json', 'r') data = json.loads(f.read())[0] f.close() data_provider = USDataProvider({}) key = data_provider.KEY_OPENSTATES.format(**data) adapter = adapt_by_key(key) target = adapter.target(data) target['offices'] = adapter.offices(data) self.assertEqual(target['uid'], data['id']) self.assertEqual(target['name'], data['full_name']) self.assertEqual(target['title'], 'Senator') self.assertEqual(target['number'], data['offices'][0]['phone']) self.assertEqual(target['offices'][0]['number'], data['offices'][1]['phone'])
def setUpClass(cls): cls.mock_cache = {} # mock flask-cache outside of application context cls.us_data = USDataProvider(cls.mock_cache, 'localmem') cls.us_data.load_data()