Geoffrey-Django/geoffrey/tests/test_geoffreyDatabase.py

import os
from unittest import TestCase

from DatabaseInterface import *
from BotConfig import *

zerohd = 'BirbHD'


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 test_find_top_shops_selling_items(self):
        owner = self.add_player()
        shop = self.add_shop(owner)
        self.interface.add_item(self.session, owner, 'test', 'dirt', 1, 15)
        self.interface.add_item(self.session, owner, 'test', 'dirt', 1, 5)
        self.interface.add_loc(self.session, owner, 'test2', 1, 3, "nether", loc_type=Shop)
        self.interface.add_item(self.session, owner, 'test2', 'dirt', 1, 5)
        shop_list = self.interface.find_top_shops_selling_item(self.session, "Dirt")

        self.assertEquals(shop_list[0][0].id, shop.id)

    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_player(self):
        self.add_player()
        self.assertRaises(PlayerInDBError, self.interface.add_player, self.session, zerohd,
                          discord_uuid='143072699567177728')

    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)

        self.assertRaises(EntryNameNotUniqueError, self.interface.add_tunnel, self.session, player, "South", 155, None)

        loc = self.add_loc(player)

        tunnel3 = self.interface.add_tunnel(self.session, player, "South", 155, 'test')

        tunnel4 = self.interface.find_tunnel_by_owner_name(self.session, zerohd)[0]
        self.assertEqual(tunnel3, tunnel4)

        self.assertRaises(LocationHasTunnelError, self.interface.add_tunnel, self.session, player, "South", 155,
                          'test')

        self.assertRaises(LocationLookUpError, self.interface.add_tunnel, self.session, player, "South", 155,
                          'no u')

    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)

        self.assertRaises(LocationLookUpError, self.interface.add_item, self.session, owner, 'no u', 'dirt', 1, 15)

    def test_find_player_by_discord_uuid(self):
        p1 = self.add_player()
        p2 = self.interface.find_player_by_discord_uuid(self.session, 143072699567177728)

        self.assertEqual(p1, p2)

        self.assertRaises(PlayerNotFound, self.interface.find_player_by_discord_uuid, self.session, 143072698)

    def test_find_player_by_mc_uuid(self):
        p1 = self.add_player()
        p2 = self.interface.find_player_by_mc_uuid(self.session, 'fe7e84132570458892032b69ff188bc3')

        self.assertEqual(p1, p2)

        self.assertRaises(PlayerNotFound, self.interface.find_player_by_discord_uuid, self.session, 143072698)

    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_find_matching_location_by_name(self):
        owner = self.add_player()
        loc = self.add_loc(owner)

        loc_list = self.interface.find_location_by_name_closest_match(self.session, 'test')

        self.assertEqual(loc_list.name, loc.name)

        shop = self.interface.add_loc(self.session, owner, 'tes', 1, 3, "nether", loc_type=Shop)

        loc_list = self.interface.find_location_by_name_closest_match(self.session, shop.name)

        self.assertEqual(loc_list.name, shop.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)