def test_get_order_status():
""" Tests - get order status """
game = Game()
game.clear_units()
game.set_units('ITALY', 'A VEN')
game.set_units('AUSTRIA', 'A VIE')
game.set_orders('ITALY', 'A VEN - TYR')
game.set_orders('AUSTRIA', 'A VIE - TYR')
game.process()
results = game.get_order_status()
assert BOUNCE in results['ITALY']['A VEN']
assert BOUNCE in results['AUSTRIA']['A VIE']
assert BOUNCE in game.get_order_status(unit='A VEN')
assert BOUNCE in game.get_order_status(unit='A VIE')
assert BOUNCE in game.get_order_status('ITALY')['A VEN']
assert BOUNCE in game.get_order_status('AUSTRIA')['A VIE']
def test_clear_orders():
""" Test - Clear orders"""
game = Game()
game.clear_units()
game.set_units('ITALY', 'A VEN')
game.set_units('AUSTRIA', 'A VIE')
game.set_orders('ITALY', 'A VEN - TYR')
game.set_orders('AUSTRIA', 'A VIE - TYR')
game.clear_orders()
game.process()
results = game.get_order_status()
assert results['ITALY']['A VEN'] == []
assert results['AUSTRIA']['A VIE'] == []
def test_set_orders():
""" Test - Sets orders """
game = Game()
game.clear_units()
game.set_units('ITALY', 'A VEN')
game.set_units('AUSTRIA', 'A VIE')
game.set_units('FRANCE', 'A PAR')
game.set_orders('ITALY', 'A VEN - TYR')
game.set_orders('AUSTRIA', 'A VIE - TYR')
game.set_orders('FRANCE', ['', '', 'A PAR - GAS', '', '', ''])
game.set_orders('RUSSIA', '')
game.set_orders('GERMANY', [])
assert game.get_orders('FRANCE') == ['A PAR - GAS']
assert not game.get_orders('RUSSIA')
assert not game.get_orders('GERMANY')
game.process()
results = game.get_order_status()
assert BOUNCE in results['ITALY']['A VEN']
assert BOUNCE in results['AUSTRIA']['A VIE']
