Refactored how config data is handled and added new stress tests.

doc_update
Joey Hines 2018-08-09 21:25:55 -05:00
parent 311b94a1c8
commit 73844c0c8f
10 changed files with 186 additions and 129 deletions

View File

@ -32,5 +32,6 @@ class Admin_Commands:
else: else:
raise NoPermissionError raise NoPermissionError
def setup(bot): def setup(bot):
bot.add_cog(Admin_Commands(bot)) bot.add_cog(Admin_Commands(bot))

View File

@ -1,34 +1,50 @@
import configparser import configparser
def read_config():
config = configparser.ConfigParser()
config.read('GeoffreyConfig.ini')
if len(config.sections()) == 0: class Config:
create_config(config)
print("GeoffreyConfig.ini generated.")
quit(0)
return config def __init__(self):
try:
self.config = self.read_config()
self.engine_args = self.read_engine_arg()
self.token = self.config['Discord']['Token']
self.world_name = self.config['Minecraft']['World_Name']
self.status = self.config['Discord']['Status']
self.prefix = self.config['Discord']['Prefix']
except:
print("Invalid config file")
quit(1)
def read_config(self):
config = configparser.ConfigParser()
config.read('GeoffreyConfig.ini')
if len(config.sections()) == 0:
self.create_config(config)
print("GeoffreyConfig.ini generated.")
quit(0)
return config
def create_config(self, config):
config['Discord'] = {'Token': '', 'Status': '', 'Prefix': '', }
config['SQL'] = {'Dialect+Driver': '', 'username': '', 'password': '', 'host': '', 'port': '',
'database':'','test_args':''}
config['Minecraft'] = {'World_Name': ''}
with open('GeoffreyConfig.ini', 'w') as configfile:
config.write(configfile)
def read_engine_arg(self):
driver = self.config['SQL']['Dialect+Driver']
username = self.config['SQL']['username']
password = self.config['SQL']['password']
host = self.config['SQL']['host']
port = self.config['SQL']['port']
database_name = self.config['SQL']['database']
engine_args = '{}://{}:{}@{}:{}/{}?charset=utf8mb4&use_unicode=1'
return engine_args.format(driver, username, password, host, port, database_name)
def create_config(config): bot_config = Config()
config['Discord'] = {'Token': ''}
config['SQL'] = {'Dialect+Driver': '', 'username': '', 'password':'', 'host': '', 'port': '', 'database':'',
'test_args':''}
with open('GeoffreyConfig.ini', 'w') as configfile:
config.write(configfile)
def get_engine_arg(config):
driver = config['SQL']['Dialect+Driver']
username = config['SQL']['username']
password = config['SQL']['password']
host = config['SQL']['host']
port = config['SQL']['port']
database_name = config['SQL']['database']
engine_args = '{}://{}:{}@{}:{}/{}?charset=utf8mb4&use_unicode=1'
return engine_args.format(driver, username, password, host, port, database_name)

View File

@ -1,12 +1,9 @@
from DatabaseInterface import * from DatabaseInterface import *
from itertools import zip_longest
from discord.ext.commands import UserInputError
class Commands: class Commands:
def __init__(self, engine_args=None):
def __init__(self, db_engine_arg): self.interface = DatabaseInterface(engine_args)
self.interface = DatabaseInterface(db_engine_arg)
def get_player(self, session, discord_uuid=None, mc_uuid=None): def get_player(self, session, discord_uuid=None, mc_uuid=None):
if discord_uuid is not None: if discord_uuid is not None:

View File

@ -3,8 +3,11 @@ from DatabaseModels import *
class DatabaseInterface: class DatabaseInterface:
def __init__(self, db_engine_arg): def __init__(self, engine_args=None):
self.database = GeoffreyDatabase(db_engine_arg) if engine_args is None:
self.database = GeoffreyDatabase()
else:
self.database = GeoffreyDatabase(engine_args)
def add_base(self, session, owner, name, x_pos, z_pos, dimension=None): def add_base(self, session, owner, name, x_pos, z_pos, dimension=None):
base = Base(name, x_pos, z_pos, owner, dimension) base = Base(name, x_pos, z_pos, owner, dimension)

