diff --git a/test/test_commands.py b/test/test_commands.py index b62071f..f97bed6 100644 --- a/test/test_commands.py +++ b/test/test_commands.py @@ -181,7 +181,7 @@ class CommandsAPITestCase(TestCase): self.assertEquals(shop.id, self.shop.id) - self.assertRaises(PlayerInDBError, add_owner, new_owner.name, self.shop.name, discord_uuid=DISCORD_UUID) + self.assertRaises(IsOwnerError, add_owner, new_owner.name, self.shop.name, discord_uuid=DISCORD_UUID) def test_add_resident(self): self.populate() @@ -192,8 +192,8 @@ class CommandsAPITestCase(TestCase): self.assertEquals(town.id, self.town.id) - self.assertRaises(PlayerInDBError, add_resident, new_resident.name, self.town.name, discord_uuid=DISCORD_UUID) - self.assertRaises(PlayerInDBError, add_resident, self.player.name, self.town.name, discord_uuid=DISCORD_UUID) + 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()