190 lines
7.6 KiB
Python
190 lines
7.6 KiB
Python
"""
|
|
Test models, managers, and validators.
|
|
"""
|
|
|
|
from __future__ import absolute_import, division, print_function, unicode_literals
|
|
|
|
from django.core.exceptions import ValidationError
|
|
from django.test import TestCase
|
|
from opaque_keys.edx.keys import UsageKey, CourseKey
|
|
|
|
from student.tests.factories import UserFactory, CourseEnrollmentFactory
|
|
|
|
from .. import models
|
|
from .. import waffle
|
|
|
|
|
|
class PercentValidatorTestCase(TestCase):
|
|
"""
|
|
Test that validate_percent only allows floats (and ints) between 0.0 and 1.0.
|
|
"""
|
|
def test_valid_percents(self):
|
|
for value in [1.0, 0.0, 1, 0, 0.5, 0.333081348071397813987230871]:
|
|
models.validate_percent(value)
|
|
|
|
def test_invalid_percent(self):
|
|
for value in [-0.00000000001, 1.0000000001, 47.1, 1000, None, float('inf'), float('nan')]:
|
|
self.assertRaises(ValidationError, models.validate_percent, value)
|
|
|
|
|
|
class CompletionSetUpMixin(object):
|
|
def set_up_completion(self):
|
|
self.user = UserFactory()
|
|
self.block_key = UsageKey.from_string(u'block-v1:edx+test+run+type@video+block@doggos')
|
|
self.completion = models.BlockCompletion.objects.create(
|
|
user=self.user,
|
|
course_key=self.block_key.course_key,
|
|
block_type=self.block_key.block_type,
|
|
block_key=self.block_key,
|
|
completion=0.5,
|
|
)
|
|
|
|
|
|
class SubmitCompletionTestCase(CompletionSetUpMixin, TestCase):
|
|
"""
|
|
Test that BlockCompletion.objects.submit_completion has the desired
|
|
semantics.
|
|
"""
|
|
def setUp(self):
|
|
super(SubmitCompletionTestCase, self).setUp()
|
|
_overrider = waffle.waffle().override(waffle.ENABLE_COMPLETION_TRACKING, True)
|
|
_overrider.__enter__()
|
|
self.addCleanup(_overrider.__exit__, None, None, None)
|
|
self.set_up_completion()
|
|
|
|
def test_changed_value(self):
|
|
with self.assertNumQueries(4): # Get, update, 2 * savepoints
|
|
completion, isnew = models.BlockCompletion.objects.submit_completion(
|
|
user=self.user,
|
|
course_key=self.block_key.course_key,
|
|
block_key=self.block_key,
|
|
completion=0.9,
|
|
)
|
|
completion.refresh_from_db()
|
|
self.assertEqual(completion.completion, 0.9)
|
|
self.assertFalse(isnew)
|
|
self.assertEqual(models.BlockCompletion.objects.count(), 1)
|
|
|
|
def test_unchanged_value(self):
|
|
with self.assertNumQueries(1): # Get
|
|
completion, isnew = models.BlockCompletion.objects.submit_completion(
|
|
user=self.user,
|
|
course_key=self.block_key.course_key,
|
|
block_key=self.block_key,
|
|
completion=0.5,
|
|
)
|
|
completion.refresh_from_db()
|
|
self.assertEqual(completion.completion, 0.5)
|
|
self.assertFalse(isnew)
|
|
self.assertEqual(models.BlockCompletion.objects.count(), 1)
|
|
|
|
def test_new_user(self):
|
|
newuser = UserFactory()
|
|
with self.assertNumQueries(4): # Get, update, 2 * savepoints
|
|
_, isnew = models.BlockCompletion.objects.submit_completion(
|
|
user=newuser,
|
|
course_key=self.block_key.course_key,
|
|
block_key=self.block_key,
|
|
completion=0.0,
|
|
)
|
|
self.assertTrue(isnew)
|
|
self.assertEqual(models.BlockCompletion.objects.count(), 2)
|
|
|
|
def test_new_block(self):
|
|
newblock = UsageKey.from_string(u'block-v1:edx+test+run+type@video+block@puppers')
|
|
with self.assertNumQueries(4): # Get, update, 2 * savepoints
|
|
_, isnew = models.BlockCompletion.objects.submit_completion(
|
|
user=self.user,
|
|
course_key=newblock.course_key,
|
|
block_key=newblock,
|
|
completion=1.0,
|
|
)
|
|
self.assertTrue(isnew)
|
|
self.assertEqual(models.BlockCompletion.objects.count(), 2)
|
|
|
|
def test_invalid_completion(self):
|
|
with self.assertRaises(ValidationError):
|
|
models.BlockCompletion.objects.submit_completion(
|
|
user=self.user,
|
|
course_key=self.block_key.course_key,
|
|
block_key=self.block_key,
|
|
completion=1.2
|
|
)
|
|
completion = models.BlockCompletion.objects.get(user=self.user, block_key=self.block_key)
|
|
self.assertEqual(completion.completion, 0.5)
|
|
self.assertEqual(models.BlockCompletion.objects.count(), 1)
|
|
|
|
|
|
class CompletionDisabledTestCase(CompletionSetUpMixin, TestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super(CompletionDisabledTestCase, cls).setUpClass()
|
|
cls.overrider = waffle.waffle().override(waffle.ENABLE_COMPLETION_TRACKING, False)
|
|
cls.overrider.__enter__()
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
cls.overrider.__exit__(None, None, None)
|
|
super(CompletionDisabledTestCase, cls).tearDownClass()
|
|
|
|
def setUp(self):
|
|
super(CompletionDisabledTestCase, self).setUp()
|
|
self.set_up_completion()
|
|
|
|
def test_cannot_call_submit_completion(self):
|
|
self.assertEqual(models.BlockCompletion.objects.count(), 1)
|
|
with self.assertRaises(RuntimeError):
|
|
models.BlockCompletion.objects.submit_completion(
|
|
user=self.user,
|
|
course_key=self.block_key.course_key,
|
|
block_key=self.block_key,
|
|
completion=0.9,
|
|
)
|
|
self.assertEqual(models.BlockCompletion.objects.count(), 1)
|
|
|
|
|
|
class SubmitBatchCompletionTestCase(TestCase):
|
|
"""
|
|
Test that BlockCompletion.objects.submit_batch_completion has the desired
|
|
semantics.
|
|
"""
|
|
|
|
def setUp(self):
|
|
super(SubmitBatchCompletionTestCase, self).setUp()
|
|
_overrider = waffle.waffle().override(waffle.ENABLE_COMPLETION_TRACKING, True)
|
|
_overrider.__enter__()
|
|
self.addCleanup(_overrider.__exit__, None, None, None)
|
|
|
|
self.block_key = UsageKey.from_string('block-v1:edx+test+run+type@video+block@doggos')
|
|
self.course_key_obj = CourseKey.from_string('course-v1:edx+test+run')
|
|
self.user = UserFactory()
|
|
CourseEnrollmentFactory.create(user=self.user, course_id=unicode(self.course_key_obj))
|
|
|
|
def test_submit_batch_completion(self):
|
|
blocks = [(self.block_key, 1.0)]
|
|
models.BlockCompletion.objects.submit_batch_completion(self.user, self.course_key_obj, blocks)
|
|
self.assertEqual(models.BlockCompletion.objects.count(), 1)
|
|
self.assertEqual(models.BlockCompletion.objects.last().completion, 1.0)
|
|
|
|
def test_submit_batch_completion_without_waffle(self):
|
|
with waffle.waffle().override(waffle.ENABLE_COMPLETION_TRACKING, False):
|
|
with self.assertRaises(RuntimeError):
|
|
blocks = [(self.block_key, 1.0)]
|
|
models.BlockCompletion.objects.submit_batch_completion(self.user, self.course_key_obj, blocks)
|
|
|
|
def test_submit_batch_completion_with_same_block_new_completion_value(self):
|
|
blocks = [(self.block_key, 0.0)]
|
|
self.assertEqual(models.BlockCompletion.objects.count(), 0)
|
|
models.BlockCompletion.objects.submit_batch_completion(self.user, self.course_key_obj, blocks)
|
|
self.assertEqual(models.BlockCompletion.objects.count(), 1)
|
|
model = models.BlockCompletion.objects.first()
|
|
self.assertEqual(model.completion, 0.0)
|
|
blocks = [
|
|
(UsageKey.from_string('block-v1:edx+test+run+type@video+block@doggos'), 1.0),
|
|
]
|
|
models.BlockCompletion.objects.submit_batch_completion(self.user, self.course_key_obj, blocks)
|
|
self.assertEqual(models.BlockCompletion.objects.count(), 1)
|
|
model = models.BlockCompletion.objects.first()
|
|
self.assertEqual(model.completion, 1.0)
|