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"