# vim: set fileencoding=utf-8 :
# encoding: utf-8
###############################################################################
# #
# Copyright (c) 2020 Idiap Research Institute, http://www.idiap.ch/ #
# Contact: beat.support@idiap.ch #
# #
# This file is part of the beat.web module of the BEAT platform. #
# #
# Commercial License Usage #
# Licensees holding valid commercial BEAT licenses may use this file in #
# accordance with the terms contained in a written agreement between you #
# and Idiap. For further information contact tto@idiap.ch #
# #
# Alternatively, this file may be used under the terms of the GNU Affero #
# Public License version 3 as published by the Free Software and appearing #
# in the file LICENSE.AGPL included in the packaging of this file. #
# The BEAT platform 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. #
# #
# You should have received a copy of the GNU Affero Public License along #
# with the BEAT platform. If not, see http://www.gnu.org/licenses/. #
# #
###############################################################################
import io
from unittest.mock import patch
from django.contrib.auth.models import User
from django.contrib.sites.models import Site
from django.core import mail
from django.core.management import call_command
from django.test import override_settings
from ..algorithms.tests.core import setup_users
from ..common.testutils import BaseTestCase
from ..common.testutils import tearDownModule # noqa test runner will call it
from ..utils.tests.helpers import reload_urlconf
from .models import Leaderboard
from .models import Search
[docs]class UpdateLeaderBoardManagementTestCase(BaseTestCase):
[docs] def setUp(self):
setup_users()
self.johndoe = User.objects.get(username="johndoe")
self.jackdoe = User.objects.get(username="jackdoe")
self.janedoe = User.objects.get(username="janedoe")
[docs] def run_command(self):
new_io = io.StringIO()
call_command("update_leaderboards", stdout=new_io)
return new_io.getvalue().strip()
[docs] def test_leaderboard_update(self):
current_site = Site.objects.get_current()
search = Search.objects.create(author=self.johndoe, name="test")
search.filters = "{}"
search.settings = "{}"
search.save()
leaderboard = Leaderboard.objects.create(search=search)
leaderboard.notify.add(self.johndoe, self.jackdoe, self.janedoe)
leaderboard.save()
# Simulating update
with patch.object(Leaderboard, "update_experiments", return_value=True):
for prefix in ["", "/platform"]:
with self.subTest(url_prefix=prefix):
with override_settings(URL_PREFIX=prefix):
reload_urlconf()
mail.outbox = []
self.run_command()
self.assertEqual(len(mail.outbox), 1)
self.assertTrue(
f"https://{current_site.domain}{prefix}"
in mail.outbox[0].body
)
[docs]class EmailOnLeaderboarDeletionTestCase(BaseTestCase):
[docs] def setUp(self):
setup_users()
self.johndoe = User.objects.get(username="johndoe")
self.jackdoe = User.objects.get(username="jackdoe")
self.janedoe = User.objects.get(username="janedoe")
[docs] def test_leaderboard_deletion(self):
current_site = Site.objects.get_current()
for prefix in ["", "/platform"]:
with self.subTest(url_prefix=prefix):
search = Search.objects.create(author=self.johndoe, name="test")
search.filters = "{}"
search.settings = "{}"
search.save()
leaderboard = Leaderboard.objects.create(search=search)
leaderboard.notify.add(self.johndoe, self.jackdoe, self.janedoe)
leaderboard.save()
with override_settings(URL_PREFIX=prefix):
reload_urlconf()
mail.outbox = []
leaderboard.delete()
self.assertEqual(len(mail.outbox), 1)
self.assertTrue(
f"https://{current_site.domain}{prefix}" in mail.outbox[0].body
)