From cb280ed5a040eba33b927c07a86b995124ac62a6 Mon Sep 17 00:00:00 2001 From: Michael Wittig Date: Sat, 10 Dec 2016 18:40:38 +0100 Subject: [PATCH] #99 don't use _meta from models --- core/tests/unit/parser/test_player_statistics_parser.py | 6 +++--- core/tests/unit/test_core_models.py | 2 ++ 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/core/tests/unit/parser/test_player_statistics_parser.py b/core/tests/unit/parser/test_player_statistics_parser.py index 3365c3d..65e6a98 100644 --- a/core/tests/unit/parser/test_player_statistics_parser.py +++ b/core/tests/unit/parser/test_player_statistics_parser.py @@ -16,9 +16,9 @@ def setUp(self): user = OFMUserFactory.create() season = SeasonFactory.create() - country_choices = dict(Country._meta.get_field('country').choices) - country_no_greece = list(country_choices.keys())[list(country_choices.values()).index('Griechenland')] - nationality_greece, _ = Country.objects.get_or_create(country=country_no_greece) + country_choices = Country.get_choices() + country_iso_greece = list(country_choices.keys())[list(country_choices.values()).index('Griechenland')] + nationality_greece, _ = Country.objects.get_or_create(country=country_iso_greece) self.player = Player.objects.create( id='159883060', diff --git a/core/tests/unit/test_core_models.py b/core/tests/unit/test_core_models.py index 9e11d4e..170305e 100644 --- a/core/tests/unit/test_core_models.py +++ b/core/tests/unit/test_core_models.py @@ -119,6 +119,7 @@ def test_create_finance(self): self.assertTrue(f.matchday is not None) self.assertEqual(f.balance, 1000) + self.assertEqual(f.income_visitors_league, 100) self.assertEqual(f.income_sponsoring, 0) self.assertEqual(f.income_cup, 0) @@ -129,6 +130,7 @@ def test_create_finance(self): self.assertEqual(f.income_friendlies, 0) self.assertEqual(f.income_funcup, 0) self.assertEqual(f.income_betting, 0) + self.assertEqual(f.expenses_player_salaries, 100) self.assertEqual(f.expenses_stadium, 0) self.assertEqual(f.expenses_youth, 0)