diff --git a/tests.py b/tests.py index 1b3e6ee..b00b389 100644 --- a/tests.py +++ b/tests.py @@ -22,7 +22,7 @@ def test_get_booking_page(self): @classmethod def test_prep_data(self): """ Verify prep_data """ - i = b.prep_data(rooms, country, start_date, end_date) + i: list = b.prep_data(rooms, country, start_date, end_date) assert i @classmethod @@ -43,6 +43,23 @@ def test_default_start_end_date(self): except Exception as err: raise ValueError(err) + @classmethod + def test_prep_data(self): + """ Test Process Hotels Method """ + try: + session = requests.Session() + out = b.prep_data( + rooms, + country, + 'json', + b.default_start_date(), + b.default_end_date() + ) + assert isinstance(out, list) + assert len(out) > 0 + except Exception as err: + raise Exception(err) + @classmethod def test_get_excel(self): """ Verify excel output """