Skip to content

Commit

Permalink
Added unit to .miz file
Browse files Browse the repository at this point in the history
* Making sure units are properly loaded and saved
  for countries which do have them.
  • Loading branch information
332fg-raven committed Jan 17, 2024
1 parent 42a28ce commit 44e3e58
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 2 deletions.
Binary file not shown.
Binary file removed tests/missions/empty-mission-with-coalitions.miz
Binary file not shown.
8 changes: 6 additions & 2 deletions tests/test_mission.py
Original file line number Diff line number Diff line change
Expand Up @@ -1240,18 +1240,19 @@ def test_action_a_out_picture_u(self) -> None:

def test_empty_mission_with_coalitions(self) -> None:
m = dcs.mission.Mission()
m_filename = "tests/missions/empty-mission-with-coalitions.miz"
m_filename = "tests/missions/countries-without-units-on-the-map.miz"
m.load_file(m_filename)

m_blue_countries = m.coalition['blue'].countries
for country in ["Australia", "UK", "USA", "USSR"]:
self.assertIn(country, m_blue_countries)
self.assertEqual(len(m.coalition['blue'].countries["UK"].plane_group), 1)

m_red_countries = m.coalition['red'].countries
for country in ["Third Reich", "Bulgaria", "Romania", "Finland"]:
self.assertIn(country, m_red_countries)

m2_miz_filename = "missions/saved.empty-mission-with-coalitions.miz"
m2_miz_filename = "missions/saved.countries-without-units-on-the-map.miz"
m.save(m2_miz_filename)

m2 = dcs.mission.Mission()
Expand All @@ -1260,3 +1261,6 @@ def test_empty_mission_with_coalitions(self) -> None:
m2_red_countries = m2.coalition['red'].countries
self.assertTrue(sorted(m_blue_countries.keys()), sorted(m2_blue_countries.keys()))
self.assertTrue(sorted(m_red_countries.keys()), sorted(m2_red_countries.keys()))
self.assertEqual(len(m.coalition['blue'].countries["UK"].plane_group),
len(m2.coalition['blue'].countries["UK"].plane_group))

0 comments on commit 44e3e58

Please sign in to comment.