Geoffrey-Django/GeoffreyApp/test/test_commands.py

from django.test import TestCase
from GeoffreyApp.GeoffreyApp.models import *
from time import sleep

# Create your tests here.

MC_UUID = "fe7e84132570458892032b69ff188bc3"
DISCORD_UUID = "143072699567177728"
USERNAME = "ZeroHD"


class CommandsAPITestCase(TestCase):

    def setUp(self):
        self.player = Player.objects.create(name=USERNAME, mc_uuid=MC_UUID, discord_uuid=DISCORD_UUID)

    def tearDown(self):
        Player.objects.all().delete()
        Base.objects.all().delete()
        Location.objects.all().delete()
        Shop.objects.all().delete()
        ItemListing.objects.all().delete()
        Tunnel.objects.all().delete()
        PublicFarm.objects.all().delete()

    def populate(self):
        self.base = Base.objects.create(name="test", x_coord=0, z_coord=0, dimension="O")
        self.base.owner.add(self.player)
        self.shop = Shop.objects.create(name="test shop", x_coord=500, z_coord=500,
                                   dimension="O")

        self.town = Town.objects.create(name="test town", x_coord=-500, z_coord=-500,
                                        dimension="O")

        self.town.owner.add(self.player)
        self.shop.owner.add(self.player)
        self.item = ItemListing.objects.create(shop=self.shop, price=1, amount=5, item_name="sed")

        self.tunnel = Tunnel.objects.create(tunnel_number="42", tunnel_direction="S", location=self.base)

        self.farm = PublicFarm.objects.create(name="test farm", x_coord=0, z_coord=0, dimension="O")
        self.farm.owner.add(self.player)

    def test_register(self):
        register(player_name="DrVakky", discord_uuid="229423434256351233")

        count = Player.objects.filter(mc_uuid__iexact="7afbf6632bf049ef915f22e81b298d17").count()

        self.assertEqual(count, 1)

        self.assertRaises(PlayerInDBError, register, player_name="DrVakky", discord_uuid="229423434256351233")

    def test_add_base(self):
        add_base(x_pos=0, z_pos=0, name=None, discord_uuid=DISCORD_UUID)

        base = Base.objects.filter(name__icontains=USERNAME).all().first()

        self.assertEqual(base.owner.all()[0].name, "ZeroHD")

        self.assertRaises(LocationLookUpError, add_base, x_pos=0, z_pos=0, name=None, discord_uuid=DISCORD_UUID)

    def test_add_shop(self):
        add_shop(x_pos=0, z_pos=0, name=None, discord_uuid=DISCORD_UUID)

        shop = Shop.objects.filter(name__icontains=USERNAME).all().first()

        self.assertEqual(shop.owner.all()[0].name, "ZeroHD")

    def test_add_town(self):
        add_town(x_pos=0, z_pos=0, name=None, discord_uuid=DISCORD_UUID)

        town = Town.objects.filter(name__icontains=USERNAME).all().first()

        self.assertEqual(town.owner.all()[0].name, "ZeroHD")

    def test_add_tunnel(self):
        base = Base.objects.create(name="Test", x_coord=0, z_coord=0)

        base.owner.add(self.player)
        add_tunnel(tunnel_direction="N", tunnel_number=500, discord_uuid=DISCORD_UUID)

        qbase = Base.objects.filter(tunnel_location__tunnel_direction="N").all().first()

        self.assertEqual(base.id, qbase.id)

    def test_add_item(self):
        shop = Shop.objects.create(name="Test", x_coord=0, z_coord=0)
        shop.owner.add(self.player)
        add_item(item_name="sed", quantity=5, diamond_price=5, discord_uuid=DISCORD_UUID)

        item = ItemListing.objects.filter(shop_id=shop.id).all().first()

        self.assertEqual(item.item_name, "sed")

    def test_edit_post(self):
        shop = Shop.objects.create(name="Test", x_coord=0, z_coord=0)
        shop.owner.add(self.player)

        edit_pos(x=500, z=500, loc_name="Test", discord_uuid=DISCORD_UUID)

        qshop = Shop.objects.filter(name__exact="Test").all().first()

        self.assertEqual(qshop.x_coord, 500)

    def test_find_location(self):
        self.populate()
        locations = find_location(search="Test")

        count = len(locations)

        self.assertGreater(count, 1)

    def test_find_around(self):
        self.populate()
        locations = find_around(x_pos=0, z_pos=0, radius=50)

        self.assertEqual(len(locations), 2)

    def test_selling(self):
        self.populate()

        items = selling(item_name="sED")

        self.assertEqual(len(items), 1)

    def test_info(self):
        self.populate()
        location = info(location_name="test")

        self.assertEqual(location["name"], "test")

    def test_tunnel(self):
        self.populate()

        tunnels = tunnel(player_name=USERNAME)

        self.assertEqual(len(tunnels), 1)

    def test_me(self):
        self.populate()

        locations = me(discord_uuid=DISCORD_UUID)

        self.assertGreater(len(locations), 1)

    def test_delete(self):
        self.populate()

        count = Location.objects.filter(name__icontains="test").all().count()

        delete(name="test", discord_uuid=DISCORD_UUID)

        locations = Location.objects.filter(name__icontains="test").all()

        self.assertEqual(len(locations), count - 1)

    def test_delete_item(self):
        self.populate()

        delete_item(item="sed", shop_name="test shop", discord_uuid=DISCORD_UUID)

        items = ItemListing.objects.filter(item_name__iexact="sed").all()

        self.assertEqual(len(items), 0)

    def test_restock(self):
        self.populate()
        sleep(1)
        item = ItemListing.objects.create(shop=self.shop, price=1, amount=5, item_name="sed2")

        restock(item_name="sed", shop_name=self.shop.name, discord_uuid=DISCORD_UUID)

        new_item = ItemListing.objects.get(item_name="sed2")
        old_item = ItemListing.objects.get(item_name="sed")

        self.assertGreater(old_item.date_restocked, new_item.date_restocked)

    def test_add_owner(self):
        self.populate()
        new_owner = Player.objects.create(name="Vakky", mc_uuid="5", discord_uuid="5")
        add_owner(new_owner.name, self.shop.name, discord_uuid=DISCORD_UUID)

        shop = Shop.objects.get(owner__name__icontains="Vakky")

        self.assertEquals(shop.id, self.shop.id)

        self.assertRaises(IsOwnerError, add_owner, new_owner.name, self.shop.name, discord_uuid=DISCORD_UUID)

    def test_add_resident(self):
        self.populate()
        new_resident = Player.objects.create(name="Vakky", mc_uuid="5", discord_uuid="5")
        add_resident(new_resident.name, self.town.name, discord_uuid=DISCORD_UUID)

        town = Town.objects.get(residents__name__icontains="Vakky")

        self.assertEquals(town.id, self.town.id)

        self.assertRaises(IsResidentError, add_resident, new_resident.name, self.town.name, discord_uuid=DISCORD_UUID)
        self.assertRaises(IsResidentError, add_resident, self.player.name, self.town.name, discord_uuid=DISCORD_UUID)

    def test_remove_resident(self):
        self.populate()
        new_resident = Player.objects.create(name="Vakky", mc_uuid="5", discord_uuid="5")
        add_resident(new_resident.name, self.town.name, discord_uuid=DISCORD_UUID)

        Town.objects.get(residents__name__icontains="Vakky")

        remove_resident(new_resident.name, self.town.name, discord_uuid=DISCORD_UUID)

        self.assertRaises(Town.DoesNotExist, Town.objects.get, residents__name__icontains="Vakky")

        self.assertRaises(ResidentNotFoundError, remove_resident, new_resident.name, self.town.name,
                          discord_uuid=DISCORD_UUID)

    def test_add_resource(self):
        self.populate()

        resource = add_resource(resource_name="sed", farm_name="test farm", discord_uuid=DISCORD_UUID)

        farm = PublicFarm.objects.filter(resource__resource_name__icontains="sed").first()

        self.assertEqual(resource["farm_id"], farm.id)

    def test_remove_resource(self):
        self.populate()

        resource = add_resource(resource_name="sed", farm_name="test farm", discord_uuid=DISCORD_UUID)

        delete_resource(resource_name="sed", farm_name="test farm", discord_uuid=DISCORD_UUID)

        farm = PublicFarm.objects.filter(resource__resource_name__icontains="sed").all()

        self.assertEqual(len(farm), 0)

    def test_find_farm(self):
        self.populate()
        add_resource(resource_name="sed", farm_name="test farm", discord_uuid=DISCORD_UUID)

        list = find_farm(resource_name="sed")

        for farm in list:
            if farm["name"] == "test farm":
                return

        self.fail("Farm not found")