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