1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 | package com.ibm.ei.engine.golf; import static org.hamcrest.Matchers.containsInAnyOrder; import static org.hamcrest.Matchers.is; import static org.hamcrest.Matchers.notNullValue; import static org.junit.Assert.assertThat; import java.net.URL; import javax.persistence.EntityManager; import org.dbunit.dataset.DataSetException; import org.dbunit.dataset.IDataSet; import org.dbunit.dataset.xml.FlatXmlDataSetBuilder; import org.junit.Test; import com.google.common.base.Function; import com.google.common.base.Optional; import com.google.common.collect.ImmutableList; import com.google.common.collect.Lists; import com.google.common.io.Resources; import com.ibm.ei.persistence.Player; import com.ibm.ei.persistence.Site; import com.ibm.ei.persistence.jpaloaders.Sites; import com.ibm.ei.zepplin.H2DatabaseTestCase; public class XMLFeaturedGroupsTest extends H2DatabaseTestCase { private final Function<Player, String> playerIDFunction = new Function<Player, String>() { @Override public String apply(Player input) { return input.getPlayerId(); } }; @Test public void testFromInputSupplier() throws Exception { EntityManager manager = getEntityManager(); Site site = Sites.loadSite(manager, "masters", 2013); URL group = Resources.getResource("featuredGroup.xml"); ImmutableList<FeaturedGroup> groups = XMLFeaturedGroups .fromInputSupplier(manager, site, Resources.asByteSource(group)); FeaturedGroup group1 = groups.get(0); FeaturedGroup group2 = groups.get(1); assertThat(group1, notNullValue()); assertThat(group2, notNullValue()); assertThat(group1.getPlayers(), notNullValue()); assertThat(group2.getPlayers(), notNullValue()); assertThat(group1.getPlayers().size(), is(3)); assertThat(group2.getPlayers().size(), is(3)); assertThat(Lists.transform(group1.getPlayers(), playerIDFunction), containsInAnyOrder("20396", "1810", "10885")); assertThat(Lists.transform(group2.getPlayers(), playerIDFunction), containsInAnyOrder("24357", "8793", "20229")); } @Override protected Optional<IDataSet> getDataSet() throws DataSetException { return Optional.of((IDataSet)new FlatXmlDataSetBuilder().build(Resources.getResource("players.xml"))); } } |
欢迎光临 电子技术论坛_中国专业的电子工程师学习交流社区-中电网技术论坛 (http://bbs.eccn.com/) | Powered by Discuz! 7.0.0 |