92 lines
3.6 KiB
Python
92 lines
3.6 KiB
Python
import re
|
|
import unittest
|
|
from io import StringIO
|
|
from unittest.mock import patch
|
|
|
|
from tests.helpers import TestContext, temporary_root
|
|
from tutor import config as tutor_config
|
|
from tutor import jobs
|
|
|
|
|
|
class JobsTests(unittest.TestCase):
|
|
@patch("sys.stdout", new_callable=StringIO)
|
|
def test_initialise(self, mock_stdout: StringIO) -> None:
|
|
with temporary_root() as root:
|
|
context = TestContext(root)
|
|
config = tutor_config.load_full(root)
|
|
runner = context.job_runner(config)
|
|
jobs.initialise(runner)
|
|
|
|
output = mock_stdout.getvalue().strip()
|
|
service = re.search(r"Service: (\w*)", output)
|
|
commands = re.search(r"(-----)([\S\s]+)(-----)", output)
|
|
assert service is not None
|
|
assert commands is not None
|
|
self.assertTrue(output.startswith("Initialising all services..."))
|
|
self.assertTrue(output.endswith("All services initialised."))
|
|
self.assertEqual(service.group(1), "mysql")
|
|
self.assertTrue(commands.group(2))
|
|
|
|
def test_create_user_command_without_staff(self) -> None:
|
|
command = jobs.create_user_command("superuser", False, "username", "email")
|
|
self.assertNotIn("--staff", command)
|
|
|
|
def test_create_user_command_with_staff(self) -> None:
|
|
command = jobs.create_user_command("superuser", True, "username", "email")
|
|
self.assertIn("--staff", command)
|
|
|
|
def test_create_user_command_with_staff_with_password(self) -> None:
|
|
command = jobs.create_user_command(
|
|
"superuser", True, "username", "email", "command"
|
|
)
|
|
self.assertIn("set_password", command)
|
|
|
|
@patch("sys.stdout", new_callable=StringIO)
|
|
def test_import_demo_course(self, mock_stdout: StringIO) -> None:
|
|
with temporary_root() as root:
|
|
context = TestContext(root)
|
|
config = tutor_config.load_full(root)
|
|
runner = context.job_runner(config)
|
|
jobs.import_demo_course(runner)
|
|
|
|
output = mock_stdout.getvalue()
|
|
service = re.search(r"Service: (\w*)", output)
|
|
commands = re.search(r"(-----)([\S\s]+)(-----)", output)
|
|
assert service is not None
|
|
assert commands is not None
|
|
self.assertEqual(service.group(1), "cms")
|
|
self.assertTrue(
|
|
commands.group(2)
|
|
.strip()
|
|
.startswith('echo "Loading settings $DJANGO_SETTINGS_MODULE"')
|
|
)
|
|
|
|
@patch("sys.stdout", new_callable=StringIO)
|
|
def test_set_theme(self, mock_stdout: StringIO) -> None:
|
|
with temporary_root() as root:
|
|
context = TestContext(root)
|
|
config = tutor_config.load_full(root)
|
|
runner = context.job_runner(config)
|
|
jobs.set_theme("sample_theme", ["domain1", "domain2"], runner)
|
|
|
|
output = mock_stdout.getvalue()
|
|
service = re.search(r"Service: (\w*)", output)
|
|
commands = re.search(r"(-----)([\S\s]+)(-----)", output)
|
|
assert service is not None
|
|
assert commands is not None
|
|
self.assertEqual(service.group(1), "lms")
|
|
self.assertTrue(
|
|
commands.group(2)
|
|
.strip()
|
|
.startswith(
|
|
"export DJANGO_SETTINGS_MODULE=$SERVICE_VARIANT.envs.$SETTINGS"
|
|
)
|
|
)
|
|
|
|
def test_get_all_openedx_domains(self) -> None:
|
|
with temporary_root() as root:
|
|
config = tutor_config.load_full(root)
|
|
domains = jobs.get_all_openedx_domains(config)
|
|
self.assertTrue(domains)
|
|
self.assertEqual(6, len(domains))
|