import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@ShouldMatchDataSet("datasets/after_create.xml")
@InSequence(2)
public void shouldCreateEntity() throws Exception {        // <4>
    MyEntity myentity = new MyEntity("Some Key", "Some Value");
    em.persist(myentity);
}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@UsingDataSet("datasets/before_update.xml")
@ShouldMatchDataSet("datasets/after_update.xml")
@InSequence(2)
public void shouldUpdateEntity() throws Exception {        // <5>
    MyEntity myentity = em.find(MyEntity.class, "Key 1");
    myentity.setValue("Another Value 1");
}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@UsingDataSet("datasets/seats.yml")
@ShouldMatchDataSet("datasets/expected-seats.yml")
public void shouldMakeACleanup() throws Exception {
    // given
    // from annotation

    // when
    ticketService.doCleanUp();

    // then
    // from annotation
}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@UsingDataSet(value = {"guests.json","sectors.json", "flight.json"})
@ShouldMatchDataSet(value = {"schedule.yml"}, excludeColumns = {"id"})
public void save_NewSchedule_ShouldBeCreated() throws Exception {

    Guest guest = guestService.find(Guest.class, 1);
    Flight flight = flightService.find(Flight.class, 1);

    Schedule saved = scheduleService.create(createSchedule(flight, guest));

    assertNotNull(saved);
    assertNotNull(saved.getId());
}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@UsingDataSet(value = {"schedule.yml","guests.json","sectors.json", "flight.json"})
@ShouldMatchDataSet(value = {"schedule_updated.yml"}, excludeColumns = {"id"})
public void shouldUpdateSchedule(){

	 Schedule schedule = scheduleService.find(Schedule.class, 1);

	 schedule.setScheduleDate(LocalDate.of(2016,9,14));

	 scheduleService.edit(schedule);
}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@UsingDataSet(value = {"schedule.yml","guests.json","sectors.json", "flight.json"})
@ShouldMatchDataSet(value = {"schedule.yml"}, excludeColumns = {"id"})
public void get_ExistingSchedule_Found() throws Exception {

    Schedule schedule = scheduleService.find(Schedule.class, 1);
    assertNotNull(schedule);
}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@UsingDataSet(value = {"schedule.yml","guests.json","sectors.json", "flight.json"})
@ShouldMatchDataSet(value = {"itinerary.yml"}, excludeColumns = {"id"})
public void save_NewItinerary_ShouldBeCreated() throws Exception {

    Guest guest = guestService.find(Guest.class, 1);
    Flight flight = flightService.find(Flight.class, 1);
    Schedule schedule = scheduleService.find(Schedule.class, 1);

    Itinerary  saved =  itineraryService.createItinerary(createItinerary(flight, guest, schedule));

    assertNotNull(saved);
    assertNotNull(saved.getId());
}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@UsingDataSet(value = {"itinerary.yml","schedule.yml","guests.json","sectors.json", "flight.json"})
@ShouldMatchDataSet(value = {"itinerary.yml"}, excludeColumns = {"id"})
public void get_ExistingItinerary_Found() throws Exception {

    Itinerary itinerary = itineraryService.find(Itinerary.class, 1);
    assertNotNull(itinerary);
}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@ShouldMatchDataSet(value = { "guest.json" }, excludeColumns = { "id" })
public void save_NewGuest_ShouldBeCreated() throws Exception {

	Guest saved = guestService.create(createGuest());

	assertNotNull(saved);
	assertNotNull(saved.getId());
}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@UsingDataSet(value = { "guest.json" })
@ShouldMatchDataSet(value = { "guest_updated.json" }, excludeColumns = { "id" })
public void shouldUpdateGuest(){

	Guest guest = guestService.find(Guest.class, 1);
	guest.setFirstname("James");
	guest.setLastname("Gosling");

	guestService.edit(guest);

}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@UsingDataSet(value = { "guest.json" })
@ShouldMatchDataSet(value = { "guest.json" }, excludeColumns = { "id" })
public void get_ExistingGuest_Found() throws Exception {

	Guest guest = guestService.find(Guest.class, 1);
	assertNotNull(guest);
}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@UsingDataSet(value = {"sectors.json"})
@ShouldMatchDataSet(value = {"flight.json"}, excludeColumns = {"id"})
public void save_NewFlight_ShouldBeCreated() throws Exception {

    Sector sourceSector = sectorService.find(Sector.class, 1);

    Sector destSector = sectorService.find(Sector.class, 3);

    Flight saved = flightService.create(createFlight(destSector, sourceSector));

    assertNotNull(saved);
    assertNotNull(saved.getId());
}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@UsingDataSet(value = {"sectors.json", "flight.json"})
@ShouldMatchDataSet(value = {"flight.json"}, excludeColumns = {"id"})
public void get_ExistingFlight_Found() throws Exception {

    Flight flight = flightService.find(Flight.class, 1);
    assertNotNull(flight);
}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@ShouldMatchDataSet(value = {"sector.json"}, excludeColumns = {"id"})
public void save_NewSector_ShouldBeCreated() throws Exception {

    Sector saved = sectorService.create(createSector());

    assertNotNull(saved);
    assertNotNull(saved.getId());
}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@UsingDataSet(value = {"sector.json"})
@ShouldMatchDataSet(value = {"sector_updated.json"}, excludeColumns = {"id"})
public void shouldUpdateSector(){

	Sector sector = sectorService.find(Sector.class, 1);
	sector.setSector("DEL");

	sectorService.edit(sector);
}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@UsingDataSet(value = {"sector.json"})
@ShouldMatchDataSet(value = {"sector.json"}, excludeColumns = {"id"})
public void get_ExistingSector_Found() throws Exception {

    Sector sector = sectorService.find(Sector.class, 1);
    assertNotNull(sector);
}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@Transactional(TransactionMode.COMMIT) // default with persistence extension
@UsingDataSet("datasets/users.yml")
@ShouldMatchDataSet("datasets/expected-users.yml")
public void seriouslyYouAlreadyForgotOpenEJB_questionMark() throws Exception {
    assertEquals(2, em.createQuery("select count(e) from User e", Number.class).getSingleResult().intValue());

    final User user = em.find(User.class, 2L);
    assertNotNull(user);

    user.setName("OpenEJB"); // @Transactional(TransactionMode.COMMIT) will commit it and datasets/expected-users.yml will check it
}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@ShouldMatchDataSet(value = "expected-beers.yml", orderBy = "name")
@Cleanup(phase = TestExecutionPhase.AFTER)
public void should_return_all_beers() throws Exception
{
   assertThat(beersInserter).isNotNull();
}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@UsingDataSet(value = {"itinerary.yml","schedule.yml","guests.json","sectors.json", "flight.json"})
@ShouldMatchDataSet(value = {"itinerary_updated.yml"}, excludeColumns = {"id"})
public void shouldUpdateItinerary(){

	Itinerary itinerary = itineraryService.find(Itinerary.class, 1);

	Guest guest = guestService.find(Guest.class, 2);

	itinerary.setGuest(guest);

	itineraryService.edit(itinerary);
}
 

import org.jboss.arquillian.persistence.ShouldMatchDataSet; //导入依赖的package包/类
@Test
@UsingDataSet(value = {"sectors.json", "flight.json"})
@ShouldMatchDataSet(value = {"flight_updated.json"}, excludeColumns = {"id"})
public void shouldUpdateFlight(){

	   Flight flight = flightService.find(Flight.class, 1);

	   Sector destSector = sectorService.find(Sector.class, 9);

	   flight.setName("AA057");
	   flight.setDestSectorId(destSector);

	   flightService.edit(flight);

}