@Test(expected = IllegalArgumentException.class)
public void addWorkerTwiceErrorTest() {
Person newWorker = new Person(crs.getModel());
crs.addWorker(newWorker);
crs.addWorker(newWorker);
}
@Test
public void testWorkerSize() throws Exception {
assertEquals(crs.getNumberOfWorkers(), 2);
Person w = crs.removeLastWorker();
assertEquals(crs.getNumberOfWorkers(), 1);
crs.addWorker(w);
assertEquals(crs.getNumberOfWorkers(), 2);
}
@Test
public void testCheckForWorkers() throws Exception {
assertTrue(
crs.checkForWorkers()); // you have two workers, the minimum is one, you should be okay
crs.removeLastWorker(); // remove a worker
assertTrue(crs.checkForWorkers()); // should still work
Person w = crs.removeLastWorker();
assertTrue(!crs.checkForWorkers()); // should still work
crs.addWorker(w);
assertTrue(crs.checkForWorkers()); // should still work
}
@Test
public void testStep() throws Exception {
Blueprint b =
Blueprint.simpleBlueprint(
DifferentiatedGoodType.CAPITAL, 1, DifferentiatedGoodType.CAPITAL, 1);
MacroII localMacroII = new MacroII(1);
localMacroII.schedule =
mock(Schedule.class); // put in a fake schedule so we avoid steppables firing at random
Firm f = new Firm(localMacroII);
Plant localCRS = new Plant(b, new Firm(localMacroII));
localCRS.setPlantMachinery(
new CRSExponentialMachinery(DifferentiatedGoodType.CAPITAL, f, 0, localCRS, 1f, 1f));
f.addPlant(localCRS);
SalesDepartmentAllAtOnce stub = mock(SalesDepartmentAllAtOnce.class);
when(stub.isSelling(any(Good.class))).thenReturn(Boolean.TRUE);
localCRS.getOwner().registerSaleDepartment(stub, DifferentiatedGoodType.CAPITAL); // fake sales
// department so that you don't try selling the stuff you build
localCRS.setCostStrategy(new EmptyCostStrategy());
try {
localCRS.getModel();
localCRS.getModel().getPhaseScheduler().step(localCRS.getModel());
assertEquals(localCRS.getStatus(), PlantStatus.WAITING_FOR_WORKERS);
localCRS.addWorker(new Person(localCRS.getModel()));
localCRS.addWorker(new Person(localCRS.getModel()));
localCRS.getModel().getPhaseScheduler().step(localCRS.getModel());
assertEquals(localCRS.getStatus(), PlantStatus.WAITING_FOR_INPUT);
localCRS
.getOwner()
.receive(
Good.getInstanceOfDifferentiatedGood(
DifferentiatedGoodType.CAPITAL, drs.getOwner(), 1),
drs.getOwner());
localCRS.getModel().getPhaseScheduler().step(localCRS.getModel());
assertEquals(
localCRS.getStatus(), PlantStatus.READY); // you should automatically start production!
} catch (Exception e) {
fail("Can't throw exceptions now");
}
}
@Before
public void setUp() throws Exception {
Blueprint b =
new Blueprint.Builder()
.output(DifferentiatedGoodType.CAPITAL, 2)
.build(); // create a simple output
macro = new MacroII(1);
macro.schedule =
mock(Schedule.class); // put in a fake schedule so we avoid steppables firing at random
Firm f = new Firm(macro);
crs = new Plant(b, new Firm(macro));
crs.setPlantMachinery(
new CRSExponentialMachinery(DifferentiatedGoodType.CAPITAL, f, 0, crs, 1f, 1f));
irs = new Plant(b, new Firm(macro));
irs.setPlantMachinery(
new IRSExponentialMachinery(DifferentiatedGoodType.CAPITAL, f, 0, irs, 1f, 1f));
drs = new Plant(b, new Firm(macro));
drs.setPlantMachinery(
new DRSExponentialMachinery(DifferentiatedGoodType.CAPITAL, f, 0, drs, 1f, 1f));
Person w1 = new Person(macro);
Person w2 = new Person(macro);
crs.addWorker(w1);
crs.addWorker(w2);
w1.hired(crs.getOwner(), 9999999);
w2.hired(crs.getOwner(), 9999999);
w1 = new Person(macro);
w2 = new Person(macro);
irs.addWorker(w1);
irs.addWorker(w2);
w1.hired(crs.getOwner(), 9999999);
w2.hired(crs.getOwner(), 9999999);
w1 = new Person(macro);
w2 = new Person(macro);
drs.addWorker(w1);
drs.addWorker(w2);
w1.hired(crs.getOwner(), 9999999);
w2.hired(crs.getOwner(), 9999999);
SalesDepartmentAllAtOnce stub = mock(SalesDepartmentAllAtOnce.class);
crs.getOwner()
.registerSaleDepartment(
stub,
DifferentiatedGoodType
.CAPITAL); // fake sales department so that you don't sell the stuff you
// completeProductionRunNow
irs.getOwner()
.registerSaleDepartment(
stub,
DifferentiatedGoodType
.CAPITAL); // fake sales department so that you don't sell the stuff you
// completeProductionRunNow
drs.getOwner()
.registerSaleDepartment(
stub,
DifferentiatedGoodType
.CAPITAL); // fake sales department so that you don't sell the stuff you
// completeProductionRunNow
when(stub.isSelling(any(Good.class))).thenReturn(Boolean.TRUE);
crs.setCostStrategy(new EmptyCostStrategy());
irs.setCostStrategy(new EmptyCostStrategy());
drs.setCostStrategy(new EmptyCostStrategy());
}
@Test
public void testTotalInputs() {
Blueprint b = Blueprint.simpleBlueprint(DifferentiatedGoodType.CAPITAL, 1, BEEF, 1);
MacroII localMacroII = mock(MacroII.class);
PhaseScheduler scheduler = mock(PhaseScheduler.class);
when(localMacroII.getCurrentPhase()).thenReturn(ActionOrder.PRODUCTION);
when(localMacroII.getPhaseScheduler()).thenReturn(scheduler);
when(localMacroII.getWeekLength()).thenReturn(7f);
Firm f = new Firm(localMacroII);
Plant localCRS = new Plant(b, new Firm(localMacroII));
localCRS.setPlantMachinery(
new LinearConstantMachinery(DifferentiatedGoodType.CAPITAL, f, 0, localCRS));
f.addPlant(localCRS);
localCRS.addWorker(new Person(localCRS.getModel()));
localCRS.addWorker(new Person(localCRS.getModel()));
localCRS
.getOwner()
.registerSaleDepartment(
mock(SalesDepartmentAllAtOnce.class),
BEEF); // fake sales department so that you don't sell the stuff you
// completeProductionRunNow
localCRS.setCostStrategy(new InputCostStrategy(localCRS));
// initially they should be set up to 0
assertEquals(localCRS.getThisWeekInputCosts(), 0);
assertEquals(localCRS.getLastWeekInputCosts(), 0);
// now when I step you, you should say you are waiting for inputs
localCRS.getModel();
localCRS.step(localMacroII);
assertEquals(localCRS.getStatus(), PlantStatus.WAITING_FOR_INPUT);
assertEquals(localCRS.getThisWeekInputCosts(), 0); // input costs should still be 0
assertEquals(localCRS.getLastWeekInputCosts(), 0);
// now I give you one input that costs 10$
localCRS
.getOwner()
.receive(
Good.getInstanceOfDifferentiatedGood(
DifferentiatedGoodType.CAPITAL, drs.getOwner(), 10),
drs.getOwner());
localCRS.step(localCRS.getModel());
assertEquals(localCRS.getThisWeekInputCosts(), 10); // input costs should still now be 10
assertEquals(localCRS.getLastWeekInputCosts(), 0);
// I'll give you another and force you to step again
localCRS
.getOwner()
.receive(
Good.getInstanceOfDifferentiatedGood(
DifferentiatedGoodType.CAPITAL, drs.getOwner(), 10),
drs.getOwner());
localCRS.step(localCRS.getModel());
assertEquals(localCRS.getThisWeekInputCosts(), 20); // input costs should still now be 20
assertEquals(localCRS.getLastWeekInputCosts(), 0);
// with weekend it should reset this week and make lastweekinput costs equal 20
localCRS.weekEnd(7);
assertEquals(localCRS.getThisWeekInputCosts(), 0); // reset to 0
assertEquals(localCRS.getLastWeekInputCosts(), 20); // now it's 20
// try one more time
localCRS
.getOwner()
.receive(
Good.getInstanceOfDifferentiatedGood(
DifferentiatedGoodType.CAPITAL, drs.getOwner(), 10),
drs.getOwner());
localCRS.step(localCRS.getModel());
assertEquals(localCRS.getThisWeekInputCosts(), 10); // input costs should now be 10
assertEquals(localCRS.getLastWeekInputCosts(), 20); // this should be stuck at 20
// new weekend, forget 20$ and reset this week
localCRS.weekEnd(14);
assertEquals(localCRS.getThisWeekInputCosts(), 0); // reset to 0
assertEquals(localCRS.getLastWeekInputCosts(), 10); // now it's 10
// another weekend, everything is at 0
localCRS.weekEnd(0);
assertEquals(localCRS.getThisWeekInputCosts(), 0); // reset to 0
assertEquals(localCRS.getLastWeekInputCosts(), 0); // now it's 10
}