227 lines
8.0 KiB
Python
227 lines
8.0 KiB
Python
import os
|
|
from unittest import TestCase
|
|
|
|
from DatabaseInterface import *
|
|
from BotConfig import *
|
|
|
|
|
|
class TestGeoffreyDatabase(TestCase):
|
|
def setUp(self):
|
|
path = os.path.dirname(os.path.abspath(__file__))
|
|
self.bot_config = get_config('{}/GeoffreyConfig.ini'.format(path))
|
|
|
|
self.interface = DatabaseInterface(self.bot_config, True)
|
|
self.session = self.interface.database.Session()
|
|
self.interface.database.clear_all(self.session)
|
|
self.owner = Player('ZeroHD', '143072699567177728')
|
|
self.loc = Location('test', 1, 3, self.owner, dimension='Nether')
|
|
self.tunnel = Tunnel(self.owner, "west", 105, self.loc)
|
|
|
|
def tearDown(self):
|
|
self.session.commit()
|
|
self.session.close()
|
|
|
|
def add_shop(self, player):
|
|
shop = self.interface.add_loc(self.session, player, 'test', 1, 3, "nether", loc_type=Shop)
|
|
return shop
|
|
|
|
def add_player(self):
|
|
player = self.interface.add_player(self.session, 'ZeroHD', discord_uuid='143072699567177728')
|
|
return player
|
|
|
|
def add_loc(self, player):
|
|
loc = self.interface.add_loc(self.session, player, 'test', 0, 0, loc_type=Base)
|
|
return loc
|
|
|
|
def test_add_object(self):
|
|
self.interface.database.add_object(self.session, self.loc)
|
|
self.interface.database.add_object(self.session, self.owner)
|
|
self.interface.database.add_object(self.session, self.tunnel)
|
|
|
|
uuid = grab_UUID('ZeroHD')
|
|
expr = Player.mc_uuid == uuid
|
|
p = self.interface.database.query_by_filter(self.session, Player, expr)[0]
|
|
|
|
expr = Location.owner == p
|
|
loc2 = self.interface.database.query_by_filter(self.session, Location, expr)[0]
|
|
|
|
self.assertEqual(self.loc.id, loc2.id)
|
|
|
|
def test_query_by_filter(self):
|
|
self.interface.database.add_object(self.session, self.loc)
|
|
self.interface.database.add_object(self.session, self.owner)
|
|
expr = (Location.owner == self.owner)
|
|
loc2 = self.interface.database.query_by_filter(self.session, Location, expr)[0]
|
|
self.assertEqual(loc2.id, self.loc.id)
|
|
|
|
def test_delete_entry(self):
|
|
self.interface.database.add_object(self.session, self.loc)
|
|
self.interface.database.add_object(self.session, self.owner)
|
|
self.session.commit()
|
|
|
|
expr = Location.owner == self.owner
|
|
self.interface.database.delete_entry(self.session, Location, expr)
|
|
|
|
expr = Player.name == 'ZeroHD'
|
|
player = self.interface.database.query_by_filter(self.session, Player, expr)[0]
|
|
self.assertEqual(player.name, 'ZeroHD')
|
|
|
|
expr = Location.owner == player
|
|
|
|
loc2 = self.interface.database.query_by_filter(self.session, Location, expr)
|
|
|
|
self.assertEqual(len(loc2), 0)
|
|
|
|
self.assertRaises(DeleteEntryError, self.interface.database.delete_entry, self.session, Location, expr)
|
|
|
|
def test_add_shop(self):
|
|
owner = self.add_player()
|
|
shop = self.add_shop(owner)
|
|
|
|
self.assertEqual(type(shop), Shop)
|
|
|
|
shop_list = self.interface.find_location_by_name(self.session, 'test', loc_type=Shop)
|
|
self.assertEqual(shop_list[0].dimension, shop.dimension)
|
|
|
|
def test_add_two_shops(self):
|
|
owner = self.add_player()
|
|
self.add_shop(owner)
|
|
shop2 = self.interface.add_loc(self.session, owner, 'no u', 1, 3, loc_type=Shop)
|
|
|
|
loc_list = self.interface.find_location_by_owner(self.session, owner)
|
|
|
|
self.assertEqual(loc_list[1].id, shop2.id)
|
|
|
|
def test_add_tunnel(self):
|
|
player = self.add_player()
|
|
tunnel1 = self.interface.add_tunnel(self.session, player, "South", 155, None)
|
|
|
|
tunnel2 = self.interface.find_tunnel_by_owner_name(self.session, 'ZeroHD')[0]
|
|
self.assertEqual(tunnel1, tunnel2)
|
|
|
|
def test_add_item(self):
|
|
owner = self.add_player()
|
|
self.add_shop(owner)
|
|
self.interface.add_item(self.session, owner, 'test', 'dirt', 1, 15)
|
|
|
|
shops = self.interface.find_shop_selling_item(self.session, 'dirt')
|
|
self.assertGreater(len(shops), 0)
|
|
|
|
def test_find_location_by_owner(self):
|
|
owner = self.add_player()
|
|
shop = self.add_shop(owner)
|
|
|
|
loc_list = self.interface.find_location_by_owner(self.session, owner)
|
|
|
|
self.assertEqual(loc_list[0].id, shop.id)
|
|
|
|
def test_find_location_by_name_and_owner(self):
|
|
owner = self.add_player()
|
|
shop = self.add_shop(owner)
|
|
|
|
loc_list = self.interface.find_location_by_name_and_owner(self.session, owner, 'test')
|
|
|
|
self.assertEqual(loc_list[0].id, shop.id)
|
|
|
|
def test_delete_base(self):
|
|
owner = self.add_player()
|
|
self.add_loc(owner)
|
|
|
|
self.interface.delete_location(self.session, owner, 'test')
|
|
|
|
loc_list = self.interface.find_location_by_name(self.session, 'test')
|
|
|
|
self.assertEqual(len(loc_list), 0)
|
|
|
|
def test_find_location_around(self):
|
|
owner = self.add_player()
|
|
self.add_loc(owner)
|
|
|
|
dim = "o"
|
|
|
|
loc_list = self.interface.find_location_around(self.session, 100, 100, 100, dim)
|
|
|
|
self.assertGreater(len(loc_list), 0)
|
|
|
|
loc_list = self.interface.find_location_around(self.session, 200, 200, 100, dim)
|
|
|
|
self.assertEqual(len(loc_list), 0)
|
|
|
|
loc_list = self.interface.find_location_around(self.session, -100, -100, 100, dim)
|
|
|
|
self.assertGreater(len(loc_list), 0)
|
|
|
|
loc_list = self.interface.find_location_around(self.session, 100, -100, 100, dim)
|
|
|
|
self.assertGreater(len(loc_list), 0)
|
|
|
|
loc_list = self.interface.find_location_around(self.session, -100, 100, 100, dim)
|
|
|
|
self.assertGreater(len(loc_list), 0)
|
|
|
|
loc_list = self.interface.find_location_around(self.session, 50, -50, 100, dim)
|
|
|
|
self.assertGreater(len(loc_list), 0)
|
|
|
|
def test_find_location_by_name(self):
|
|
owner = self.add_player()
|
|
loc = self.add_loc(owner)
|
|
|
|
loc_list = self.interface.find_location_by_name(self.session, 'test')
|
|
|
|
self.assertEqual(loc_list[0].name, loc.name)
|
|
|
|
def test_search_all(self):
|
|
owner = self.add_player()
|
|
self.add_loc(owner)
|
|
|
|
loc_list = self.interface.search_all_fields(self.session, 'ZeroHD')
|
|
|
|
self.assertGreater(len(loc_list), 0)
|
|
|
|
def test_wrong_case(self):
|
|
owner = self.add_player()
|
|
loc = self.add_loc(owner)
|
|
|
|
loc_list = self.interface.find_location_by_owner_name(self.session, 'zerohd')
|
|
|
|
self.assertEqual(loc_list[0].id, loc.id)
|
|
|
|
self.interface.add_loc(self.session, owner, 'testshop', 1, 3, 'neThEr', loc_type=Shop)
|
|
|
|
self.interface.add_item(self.session, owner, 'testshop', 'dirts', 1, 15)
|
|
|
|
shops = self.interface.find_shop_selling_item(self.session, 'diRt')
|
|
|
|
self.assertGreater(len(shops), 0)
|
|
|
|
loc_list = self.interface.find_location_by_name(self.session, 'TEST')
|
|
|
|
self.assertEqual(loc_list[0].name, 'test')
|
|
|
|
def test_big_input(self):
|
|
owner = self.add_player()
|
|
|
|
self.assertRaises(DatabaseValueError, self.interface.add_loc, self.session, owner,
|
|
'TTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTT'
|
|
'TTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTT'
|
|
'TTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTT', 0, 0, Shop)
|
|
|
|
def test_duplicate_name(self):
|
|
owner = self.add_player()
|
|
self.add_loc(owner)
|
|
|
|
self.assertRaises(EntryNameNotUniqueError, self.interface.add_loc, self.session,
|
|
owner, 'test', 0, 0, Shop)
|
|
|
|
def test_delete_parent(self):
|
|
owner = self.add_player()
|
|
self.add_shop(owner)
|
|
|
|
self.interface.add_item(self.session, owner, 'test', 'dirt', 1, 15)
|
|
|
|
self.interface.delete_location(self.session, owner, 'test')
|
|
|
|
shops = self.interface.find_shop_selling_item(self.session, 'dirt')
|
|
self.assertEqual(len(shops), 0)
|