INIT

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'