View File

@ -5,10 +5,10 @@ from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.sql import expression from sqlalchemy.sql import expression
import enum import enum
from difflib import SequenceMatcher from difflib import SequenceMatcher
from BotConfig import bot_config
from BotErrors import * from BotErrors import *
from MinecraftAccountInfoGrabber import * from MinecraftAccountInfoGrabber import *
world_name = 'season3'
SQL_Base = declarative_base() SQL_Base = declarative_base()
@ -25,8 +25,8 @@ def check_similarity(a, b):
class GeoffreyDatabase: class GeoffreyDatabase:
def __init__(self, engine_arg): def __init__(self, engine_args=bot_config.engine_args):
self.engine = create_engine(engine_arg, echo=True, pool_recycle=3600, pool_pre_ping=True) self.engine = create_engine(engine_args, echo=True, pool_recycle=3600, pool_pre_ping=True)
self.Session = sessionmaker(bind=self.engine) self.Session = sessionmaker(bind=self.engine)
SQL_Base.metadata.create_all(self.engine) SQL_Base.metadata.create_all(self.engine)
@ -207,7 +207,7 @@ class Location(SQL_Base):
def dynmap_link(self): def dynmap_link(self):
return '<http://24carrotcraft.com:8123/?worldname={}&mapname=surface&zoom=4&x={}&y=65&z={}>'.\ return '<http://24carrotcraft.com:8123/?worldname={}&mapname=surface&zoom=4&x={}&y=65&z={}>'.\
format(world_name, self.x, self.z) format(bot_config.world_name, self.x, self.z)
def pos_to_str(self): def pos_to_str(self):
return '(x= {}, z= {}) in the {}'.format(self.x, self.z, self.dimension.value.title()) return '(x= {}, z= {}) in the {}'.format(self.x, self.z, self.dimension.value.title())

View File

@ -12,6 +12,9 @@ class Delete_Commands:
*You must use ?register before using any of these commands!* *You must use ?register before using any of these commands!*
''' '''
def __init__(self, bot):
self.bot = bot
@commands.command(pass_context=True) @commands.command(pass_context=True)
async def delete(self, ctx, *args): async def delete(self, ctx, *args):
''' '''

View File

