Add powerball scrape and insert test
This commit is contained in:
1 parent
9992abb8a2
commit
f41e88ac69
1 file changed
+20
@@ -2,6 +2,8 @@ from datetime import datetime
|
|||||||
from sqlite3 import IntegrityError
|
from sqlite3 import IntegrityError
|
||||||
|
|
||||||
from conftest import PowerballDraw
|
from conftest import PowerballDraw
|
||||||
|
from util.drawing import get_powerball_object
|
||||||
|
from util.scrape import scrape_game_history
|
||||||
|
|
||||||
|
|
||||||
def test_create_powerball_draw(db_session) -> None:
|
def test_create_powerball_draw(db_session) -> None:
|
||||||
@@ -66,3 +68,21 @@ def test_powerball_draw_eq_repr_and_str(db_session) -> None:
|
|||||||
assert repr(record) == repr(new_record)
|
assert repr(record) == repr(new_record)
|
||||||
|
|
||||||
|
|
||||||
|
def test_powerball_scrape_and_insert(db_session) -> None:
|
||||||
|
"""test scraping game data and inserting it into the database"""
|
||||||
|
# scrape the current year history for powerball
|
||||||
|
raw_data = scrape_game_history(game='powerball')
|
||||||
|
|
||||||
|
# loop over the records returned
|
||||||
|
for key, value in raw_data.items():
|
||||||
|
# convert each record to a PowerballDraw object
|
||||||
|
record = get_powerball_object(draw_date=key, draw_result=value)
|
||||||
|
# add the record to the database
|
||||||
|
db_session.add(record)
|
||||||
|
|
||||||
|
# commit the changes
|
||||||
|
db_session.commit()
|
||||||
|
|
||||||
|
# get all the records from the table
|
||||||
|
records = db_session.query(PowerballDraw).all()
|
||||||
|
assert len(records) == len(raw_data)
|
||||||
Reference in new issue
Block a user