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