def test_on_ready(gid, obj_guild):
# New guild test
DataConnector.run_query("DELETE FROM {}.guilds WHERE guild_id='{}'".format(
SCHEMA_NAME, gid))
df = DataConnector.read_data(
"SELECT * FROM {}.guilds WHERE guild_id='{}'".format(SCHEMA_NAME, gid))
assert df.shape[0] == 0
lst_guilds = []
lst_guilds.append(obj_guild)
DataProcessor._on_ready(lst_guilds)
df = DataConnector.read_data(
"SELECT * FROM {}.guilds WHERE guild_id='{}'".format(SCHEMA_NAME, gid))
assert df.shape[0] == 1
# Same guild list test
DataProcessor._on_ready(lst_guilds)
df = DataConnector.read_data(
"SELECT * FROM {}.guilds WHERE guild_id='{}'".format(SCHEMA_NAME, gid))
assert df.shape[0] == 1
# Additional guild test
new_gid = '54321'
DataConnector.run_query("DELETE FROM {}.guilds WHERE guild_id='{}'".format(
SCHEMA_NAME, new_gid))
new_obj_guild = guilds(new_gid)
lst_guilds.append(new_obj_guild)
DataProcessor._on_ready(lst_guilds)
df = DataConnector.read_data("""SELECT *
FROM {}.guilds
WHERE guild_id='{}' or guild_id='{}'
""".format(SCHEMA_NAME, gid, new_gid))
assert df.shape[0] == 2
# Removed guilds test
empty_lst_guilds = []
DataProcessor._on_ready(empty_lst_guilds)
df = DataConnector.read_data("""SELECT *
FROM {}.guilds
WHERE guild_id='{}' or guild_id='{}'
""".format(SCHEMA_NAME, gid, new_gid))
assert df.shape[0] == 0
# One last sanity check
lst_guilds.remove(new_obj_guild)
DataProcessor._on_ready(lst_guilds)
df = DataConnector.read_data("""SELECT *
FROM {}.guilds
WHERE guild_id='{}' or guild_id='{}'
""".format(SCHEMA_NAME, gid, new_gid))
assert df.shape[0] == 1
async def on_ready():
'''
Initial discord call when launching bot.
'''
DataProcessor._on_ready(client.guilds)