fixing broken tests

Author: twalen

oioioi/contests/processors.py

@@ -35,16 +35,6 @@ def recent_contests(request):
             for c in (mapping.get(id) for id in ids)
             if c is not None and c != request.contest
         ]
-    elif False:
-        visible_query = visible_contests_queryset(request)
-        c_views = (
-            Contest.objects.filter(contestview__user=request.real_user)
-            .filter(visible_query)
-            .order_by('-contestview__timestamp')
-            .only("id", "name")
-            .distinct()
-        )[: getattr(settings, "NUM_RECENT_CONTESTS", 5)]
-        return list(c_views)
     else:
         c_views = ContestView.objects.filter(user=request.real_user).select_related(
             'contest'

oioioi/globalmessage/processors.py

@@ -16,10 +16,10 @@ def global_message_processor(request):
     if message.visible(time):
         return {
             'extra_body_global_message': render_to_string(
-                'global-message-user.html', {'global_message': message, 'start': message.start.strftime('%s')}
+                'global-message-user.html', {'global_message': message, 'start': message.start.strftime('%s') if message.start else ''}
             ),
             'extra_admin_global_message': render_to_string(
-                'global-message-admin.html', {'global_message': message, 'start': message.start.strftime('%s')}
+                'global-message-admin.html', {'global_message': message, 'start': message.start.strftime('%s') if message.start else ''}
             ),
         }
 

oioioi/oi/files/rspo_schools.bak

@@ -0,0 +1 @@
+Numer RSPO;REGON;NIP;Typ;Nazwa;Kod terytorialny województwo;Kod terytorialny powiat;Kod terytorialny gmina;Kod terytorialny miejscowość;Kod terytorialny ulica;Województwo;Powiat;Gmina;Miejscowość;Rodzaj miejscowości;Ulica;Numer budynku;Numer lokalu;Kod pocztowy;Poczta;Telefon;Faks;E-mail;Strona www;Publiczność status;Kategoria uczniów;Specyfika placówki;Imię i nazwisko dyrektora;Data założenia;Data rozpoczęcia działalności;Data likwidacji;Typ organu prowadzącego;Nazwa organu prowadzącego;REGON organu prowadzącego;NIP organu prowadzącego;Województwo organu prowadzącego;Powiat organu prowadzącego;Gmina organu prowadzącego;Miejsce w strukturze;RSPO podmiotu nadrzędnego;Typ podmiotu nadrzędnego;Nazwa podmiotu nadrzędnego;Liczba uczniów;Tereny sportowe;Języki nauczane;Czy zatrudnia logopedę;Czy zatrudnia psychologa;Czy zatrudnia pedagoga;Oddziały podstawowe wg specyfiki;Oddziały dodatkowe;Zawód;Zawód artystyczny;Zawód w KPSS;Dziedzina BCU

oioioi/oi/files/rspo_schools.csv

@@ -0,0 +1,9 @@
+[
+{
+  "pk": 1,
+  "model": "oi.schooltype",
+  "fields": {
+    "name": "Liceum ogólnokształcące"
+  }
+}
+]

oioioi/oi/fixtures/test_school_types.json

@@ -297,7 +297,7 @@ def read_rspo_csv_file(self, filename):
                 )
             SCHOOL_TYPES = [type.name for type in SchoolType.objects.all()]
             return [
-                school
+                {key: school[key] for key in RSPO_CSV_COLUMNS}
                 for school in reader
                 if (
                     school['Typ'] in SCHOOL_TYPES

oioioi/oi/management/commands/import_schools.py

@@ -2,6 +2,7 @@
 import os
 import re
 from datetime import datetime, timedelta, timezone  # pylint: disable=E0611
+import pytest
 
 from django.contrib.admin.utils import quote
 from django.contrib.auth.models import User
@@ -25,6 +26,7 @@ class TestOIAdmin(TestCase):
         'test_contest',
         'test_oi_registration',
         'test_permissions',
+        'test_school_types',
     ]
 
     def test_admin_menu(self):
@@ -39,12 +41,18 @@ def test_admin_menu(self):
         self.assertNotContains(response, 'Regions')
 
     def test_schools_import(self):
-        filename = os.path.join(os.path.dirname(__file__), 'files', 'schools.csv')
+        filename = os.path.join(os.path.dirname(__file__), 'files', 'rspo_schools.csv')
+        backup_filename = os.path.join(os.path.dirname(__file__), 'files', 'rspo_schools.bak')
         manager = import_schools.Command()
-        manager.run_from_argv(['manage.py', 'import_schools', filename])
-        self.assertEqual(School.objects.count(), 3)
-        school = School.objects.get(postal_code='02-044')
-        self.assertEqual(school.city, u'Bielsko-Biała Zdrój')
+        manager.run_from_argv([
+            'manage.py',
+            'import_schools',
+            '--first-import', filename,
+            '--backup-filename', backup_filename
+        ])
+        self.assertEqual(School.objects.count(), 26)
+        school = School.objects.get(postal_code='02-172')
+        self.assertEqual(school.city, 'Włochy')
 
     def test_safe_exec_mode(self):
         contest = Contest.objects.get()
@@ -172,7 +180,6 @@ def test_participants_registration(self):
 
         self.assertContains(response, 'Postal code')
         self.assertContains(response, 'School')
-        self.assertContains(response, 'add it')
         self.assertContains(response, 'Test terms accepted')
         self.assertContains(response, '1977')
 
@@ -201,15 +208,16 @@ def test_participants_registration(self):
         self.assertEqual(registration.address, reg_data['address'])
         self.assertEqual(registration.school.address, 'Nowowiejska 37a')
 
+    @pytest.mark.xfail
     def test_registration_with_new_school(self):
+        """currently the registration form for OI does not support adding new school"""
         contest = Contest.objects.get()
         user = User.objects.get(username='test_user')
         url = reverse('participants_register', kwargs={'contest_id': contest.id})
         self.assertTrue(self.client.login(username='test_user'))
         response = self.client.get(url)
         self.assertContains(response, 'Postal code')
         self.assertContains(response, 'School')
-        self.assertContains(response, 'add it')
 
         user.first_name = 'Sir Lancelot'
         user.last_name = 'du Lac'