diff --git a/src/currency_quote/domain/entities/currency.py b/src/currency_quote/domain/entities/currency.py index 5c6ca5c..a0dc244 100644 --- a/src/currency_quote/domain/entities/currency.py +++ b/src/currency_quote/domain/entities/currency.py @@ -5,7 +5,7 @@ def __init__(self, currency_list: list): self.currency_list = currency_list def get_currency_list(self) -> list: - if self.currency_list is None: + if len(self.currency_list) == 0: raise ValueError("Currency list is empty") return self.currency_list diff --git a/src/currency_quote/domain/entities/test_currency.py b/src/currency_quote/domain/entities/test_currency.py new file mode 100644 index 0000000..131351c --- /dev/null +++ b/src/currency_quote/domain/entities/test_currency.py @@ -0,0 +1,8 @@ +import pytest +from currency_quote.domain.entities.currency import CurrencyQuote + + +def test_currency(): + client = CurrencyQuote(currency_list=[]) + with pytest.raises(ValueError): + client.get_currency_list()