async def test_retrieve_beatmapset_events_cache(scraper_test_database):
beatmapset = Beatmapset(3,
"artist",
"title",
User(4, "creator"), ["osu"],
genre="g",
language="l")
nom_event = Event("nominate",
from_string("2020-01-01 00:00:00"),
beatmapset,
user=User(1, "someone"))
qual_event = Event("qualify",
from_string("2020-01-01 05:00:00"),
beatmapset,
user=User(2, "sometwo"))
scraper_test_database.insert_event(nom_event)
scraper_test_database.insert_event(qual_event)
await scraper_test_database.retrieve_beatmapset_events(beatmapset)
assert db_module.beatmapset_event_cache[SCRAPER_TEST_DB_NAME][3] == [
qual_event, nom_event
]
db_module.clear_cache(SCRAPER_TEST_DB_NAME)
assert not db_module.beatmapset_event_cache[SCRAPER_TEST_DB_NAME]
def bot_test_database():
database = Database(BOT_TEST_DB_NAME)
database.clear_table_data("subscriptions")
database.clear_table_data("prefixes")
database.clear_table_data("permissions")
db_module.clear_cache(BOT_TEST_DB_NAME)
return database
def test_database():
database = Database(SCRAPER_TEST_DB_NAME)
database.clear_table_data("events")
db_module.clear_cache(SCRAPER_TEST_DB_NAME)
return database
async def on_events(self, events):
# This is called before any on_event for each batch.
database.clear_cache(SCRAPER_DB_NAME)
def scraper_test_database():
database = Database(SCRAPER_TEST_DB_NAME)
# Reset database to state before any tests ran.
database.clear_table_data("events")
db_module.clear_cache(SCRAPER_TEST_DB_NAME)
return database