/*
* Copyright (c) 2012 Tom Parker <thpr@users.sourceforge.net>
*
* This program is free software; you can redistribute it and/or modify it under
* the terms of the GNU Lesser General Public License as published by the Free
* Software Foundation; either version 2.1 of the License, or (at your option)
* any later version.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
* details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this library; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
*/
package tokencontent;
import org.junit.Test;
import pcgen.cdom.facet.FacetLibrary;
import pcgen.cdom.facet.analysis.FavoredClassFacet;
import pcgen.cdom.facet.input.RaceInputFacet;
import pcgen.core.PCClass;
import pcgen.core.Race;
import pcgen.gui2.facade.MockUIDelegate;
import pcgen.persistence.PersistenceLayerException;
import pcgen.rules.persistence.token.CDOMToken;
import pcgen.rules.persistence.token.ParseResult;
import pcgen.util.chooser.ChooserFactory;
import plugin.lsttokens.choose.ClassToken;
import plugin.lsttokens.race.FavclassToken;
import plugin.lsttokens.testsupport.TokenRegistration;
import tokenmodel.testsupport.AbstractTokenModelTest;
public class RaceFavClassTest extends AbstractTokenModelTest
{
private static FavclassToken token = new FavclassToken();
private static ClassToken CHOOSE_CLASS_TOKEN = new ClassToken();
protected RaceInputFacet raceInputFacet = FacetLibrary
.getFacet(RaceInputFacet.class);
private FavoredClassFacet fcFacet;
@Override
protected void setUp() throws Exception
{
super.setUp();
fcFacet = FacetLibrary.getFacet(FavoredClassFacet.class);
context.getReferenceContext().constructCDOMObject(PCClass.class, "Favorite");
TokenRegistration.register(CHOOSE_CLASS_TOKEN);
ChooserFactory.setDelegate(new MockUIDelegate());
}
@Test
public void testDirect() throws PersistenceLayerException
{
Race source = create(Race.class, "Source");
ParseResult result = token.parseToken(context, source, "Favorite");
if (result != ParseResult.SUCCESS)
{
result.printMessages();
fail("Test Setup Failed");
}
finishLoad();
assertEquals(baseCount(), targetFacetCount());
raceFacet.directSet(id, source, getAssoc());
assertTrue(containsExpected());
assertEquals(baseCount() + 1, targetFacetCount());
raceFacet.remove(id);
assertEquals(baseCount(), targetFacetCount());
}
@Test
public void testList() throws PersistenceLayerException
{
Race source = create(Race.class, "Source");
ParseResult result = token.parseToken(context, source, "%LIST");
if (result != ParseResult.SUCCESS)
{
result.printMessages();
fail("Test Setup Failed");
}
result = CHOOSE_CLASS_TOKEN.parseToken(context, source, "Favorite");
if (result != ParseResult.SUCCESS)
{
result.printMessages();
fail("Test Setup Failed");
}
finishLoad();
assertEquals(baseCount(), targetFacetCount());
raceInputFacet.set(id, source);
assertTrue(containsExpected());
assertEquals(baseCount() + 1, targetFacetCount());
raceInputFacet.remove(id);
assertEquals(baseCount(), targetFacetCount());
}
@Override
public CDOMToken<?> getToken()
{
return token;
}
protected boolean containsExpected()
{
return fcFacet.contains(id, context.getReferenceContext()
.silentlyGetConstructedCDOMObject(PCClass.class, "Favorite"));
}
protected int targetFacetCount()
{
return fcFacet.getCount(id);
}
protected int baseCount()
{
return 0;
}
}