class StyleDB: def __init__(self): self.cache = StyleCache('src/abv/inventory_api/beer_styles.csv') self.brew_db = BreweryDBQueries() def get_style(self, beer_name): beer_name = beer_name.title() if self.cache.look_up(beer_name) is not None: return self.cache.cache_dict[beer_name] try: style = self.brew_db.get_beer_style(beer_name) self.cache.add(beer_name, style) return style # pylint: disable=broad-except except Exception: return 'Unknown'
def test_add_beer(): temp_file = tempfile.NamedTemporaryFile(dir='tests/sample_csv_files', prefix='b') test_cache = StyleCache(temp_file.name) test_cache.add("Rude Elf Reserve", "Belgian") assert test_cache.look_up("Rude Elf Reserve") == "Belgian"