MazeObjectsTests.hs

-- | Items and Mazes hunit tests
-- (c) JP Moresmau 2009
module MoresmauJP.Rpg.MazeObjectsTests where

import qualified Data.Map as M

import MoresmauJP.Maze1.Maze
import MoresmauJP.Rpg.Character
import MoresmauJP.Rpg.CharacterTests
import MoresmauJP.Rpg.Inventory
import MoresmauJP.Rpg.MazeObjects
import MoresmauJP.Rpg.NPC
import MoresmauJP.Rpg.NPCTests

import MoresmauJP.Util.Random

import System.Random

import Test.HUnit

mazeObjectsTests=TestList [testScrollFill,testKillNPC,testKillNPC2HandBag,testKillNPC2HandHand]

testScrollFill= TestLabel "Test filling scroll" (TestCase (do
	let sc=Scroll "" "" 5
	sc2<-evalRandT (generateScroll sc) (mkTestWrapper [2])
	assertEqual "not proper scroll" (Scroll "Scroll of Feel Better" "Feel Better" 5) sc2
	))
	
testKillNPC=TestLabel "Test kill NPC" (TestCase (do
	std<-getStdGen
	gw0<-(evalRandT (generateGameWorld (2,2)) (ProductionRandom std)) 
	
	let 
		gw=gw0{position=(0,0)}
		jp=createTestChar "JP"
		item=Weapon "Sword" 2 6 1 10
		v1=(createTestNPC "Victim")
		vc1=(npcCharacter v1){inventory=makeFullInventory [(Bag 1,item)] 10 100}
		victim=v1{npcCharacter=vc1}
		mz=MazeObjects M.empty (M.fromList [((0,0),victim)])
		(mz2,jp2,g)=killNPC gw mz jp
		dropped=M.lookup (0,0) (items mz2)
	assertEqual "Gold in JP's pocket is not 100" 100 (getGold $ inventory jp2)
	assertEqual "Gold message is not 100" 100 g
	assertEqual "item not dropped" (Just [item]) dropped
	))
	
testKillNPC2HandBag=TestLabel "Test kill NPC 2 Handed weapon in Bag" (TestCase (do
	std<-getStdGen
	gw0<-(evalRandT (generateGameWorld (2,2)) (ProductionRandom std)) 
	
	let 
		gw=gw0{position=(0,0)}
		jp=createTestChar "JP"
		item=Weapon "2 Handed Sword" 4 12 2 20
		v1=(createTestNPC "Victim")
		vc1=(npcCharacter v1){inventory=makeFullInventory [(Bag 1,item)] 10 100}
		victim=v1{npcCharacter=vc1}
		mz=MazeObjects M.empty (M.fromList [((0,0),victim)])
		(mz2,jp2,g)=killNPC gw mz jp
		dropped=M.lookup (0,0) (items mz2)
	assertEqual "Gold in JP's pocket is not 100" 100 (getGold $ inventory jp2)
	assertEqual "Gold message is not 100" 100 g
	assertEqual "item not dropped" (Just [item]) dropped
	))	
	
testKillNPC2HandHand=TestLabel "Test kill NPC 2 Handed weapon in Hand" (TestCase (do
	std<-getStdGen
	gw0<-(evalRandT (generateGameWorld (2,2)) (ProductionRandom std)) 
	
	let 
		gw=gw0{position=(0,0)}
		jp=createTestChar "JP"
		item=Weapon "2 Handed Sword" 4 12 2 20
		v1=(createTestNPC "Victim")
		inv1=makeFullInventory [] 10 100
		Right (inv2,_)=MoresmauJP.Rpg.Inventory.takeItem inv1 item RightHand
		vc1=(npcCharacter v1){inventory=inv2}
		victim=v1{npcCharacter=vc1}
		mz=MazeObjects M.empty (M.fromList [((0,0),victim)])
		(mz2,jp2,g)=killNPC gw mz jp
		dropped=M.lookup (0,0) (items mz2)
	assertEqual "Gold in JP's pocket is not 100" 100 (getGold $ inventory jp2)
	assertEqual "Gold message is not 100" 100 g
	assertEqual "item not dropped" (Just [item]) dropped
	))