def test_create_new_fleet(self):
self.turn.orders.create(
source=self.london,
outcome=OutcomeType.SUCCEEDS,
nation=self.england,
type=OrderType.BUILD,
piece_type=PieceType.FLEET,
)
piece = create_new_pieces(self.turn)[0]
self.assertEqual(piece.type, PieceType.FLEET)
def test_create_new_pieces_failing_order(self):
self.turn.orders.create(
source=self.london,
outcome=OutcomeType.FAILS,
nation=self.england,
type=OrderType.BUILD,
piece_type=PieceType.ARMY,
)
pieces = create_new_pieces(self.turn)
self.assertEqual(pieces, [])
def test_create_new_pieces_army(self):
self.turn.orders.create(
source=self.london,
outcome=OutcomeType.SUCCEEDS,
nation=self.england,
type=OrderType.BUILD,
piece_type=PieceType.ARMY,
)
pieces = create_new_pieces(self.turn)
self.assertEqual(len(pieces), 1)
piece = pieces[0]
self.assertEqual(piece.nation, self.england)
self.assertEqual(piece.game, self.game)
self.assertEqual(piece.type, PieceType.ARMY)
def test_create_both_types_same_nation(self):
self.turn.orders.create(
source=self.london,
outcome=OutcomeType.SUCCEEDS,
nation=self.england,
type=OrderType.BUILD,
piece_type=PieceType.ARMY,
)
self.turn.orders.create(
source=self.paris,
outcome=OutcomeType.SUCCEEDS,
nation=self.england,
type=OrderType.BUILD,
piece_type=PieceType.FLEET,
)
pieces = create_new_pieces(self.turn)
self.assertEqual(len(pieces), 2)
def test_create_new_pieces_no_orders(self):
pieces = create_new_pieces(self.turn)
self.assertEqual(pieces, [])