@ -1,11 +1,11 @@
from discord.ext import commands from discord.ext import commands
from discord import Game
from Commands import * from Commands import *
from BotErrors import * from BotErrors import *
from MinecraftAccountInfoGrabber import * from MinecraftAccountInfoGrabber import *
from BotConfig import * from BotConfig import *
import threading import asyncio
command_prefix = '?'
description = ''' description = '''
Geoffrey (pronounced JOFF-ree) started his life as an inside joke none of you will understand. Geoffrey (pronounced JOFF-ree) started his life as an inside joke none of you will understand.
At some point, she was to become an airhorn bot. Now, they know where your stuff is. At some point, she was to become an airhorn bot. Now, they know where your stuff is.
@ -20,9 +20,9 @@ If have a suggestion or if something is borked, you can PM my ding dong of a cre
bad_error_message = 'OOPSIE WOOPSIE!! Uwu We made a fucky wucky!! A wittle fucko boingo! The admins at our ' \ bad_error_message = 'OOPSIE WOOPSIE!! Uwu We made a fucky wucky!! A wittle fucko boingo! The admins at our ' \
'headquarters are working VEWY HAWD to fix this! (Error in command {}: {})' 'headquarters are working VEWY HAWD to fix this! (Error in command {}: {})'
bot = commands.Bot(command_prefix=command_prefix, description=description, case_insensitive=True)
# Bot Commands ******************************************************************' # Bot Commands ******************************************************************'
bot = commands.Bot(command_prefix=bot_config.prefix, description=description, case_insensitive=True)
@bot.event @bot.event
@ -31,6 +31,8 @@ async def on_ready():
print('Username: ' + bot.user.name) print('Username: ' + bot.user.name)
print('ID: ' + bot.user.id) print('ID: ' + bot.user.id)
await bot.change_presence(game=Game(name=bot_config.status))
@bot.event @bot.event
async def on_command_error(error, ctx): async def on_command_error(error, ctx):
@ -57,43 +59,54 @@ async def on_command_error(error, ctx):
else: else:
error_str = bad_error_message.format(ctx.invoked_with, error) error_str = bad_error_message.format(ctx.invoked_with, error)
await bot.send_message(ctx.message.channel, '{} **Error Running Command:** {}'.format(ctx.message.author.mention, bot.send_message(ctx.message.channel, '{} **Error Running Command:** {}'.format(ctx.message.author.mention,
error_str)) error_str))
def update_user_names(bot_commands):
threading.Timer(600, update_user_names, [bot_commands]).start()
session = bot_commands.interface.database.Session()
print("Updating MC usernames...")
player_list = session.query(Player).all()
for player in player_list: async def username_update():
player.name = grab_playername(player.mc_uuid) session = None
await bot.wait_until_ready()
while not bot.is_closed:
session = bot_commands.interface.database.Session()
try:
print("Updating MC usernames...")
session = bot_commands.interface.database.Session()
player_list = session.query(Player).all()
for player in player_list:
player.name = grab_playername(player.mc_uuid)
session.commit() session.commit()
session.close() await asyncio.sleep(600)
except UsernameLookupFailed:
print("Username lookup error, are Mojang's servers down?")
session.rollback()
finally:
print("Done.")
await session.close()
if session is not None:
await session.close()
# Bot Startup ****************************************************************** # Bot Startup ******************************************************************
bot_commands = Commands()
config = read_config()
TOKEN = config['Discord']['Token']
engine_arg = get_engine_arg(config)
bot_commands = Commands(engine_arg)
extensions = ['Add_Commands', 'Delete_Commands', 'Edit_Commands', 'Search_Commands', 'Admin_Commands'] extensions = ['Add_Commands', 'Delete_Commands', 'Edit_Commands', 'Search_Commands', 'Admin_Commands']
if __name__ == '__main__': if __name__ == '__main__':
bot_commands = Commands()
for extension in extensions: for extension in extensions:
try: try:
bot.load_extension(extension) bot.load_extension(extension)
except: except Exception as e:
print('Failed to load extension {}'.format(extension)) print('Failed to load extension {}, {}'.format(extension, e))
bot.loop.create_task(username_update())
bot.run(bot_config.token)
update_user_names(bot_commands)
bot.run(TOKEN)

View File

@ -1,16 +1,13 @@
from unittest import TestCase from unittest import TestCase
from BotConfig import *
from Commands import * from Commands import *
from BotErrors import * from BotErrors import *
from BotConfig import *
class TestCommands(TestCase): class TestCommands(TestCase):
def setUp(self): def setUp(self):
config = read_config()
engine_arg = config['SQL']['test_args'] self.commands = Commands(bot_config.config['SQL']['test_args'])
self.commands = Commands(engine_arg)
self.session = self.commands.interface.database.Session() self.session = self.commands.interface.database.Session()
self.commands.interface.database.clear_all(self.session) self.commands.interface.database.clear_all(self.session)
self.session.close() self.session.close()
@ -225,5 +222,3 @@ class TestCommands(TestCase):
self.assertRaises(LocationLookUpError, self.commands.delete_item, 'wood', None, self.assertRaises(LocationLookUpError, self.commands.delete_item, 'wood', None,
discord_uuid='143072699567177728') discord_uuid='143072699567177728')

