def test_venue_lookupID(data_teams): # Setup log = Log('test.log') v = Venue() v.connectDB() # This should raise a format error with pytest.raises(RuntimeError) as excinfo: needle = 'My favorite venue' v.lookupID(needle, log) assert 'lookupID requires a dictionary' in str(excinfo.value) # This should raise a missing-fields error with pytest.raises(RuntimeError) as excinfo: needle = { 'Venue': 'Columbus Crew Stadium' } v.lookupID(needle, log) assert 'Submitted data is missing the following fields' in str(excinfo.value) # This should bring back one record needle = { 'VenueName': 'MAPFRE Stadium' } assert len(v.lookupID(needle, log)) >= 1
def test_venue_disconnect(): v = Venue() v.connectDB() assert hasattr(v, 'db') v.disconnectDB() assert hasattr(v, 'db') is False