diff options
Diffstat (limited to 'diplomacy/integration/webdiplomacy_net/tests/test_game.py')
| -rw-r--r-- | diplomacy/integration/webdiplomacy_net/tests/test_game.py | 185 |
1 files changed, 185 insertions, 0 deletions
diff --git a/diplomacy/integration/webdiplomacy_net/tests/test_game.py b/diplomacy/integration/webdiplomacy_net/tests/test_game.py new file mode 100644 index 0000000..7a888d3 --- /dev/null +++ b/diplomacy/integration/webdiplomacy_net/tests/test_game.py @@ -0,0 +1,185 @@ +# ============================================================================== +# Copyright (C) 2019 - Philip Paquette +# +# This program is free software: you can redistribute it and/or modify it under +# the terms of the GNU Affero General Public License as published by the Free +# Software Foundation, either version 3 of the License, or (at your option) any +# later version. +# +# This program is distributed in the hope that it will be useful, but WITHOUT +# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS +# FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more +# details. +# +# You should have received a copy of the GNU Affero General Public License along +# with this program. If not, see <https://www.gnu.org/licenses/>. +# ============================================================================== +""" Test order conversion. """ +from diplomacy.integration.webdiplomacy_net.game import turn_to_phase, unit_dict_to_str, center_dict_to_str, \ + order_dict_to_str + +# ------------------------------------ +# ---- Tests for turn_to_phase ---- +# ------------------------------------ +def test_phase_s1901m(): + """ Tests S1901M """ + phase = turn_to_phase(0, 'Diplomacy') + assert phase == 'S1901M' + +def test_phase_s1901r(): + """ Tests S1901R """ + phase = turn_to_phase(0, 'Retreats') + assert phase == 'S1901R' + +def test_phase_f1901m(): + """ Tests F1901M """ + phase = turn_to_phase(1, 'Diplomacy') + assert phase == 'F1901M' + +def test_phase_f1901r(): + """ Tests F1901R """ + phase = turn_to_phase(1, 'Retreats') + assert phase == 'F1901R' + +def test_phase_w1901a(): + """ Tests W1901A """ + phase = turn_to_phase(1, 'Builds') + assert phase == 'W1901A' + +def test_phase_s1902m(): + """ Tests S1902M """ + phase = turn_to_phase(2, 'Diplomacy') + assert phase == 'S1902M' + + +# ------------------------------------ +# ---- Tests for unit_dict_to_str ---- +# ------------------------------------ +def test_army_france(): + """ Tests Army France """ + unit_dict = {'unitType': 'Army', 'terrID': 47, 'countryID': 2, 'retreating': 'No'} + power_name, unit = unit_dict_to_str(unit_dict) + assert power_name == 'FRANCE' + assert unit == 'A PAR' + +def test_dis_fleet_england(): + """ Tests Dislodged Fleet England """ + unit_dict = {'unitType': 'Fleet', 'terrID': 6, 'countryID': 1, 'retreating': 'Yes'} + power_name, unit = unit_dict_to_str(unit_dict) + assert power_name == 'ENGLAND' + assert unit == '*F LON' + +def test_invalid_unit(): + """ Tests invalid unit """ + unit_dict = {'unitType': 'Fleet', 'terrID': 99, 'countryID': 0, 'retreating': 'No'} + power_name, unit = unit_dict_to_str(unit_dict) + assert power_name == '' + assert unit == '' + + +# -------------------------------------- +# ---- Tests for center_dict_to_str ---- +# -------------------------------------- +def test_center_dict(): + """ Tests parsing centers """ + power_name, center = center_dict_to_str({'countryID': 1, 'terrID': 6}, map_id=1) + assert power_name == 'ENGLAND' + assert center == 'LON' + + power_name, center = center_dict_to_str({'countryID': 2, 'terrID': 47}, map_id=1) + assert power_name == 'FRANCE' + assert center == 'PAR' + + +# ------------------------------------- +# ---- Tests for order_dict_to_str ---- +# ------------------------------------- +def test_s1901m_hold(): + """ Tests hold in S1901M """ + order_dict = {'turn': 0, + 'phase': 'Diplomacy', + 'countryID': 2, + 'terrID': 6, + 'unitType': 'Army', + 'type': 'Hold', + 'toTerrID': '', + 'fromTerrID': '', + 'viaConvoy': ''} + power_name, order = order_dict_to_str(order_dict, phase='Diplomacy') + assert power_name == 'FRANCE' + assert order == 'A LON H' + +def test_s1901r_disband(): + """ Tests disband in S1901R """ + order_dict = {'turn': 0, + 'phase': 'Retreats', + 'countryID': 1, + 'terrID': 6, + 'unitType': 'Fleet', + 'type': 'Disband', + 'toTerrID': '', + 'fromTerrID': '', + 'viaConvoy': ''} + power_name, order = order_dict_to_str(order_dict, phase='Retreats') + assert power_name == 'ENGLAND' + assert order == 'F LON D' + +def test_f1901m_move(): + """ Tests move in F1901M """ + order_dict = {'turn': 1, + 'phase': 'Diplomacy', + 'countryID': 2, + 'terrID': 6, + 'unitType': 'Army', + 'type': 'Move', + 'toTerrID': 47, + 'fromTerrID': '', + 'viaConvoy': 'Yes'} + power_name, order = order_dict_to_str(order_dict, phase='Diplomacy') + assert power_name == 'FRANCE' + assert order == 'A LON - PAR VIA' + +def test_f1901r_retreat(): + """ Tests retreat in F1901R """ + order_dict = {'turn': 1, + 'phase': 'Retreats', + 'countryID': 3, + 'terrID': 6, + 'unitType': 'Army', + 'type': 'Retreat', + 'toTerrID': 47, + 'fromTerrID': '', + 'viaConvoy': ''} + power_name, order = order_dict_to_str(order_dict, phase='Retreats') + assert power_name == 'ITALY' + assert order == 'A LON R PAR' + +def test_w1901a_build(): + """ Tests build army in W1901A """ + order_dict = {'turn': 1, + 'phase': 'Builds', + 'countryID': 2, + 'terrID': 6, + 'unitType': 'Army', + 'type': 'Build Army', + 'toTerrID': 6, + 'fromTerrID': '', + 'viaConvoy': ''} + power_name, order = order_dict_to_str(order_dict, phase='Builds') + assert power_name == 'FRANCE' + assert order == 'A LON B' + +def test_s1902m_hold(): + """ Tests hold in S1902M """ + order_dict = {'turn': 2, + 'phase': 'Diplomacy', + 'countryID': 2, + 'terrID': 6, + 'unitType': 'Army', + 'type': 'Hold', + 'toTerrID': '', + 'fromTerrID': '', + 'viaConvoy': ''} + power_name, order = order_dict_to_str(order_dict, phase='Diplomacy') + assert power_name == 'FRANCE' + assert order == 'A LON H' |