def reset_and_setup_bots(bot_names):
"""Reset database and setup bots.
Args:
bot_names: a list of bot names to setup
"""
bots = []
DatabaseManager.reset()
register_all_modules()
for platform_name in ['dev/bb8.test.platform', 'dev/bb8.test2.platform']:
parse_platform_from_file(get_platform_filename(platform_name))
for bot_name in bot_names:
bots.append(parse_bot_from_file(get_bot_filename(bot_name)))
return bots
def setup_prerequisite(self):
register_all_modules()
self.account1 = Account(
name=u'test', email='*****@*****.**').set_passwd('12345678').add()
self.account2 = Account(
name=u'test2',
email='*****@*****.**').set_passwd('12345678').add()
DatabaseManager.commit()
self.login(self.account1)
self.create_bot()
self.create_broadcast(self.bot_ids[0])
self.login(self.account2)
self.create_bot()
self.create_broadcast(self.bot_ids[1])
# Login back as account1
self.login(self.account1)
def setup_prerequisite(self):
register_all_modules()
self.account1 = Account(
name=u'test', email='*****@*****.**').set_passwd('12345678').add()
self.account2 = Account(
name=u'test2',
email='*****@*****.**').set_passwd('12345678').add()
DatabaseManager.commit()
self.login(self.account1)
self.create_bot()
self.create_platform('dev/bb8.test.platform')
self.login(self.account2)
self.create_bot()
self.create_platform('dev/bb8.test2.platform')
# Login back as account1
self.login(self.account1)
def run(self): # pylint: disable=E0202
with DatabaseSession():
modules.register_all_modules()
def run(self): # pylint: disable=E0202
with DatabaseSession():
DatabaseManager.reset()
modules.register_all_modules()
platform_parser.build_all_platforms(include_dev=True)
bot_parser.build_all_bots()