View File

@ -2,18 +2,15 @@ from unittest import TestCase
from Commands import * from Commands import *
from BotErrors import * from BotErrors import *
from MinecraftAccountInfoGrabber import * from MinecraftAccountInfoGrabber import *
from BotConfig import * from BotConfig import bot_config
class TestGeoffreyDatabase(TestCase): class TestGeoffreyDatabase(TestCase):
def setUp(self): def setUp(self):
config = read_config() self.interface = DatabaseInterface(bot_config.config['SQL']['test_args'])
engine_arg = config['SQL']['test_args'] self.session = self.interface.database.Session()
self.commands = Commands(engine_arg) self.interface.database.clear_all(self.session)
self.session = self.commands.interface.database.Session()
self.commands.interface.database.clear_all(self.session)
self.owner = Player('ZeroHD', '143072699567177728') self.owner = Player('ZeroHD', '143072699567177728')
self.loc = Location('test', 1, 3, self.owner, dimension='Nether') self.loc = Location('test', 1, 3, self.owner, dimension='Nether')
self.tunnel = Tunnel(self.owner, 'Green', 105, self.loc) self.tunnel = Tunnel(self.owner, 'Green', 105, self.loc)
@ -23,57 +20,57 @@ class TestGeoffreyDatabase(TestCase):
self.session.close() self.session.close()
def add_shop(self, player): def add_shop(self, player):
shop = self.commands.interface.add_shop(self.session, player, 'test', 1, 3, "nether") shop = self.interface.add_shop(self.session, player, 'test', 1, 3, "nether")
return shop return shop
def add_player(self): def add_player(self):
player = self.commands.interface.add_player(self.session, 'ZeroHD', discord_uuid='143072699567177728') player = self.interface.add_player(self.session, 'ZeroHD', discord_uuid='143072699567177728')
return player return player
def add_loc(self, player): def add_loc(self, player):
loc = self.commands.interface.add_base(self.session, player, 'test', 0, 0) loc = self.interface.add_base(self.session, player, 'test', 0, 0)
return loc return loc
def test_add_object(self): def test_add_object(self):
self.commands.interface.database.add_object(self.session, self.loc) self.interface.database.add_object(self.session, self.loc)
self.commands.interface.database.add_object(self.session, self.owner) self.interface.database.add_object(self.session, self.owner)
self.commands.interface.database.add_object(self.session, self.tunnel) self.interface.database.add_object(self.session, self.tunnel)
uuid = grab_UUID('ZeroHD') uuid = grab_UUID('ZeroHD')
expr = Player.mc_uuid == uuid expr = Player.mc_uuid == uuid
p = self.commands.interface.database.query_by_filter(self.session, Player, expr)[0] p = self.interface.database.query_by_filter(self.session, Player, expr)[0]
expr = Location.owner == p expr = Location.owner == p
loc2 = self.commands.interface.database.query_by_filter(self.session, Location, expr)[0] loc2 = self.interface.database.query_by_filter(self.session, Location, expr)[0]
self.assertEqual(self.loc.id, loc2.id) self.assertEqual(self.loc.id, loc2.id)
def test_query_by_filter(self): def test_query_by_filter(self):
self.commands.interface.database.add_object(self.session, self.loc) self.interface.database.add_object(self.session, self.loc)
self.commands.interface.database.add_object(self.session, self.owner) self.interface.database.add_object(self.session, self.owner)
expr = (Location.owner == self.owner) expr = (Location.owner == self.owner)
loc2 = self.commands.interface.database.query_by_filter(self.session, Location, expr)[0] loc2 = self.interface.database.query_by_filter(self.session, Location, expr)[0]
self.assertEqual(loc2.id, self.loc.id) self.assertEqual(loc2.id, self.loc.id)
def test_delete_entry(self): def test_delete_entry(self):
self.commands.interface.database.add_object(self.session, self.loc) self.interface.database.add_object(self.session, self.loc)
self.commands.interface.database.add_object(self.session, self.owner) self.interface.database.add_object(self.session, self.owner)
self.session.commit() self.session.commit()
id = self.loc.owner_id self.loc.owner_id
expr = Location.owner == self.owner expr = Location.owner == self.owner
self.commands.interface.database.delete_entry(self.session, Location, expr) self.interface.database.delete_entry(self.session, Location, expr)
expr = Player.name == 'ZeroHD' expr = Player.name == 'ZeroHD'
player = self.commands.interface.database.query_by_filter(self.session, Player, expr)[0] player = self.interface.database.query_by_filter(self.session, Player, expr)[0]
self.assertEqual(player.name, 'ZeroHD') self.assertEqual(player.name, 'ZeroHD')
expr = Location.owner == player expr = Location.owner == player
loc2 = self.commands.interface.database.query_by_filter(self.session, Location, expr) loc2 = self.interface.database.query_by_filter(self.session, Location, expr)
self.assertEqual(len(loc2), 0) self.assertEqual(len(loc2), 0)
self.assertRaises(DeleteEntryError, self.commands.interface.database.delete_entry, self.session, Location, expr) self.assertRaises(DeleteEntryError, self.interface.database.delete_entry, self.session, Location, expr)
def test_add_shop(self): def test_add_shop(self):
owner = self.add_player() owner = self.add_player()
@ -81,40 +78,39 @@ class TestGeoffreyDatabase(TestCase):
self.assertEqual(type(shop), Shop) self.assertEqual(type(shop), Shop)
shop_list = self.commands.interface.find_shop_by_name(self.session, 'test') shop_list = self.interface.find_shop_by_name(self.session, 'test')
self.assertEqual(shop_list[0].dimension, shop.dimension) self.assertEqual(shop_list[0].dimension, shop.dimension)
def test_add_two_shops(self): def test_add_two_shops(self):
owner = self.add_player() owner = self.add_player()
shop1 = self.add_shop(owner) shop1 = self.add_shop(owner)
shop2 = self.commands.interface.add_shop(self.session, owner, 'no u', 1, 3) shop2 = self.interface.add_shop(self.session, owner, 'no u', 1, 3)
loc_list = self.commands.interface.find_location_by_owner(self.session, owner) loc_list = self.interface.find_location_by_owner(self.session, owner)
self.assertEqual(loc_list[1].id, shop2.id) self.assertEqual(loc_list[1].id, shop2.id)
def test_add_tunnel(self): def test_add_tunnel(self):
player = self.add_player() player = self.add_player()
tunnel1 = self.commands.interface.add_tunnel(self.session, player, 'green', 155, None) tunnel1 = self.interface.add_tunnel(self.session, player, 'green', 155, None)
tunnel2 = self.commands.interface.find_tunnel_by_owner_name(self.session, 'ZeroHD')[0] tunnel2 = self.interface.find_tunnel_by_owner_name(self.session, 'ZeroHD')[0]
self.assertEqual(tunnel1, tunnel2) self.assertEqual(tunnel1, tunnel2)
def test_add_item(self): def test_add_item(self):
owner = self.add_player() owner = self.add_player()
self.add_shop(owner) self.add_shop(owner)
self.commands.interface.add_item(self.session, owner, 'test', 'dirt', 1, 15) self.interface.add_item(self.session, owner, 'test', 'dirt', 1, 15)
shops = self.commands.interface.find_shop_selling_item(self.session, 'dirt') shops = self.interface.find_shop_selling_item(self.session, 'dirt')
self.assertGreater(len(shops), 0) self.assertGreater(len(shops), 0)
def test_find_location_by_owner(self): def test_find_location_by_owner(self):
owner = self.add_player() owner = self.add_player()
shop = self.add_shop(owner) shop = self.add_shop(owner)
loc_list = self.commands.interface.find_location_by_owner(self.session, owner) loc_list = self.interface.find_location_by_owner(self.session, owner)
self.assertEqual(loc_list[0].id, shop.id) self.assertEqual(loc_list[0].id, shop.id)
@ -122,7 +118,7 @@ class TestGeoffreyDatabase(TestCase):
owner = self.add_player() owner = self.add_player()
shop = self.add_shop(owner) shop = self.add_shop(owner)
loc_list = self.commands.interface.find_location_by_name_and_owner(self.session, owner, 'test') loc_list = self.interface.find_location_by_name_and_owner(self.session, owner, 'test')
self.assertEqual(loc_list[0].id, shop.id) self.assertEqual(loc_list[0].id, shop.id)
@ -130,9 +126,9 @@ class TestGeoffreyDatabase(TestCase):
owner = self.add_player() owner = self.add_player()
self.add_loc(owner) self.add_loc(owner)
self.commands.interface.delete_location(self.session, owner, 'test') self.interface.delete_location(self.session, owner, 'test')
loc_list = self.commands.interface.find_location_by_name(self.session, 'test') loc_list = self.interface.find_location_by_name(self.session, 'test')
self.assertEqual(len(loc_list), 0) self.assertEqual(len(loc_list), 0)
@ -142,27 +138,27 @@ class TestGeoffreyDatabase(TestCase):
dim = "o" dim = "o"
loc_list = self.commands.interface.find_location_around(self.session, 100, 100, 100, dim) loc_list = self.interface.find_location_around(self.session, 100, 100, 100, dim)
self.assertGreater(len(loc_list), 0) self.assertGreater(len(loc_list), 0)
loc_list = self.commands.interface.find_location_around(self.session, 200, 200, 100, dim) loc_list = self.interface.find_location_around(self.session, 200, 200, 100, dim)
self.assertEqual(len(loc_list), 0) self.assertEqual(len(loc_list), 0)
loc_list = self.commands.interface.find_location_around(self.session, -100, -100, 100, dim) loc_list = self.interface.find_location_around(self.session, -100, -100, 100, dim)
self.assertGreater(len(loc_list), 0) self.assertGreater(len(loc_list), 0)
loc_list = self.commands.interface.find_location_around(self.session, 100, -100, 100, dim) loc_list = self.interface.find_location_around(self.session, 100, -100, 100, dim)
self.assertGreater(len(loc_list), 0) self.assertGreater(len(loc_list), 0)
loc_list = self.commands.interface.find_location_around(self.session, -100, 100, 100, dim) loc_list = self.interface.find_location_around(self.session, -100, 100, 100, dim)
self.assertGreater(len(loc_list), 0) self.assertGreater(len(loc_list), 0)
loc_list = self.commands.interface.find_location_around(self.session, 50, -50, 100, dim) loc_list = self.interface.find_location_around(self.session, 50, -50, 100, dim)
self.assertGreater(len(loc_list), 0) self.assertGreater(len(loc_list), 0)
@ -170,7 +166,7 @@ class TestGeoffreyDatabase(TestCase):
owner = self.add_player() owner = self.add_player()
loc = self.add_loc(owner) loc = self.add_loc(owner)
loc_list = self.commands.interface.find_location_by_name(self.session, 'test') loc_list = self.interface.find_location_by_name(self.session, 'test')
self.assertEqual(loc_list[0].name, loc.name) self.assertEqual(loc_list[0].name, loc.name)
@ -178,7 +174,7 @@ class TestGeoffreyDatabase(TestCase):
owner = self.add_player() owner = self.add_player()
loc = self.add_loc(owner) loc = self.add_loc(owner)
loc_list = self.commands.interface.search_all_fields(self.session, 'ZeroHD') loc_list = self.interface.search_all_fields(self.session, 'ZeroHD')
self.assertGreater(len(loc_list), 0) self.assertGreater(len(loc_list), 0)
@ -186,26 +182,26 @@ class TestGeoffreyDatabase(TestCase):
owner = self.add_player() owner = self.add_player()
loc = self.add_loc(owner) loc = self.add_loc(owner)
loc_list = self.commands.interface.find_location_by_owner_name(self.session, 'zerohd') loc_list = self.interface.find_location_by_owner_name(self.session, 'zerohd')
self.assertEqual(loc_list[0].id, loc.id) self.assertEqual(loc_list[0].id, loc.id)
self.commands.interface.add_shop(self.session, owner, 'testshop', 1, 3, 'neThEr') self.interface.add_shop(self.session, owner, 'testshop', 1, 3, 'neThEr')
self.commands.interface.add_item(self.session, owner, 'testshop', 'dirts', 1, 15) self.interface.add_item(self.session, owner, 'testshop', 'dirts', 1, 15)
shops = self.commands.interface.find_shop_selling_item(self.session, 'diRt') shops = self.interface.find_shop_selling_item(self.session, 'diRt')
self.assertGreater(len(shops), 0) self.assertGreater(len(shops), 0)
loc_list = self.commands.interface.find_location_by_name(self.session, 'TEST') loc_list = self.interface.find_location_by_name(self.session, 'TEST')
self.assertEqual(loc_list[0].name, 'test') self.assertEqual(loc_list[0].name, 'test')
def test_big_input(self): def test_big_input(self):
owner = self.add_player() owner = self.add_player()
self.assertRaises(DatabaseValueError, self.commands.interface.add_base, self.session, owner, self.assertRaises(DatabaseValueError, self.interface.add_base, self.session, owner,
'TTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTT' 'TTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTT'
'TTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTT' 'TTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTT'
'TTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTT', 0, 0,) 'TTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTT', 0, 0,)
@ -214,18 +210,18 @@ class TestGeoffreyDatabase(TestCase):
owner = self.add_player() owner = self.add_player()
self.add_loc(owner) self.add_loc(owner)
self.assertRaises(EntryNameNotUniqueError, self.commands.interface.add_base, self.session, self.assertRaises(EntryNameNotUniqueError, self.interface.add_base, self.session,
owner, 'test', 0, 0, 0) owner, 'test', 0, 0, 0)
def test_delete_parent(self): def test_delete_parent(self):
owner = self.add_player() owner = self.add_player()
loc = self.add_shop(owner) loc = self.add_shop(owner)
self.commands.interface.add_item(self.session, owner, 'test', 'dirt', 1, 15) self.interface.add_item(self.session, owner, 'test', 'dirt', 1, 15)
self.commands.interface.delete_location(self.session, owner, 'test') self.interface.delete_location(self.session, owner, 'test')
shops = self.commands.interface.find_shop_selling_item(self.session, 'dirt') shops = self.interface.find_shop_selling_item(self.session, 'dirt')
self.assertEqual(len(shops), 0) self.assertEqual(len(shops), 0)

33
test_stress.py 100644
View File

@ -0,0 +1,33 @@
from unittest import TestCase
from Commands import *
from BotConfig import *
from MinecraftAccountInfoGrabber import *
from time import sleep
class StressTest(TestCase):
def setUp(self):
self.commands = Commands(bot_config.config['SQL']['test_args'])
def clr_db(self):
self.session = self.commands.interface.database.Session()
self.commands.interface.database.clear_all(self.session)
self.session.close()
def test_commands(self):
self.clr_db()
self.commands.register('ZeroHD', '143072699567177728')
for i in range(0, 1000):
self.commands.add_shop(0, 0, shop_name='test shop{}'.format(i), discord_uuid='143072699567177728')
self.commands.find('ZeroHD')
sleep(0.5)
def test_mc_query(self):
for i in range(0, 1000):
grab_playername('fe7e84132570458892032b69ff188bc3')
sleep(0.1)