Source code for oioioi.amppz.tests

from django.contrib.auth.models import User
from django.urls import reverse

from oioioi.base.tests import TestCase
from oioioi.contests.models import Contest
from oioioi.participants.models import Participant


[docs]class TestAMPPZContestController(TestCase):
[docs]    fixtures = ['test_users', 'test_contest']

[docs]    def setUp(self):
        contest = Contest.objects.get()
        contest.controller_name = 'oioioi.amppz.controllers.AMPPZContestController'
        contest.save()

        user = User.objects.get(username='test_user')
        p = Participant(contest=contest, user=user)
        p.save()

[docs]    def test_amppz_menu(self):
        contest = Contest.objects.get()
        self.assertTrue(self.client.login(username='test_user'))
        response = self.client.get(
            reverse('default_contest_view', kwargs={'contest_id': contest.id}),
            follow=True,
        )
        self.assertContains(response, 'amppz/images/menu-icon')
        self.assertContains(response, 'amppz/images/logo')

[docs]    def test_safe_exec_mode(self):
        contest = Contest.objects.get()
        self.assertEqual(contest.controller.get_safe_exec_mode(), 'cpu')