/*
   * Same as testWaitForNewTurnToComplete with one difference: it has to wait for three turns
   * to gain sufficient resources.
   */
  @Test
  public void testCompleteAfterThreeTurns() {
    // Add mock eventlistener
    AgendaEventListener ael = mock(AgendaEventListener.class);
    ksession.addEventListener(ael);
    // prepare test data
    CityDTO city = getCity(1L, "marefy");
    Set<String> improvements = new HashSet<String>();
    city.setImprovements(improvements);
    city.getEnabledUnitTypes().add("phalanx");
    city.setResourcesSurplus(60);
    UnitTypeDTO unitType = getUnitType(1L, "phalanx", 150);

    // insert test data as facts
    game.insert(unitType);
    game.insert(city);

    // this unit will be created, trigger of the process
    game.cityBeginUnit(city.getId(), "phalanx");

    ArgumentCaptor<AfterMatchFiredEvent> aafe = ArgumentCaptor.forClass(AfterMatchFiredEvent.class);
    verify(ael, atLeastOnce()).afterMatchFired(aafe.capture());
    List<String> firedRules = getFiredRules(aafe.getAllValues());

    Assert.assertTrue(firedRules.contains("City Start Build Unit"));

    // fst turn
    ksession.getEntryPoint("GameControlStream").insert(new TurnEvent());
    ksession.fireAllRules();

    // simulate production update on new turn
    city.setResourcesSurplus(60);
    ksession.update(ksession.getFactHandle(city), city);
    // snd turn
    ksession.getEntryPoint("GameControlStream").insert(new TurnEvent());
    ksession.fireAllRules();

    // simulate production update on new turn
    city.setResourcesSurplus(60);
    ksession.update(ksession.getFactHandle(city), city);

    // third turn
    ksession.getEntryPoint("GameControlStream").insert(new TurnEvent());
    ksession.fireAllRules();

    Set<Long> units = city.getHomeUnits();
    Assert.assertTrue("City Has New Unit.", units.size() == 1);

    // get new unit
    UnitDTO result = game.getUnit(units.iterator().next());
    Assert.assertNotNull("Unit Is Within ksession", result);

    Assert.assertNotNull("Unit Has Assigned Id", result.getId());

    Assert.assertTrue("City Current Unit Null", city.getCurrentUnit() == null);

    Assert.assertTrue("City Contains Courthouse Improvement.", city.getHomeUnits().size() == 1);

    Assert.assertTrue("Created Unit Is Of Given Type.", result.getType() == unitType.getIdent());
  }
  /*
   * Tests cancel signal sent to active process.
   */
  @Test
  public void testCancelUnit() {
    // Add mock eventlistener to collect fired rules
    AgendaEventListener ael = mock(AgendaEventListener.class);
    ksession.addEventListener(ael);

    // prepare test data
    CityDTO city = getCity(1L, "marefy");
    Set<String> improvements = new HashSet<String>();
    city.setImprovements(improvements);
    city.getEnabledUnitTypes().add("phalanx");
    city.setResourcesSurplus(300);

    UnitTypeDTO unitType = getUnitType(1L, "phalanx", 150);
    // insert test data as facts
    game.insert(unitType);

    game.insert(city);

    // this unit will be created, trigger of the process
    game.cityBeginUnit(city.getId(), "phalanx");

    // afterMatchFired events contain information about fired rules
    ArgumentCaptor<AfterMatchFiredEvent> aafe = ArgumentCaptor.forClass(AfterMatchFiredEvent.class);
    verify(ael, atLeastOnce()).afterMatchFired(aafe.capture());
    List<String> firedRules = getFiredRules(aafe.getAllValues());

    Assert.assertTrue(
        "Rule City Start Build Unit Fired", firedRules.contains("City Start Build Unit"));

    // cancel the waiting process
    ksession.getEntryPoint("ActionCanceledStream").insert(new UnitEvent(city.getId()));
    ksession.fireAllRules();

    Assert.assertFalse("City Has Not Any Unit.", city.getHomeUnits().size() > 0);
  }
  /**
   * Tests whether can finish unit creation when has enough resources after one turn. Process is
   * started by setting the currentUnit property of a CityDTO object.
   */
  @Test
  public void testWaitForNewTurnToComplete() {
    // add mock event listener to collect fired rules
    AgendaEventListener ael = mock(AgendaEventListener.class);
    ksession.addEventListener(ael);

    // prepare test data
    CityDTO city = getCity(1L, "marefy");
    Set<String> improvements = new HashSet<String>();
    city.setImprovements(improvements);
    city.getEnabledUnitTypes().add("phalanx");
    // this type of unit will be created and triggers the process
    city.setResourcesSurplus(300);

    UnitTypeDTO unitType = getUnitType(1L, "phalanx", 150);

    // insert test data as facts
    game.insert(unitType);
    game.insert(city);

    game.cityBeginUnit(city.getId(), unitType.getIdent());

    // get active processes
    List<ProcessInstance> processes = (List<ProcessInstance>) ksession.getProcessInstances();

    // get fired rules
    ArgumentCaptor<AfterMatchFiredEvent> aafe = ArgumentCaptor.forClass(AfterMatchFiredEvent.class);
    verify(ael, atLeastOnce()).afterMatchFired(aafe.capture());
    List<String> firedRules = getFiredRules(aafe.getAllValues());

    // test whether process started
    Assert.assertTrue(
        "City Start Build Unit Rule Fired", firedRules.contains("City Start Build Unit"));
    // and process is placed in working memory
    Assert.assertTrue("One Process Should Be Active", processes.size() == 1);
    // get the process
    Long pId = processes.get(0).getId();

    // process is active, it waits for a TurnEvent
    assertProcessInstanceActive(pId, ksession);

    // signal TurnEvent
    ksession.getEntryPoint("GameControlStream").insert(new TurnEvent());
    ksession.fireAllRules();

    // Process waited only for one turn
    assertProcessInstanceCompleted(pId, ksession);

    Set<Long> units = city.getHomeUnits();
    Assert.assertTrue("City Has New Unit.", units.size() == 1);

    // get new unit
    QueryResults results =
        ksession.getQueryResults("getUnit", new Object[] {units.iterator().next()});
    Assert.assertTrue("Unit Is Within ksession", results.size() == 1);
    QueryResultsRow row = results.iterator().next();
    UnitDTO unit = (UnitDTO) row.get("$unit");
    Assert.assertNotNull("Unit Is Not Null", unit);

    Assert.assertNotNull("Unit Has Assigned Id", unit.getId());

    Assert.assertTrue("No Pending Processes Are There", ksession.getProcessInstances().size() == 0);

    Assert.assertTrue("City Current Unit Null", city.getCurrentUnit() == null);
  }
 private static CityDTO getCity(Long id, String name) {
   CityDTO city = new CityDTO();
   city.setId(id);
   city.setName(name);
   city.setResourcesConsumption(0);
   city.setResourcesProduction(0);
   city.setUnitsSupport(0);
   city.setFoodConsumption(0);
   city.setFoodProduction(0);
   city.setFoodStock(0);
   city.setSize(0);
   city.setTradeProduction(0);
   city.setPeopleEntertainers(0);
   city.setPeopleScientists(0);
   city.setPeopleTaxmen(0);
   city.setWeLoveDay(false);
   city.setDisorder(false);
   city.setSize(1);
   city.setPeopleHappy(0);
   city.setPeopleContent(0);
   city.setPeopleUnhappy(0);
   city.setImprovements(new HashSet<String>());
   city.setHomeUnits(new HashSet<Long>());
   return city;
 }
  @Test
  public void testTwoConsecutiveUnits() {
    // add mock event listener to collect fired rules
    AgendaEventListener ael = mock(AgendaEventListener.class);
    ksession.addEventListener(ael);

    // prepare test data
    CityDTO city = getCity(1L, "marefy");
    Set<String> improvements = new HashSet<String>();
    city.setImprovements(improvements);
    city.getEnabledUnitTypes().add("phalanx");
    // this type of unit will be created and triggers the process
    city.setResourcesSurplus(300);

    UnitTypeDTO unitType = getUnitType(1L, "phalanx", 150);

    // insert test data as facts
    game.insert(unitType);
    game.insert(city);

    game.cityBeginUnit(city.getId(), unitType.getIdent());

    // get fired rules
    ArgumentCaptor<AfterMatchFiredEvent> aafe = ArgumentCaptor.forClass(AfterMatchFiredEvent.class);
    verify(ael, atLeastOnce()).afterMatchFired(aafe.capture());
    List<String> firedRules = getFiredRules(aafe.getAllValues());

    // test whether process started
    Assert.assertTrue(
        "City Start Build Unit Rule Fired", firedRules.contains("City Start Build Unit"));

    // signal TurnEvent
    ksession.getEntryPoint("GameControlStream").insert(new TurnEvent());
    ksession.fireAllRules();

    Set<Long> units = city.getHomeUnits();
    Assert.assertTrue("City Has New Unit.", units.size() == 1);

    // get new unit
    UnitDTO result = game.getUnit(units.iterator().next());
    Assert.assertNotNull("Unit Is Within ksession", result);

    Assert.assertNotNull("Unit Has Assigned Id", result.getId());

    Assert.assertTrue("City Current Unit Null", city.getCurrentUnit() == null);

    city.setResourcesSurplus(300);
    ksession.update(ksession.getFactHandle(city), city);

    game.cityBeginUnit(city.getId(), unitType.getIdent());

    // get fired rules
    ArgumentCaptor<AfterMatchFiredEvent> aafe2 =
        ArgumentCaptor.forClass(AfterMatchFiredEvent.class);
    verify(ael, atLeastOnce()).afterMatchFired(aafe2.capture());
    List<String> firedRules2 = getFiredRules(aafe2.getAllValues());

    // test whether process started
    Assert.assertTrue(
        "City Start Build Unit Rule Fired", firedRules2.contains("City Start Build Unit"));

    // signal TurnEvent
    ksession.getEntryPoint("GameControlStream").insert(new TurnEvent());
    ksession.fireAllRules();

    Set<Long> units2 = city.getHomeUnits();
    Assert.assertTrue("City Has Two New Units.", units2.size() == 2);

    // get new unit
    Collection<UnitDTO> results = game.getUnits(units2);
    Assert.assertTrue("Units Are Within ksession", results.size() == 2);

    Assert.assertTrue("City Current Unit Null", city.getCurrentUnit() == null);
  }