TreasureManagerTest.java

package greymerk.roguelike.treasure;

import java.util.Random;

import org.junit.Before;
import org.junit.Test;

import greymerk.roguelike.util.WeightedChoice;
import greymerk.roguelike.util.WeightedRandomizer;
import net.minecraft.init.Bootstrap;
import net.minecraft.init.Items;
import net.minecraft.item.ItemStack;

public class TreasureManagerTest {

	@Before
	public void setup(){
		Bootstrap.register();
	}
	
	@Test
	public void addItem() {
		
		Random rand = new Random();
		WeightedChoice<ItemStack> stick = new WeightedChoice<ItemStack>(new ItemStack(Items.STICK), 1);
		WeightedRandomizer<ItemStack> loot = new WeightedRandomizer<ItemStack>();
		loot.add(stick);
		
		TreasureManager treasure = new TreasureManager();		
		MockChest toAdd = new MockChest(Treasure.ARMOUR, 0);
		treasure.add(toAdd);
		treasure.addItem(rand, Treasure.ARMOUR, 0, loot, 1);
		
		treasure.addItem(rand, Treasure.ARMOUR, 1, loot, 1);
		
		treasure.addItem(rand, Treasure.WEAPONS, 0, loot, 1);
		
		treasure.addItem(rand, Treasure.WEAPONS, 1, loot, 1);
		
		treasure.addItem(rand, Treasure.ARMOUR, loot, 1);
		
		treasure.addItem(rand, Treasure.WEAPONS, loot, 1);
		
		treasure.addItem(rand, 0, loot, 1);
		
		treasure.addItem(rand, 1, loot, 1);
		
	}	
}