from django.test import TestCase from GeoffreyApp.api.commands import * from 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")