The existing pattern of using `override_settings(MODULESTORE=...)` prevented
us from having more than one layer of subclassing in modulestore tests.
In a structure like:
@override_settings(MODULESTORE=store_a)
class BaseTestCase(ModuleStoreTestCase):
def setUp(self):
# use store
@override_settings(MODULESTORE=store_b)
class ChildTestCase(BaseTestCase):
def setUp(self):
# use store
In this case, the store actions performed in `BaseTestCase` on behalf of
`ChildTestCase` would still use `store_a`, even though the `ChildTestCase`
had specified to use `store_b`. This is because the `override_settings`
decorator would be the innermost wrapper around the `BaseTestCase.setUp` method,
no matter what `ChildTestCase` does.
To remedy this, we move the call to `override_settings` into the
`ModuleStoreTestCase.setUp` method, and use a cleanup to remove the override.
Subclasses can just defined the `MODULESTORE` class attribute to specify which
modulestore to use _for the entire `setUp` chain_.
[PLAT-419]
115 lines
4.6 KiB
Python
115 lines
4.6 KiB
Python
"""
|
|
Tests for support dashboard
|
|
"""
|
|
import datetime
|
|
|
|
from django.contrib.auth.models import Permission
|
|
from django.test.client import Client
|
|
from django.test.utils import override_settings
|
|
|
|
from xmodule.modulestore.tests.django_utils import TEST_DATA_MOCK_MODULESTORE
|
|
from course_modes.models import CourseMode
|
|
from shoppingcart.models import CertificateItem, Order
|
|
from student.models import CourseEnrollment
|
|
from student.tests.factories import UserFactory
|
|
from xmodule.modulestore.tests.factories import CourseFactory
|
|
from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase
|
|
|
|
|
|
class RefundTests(ModuleStoreTestCase):
|
|
"""
|
|
Tests for the manual refund page
|
|
"""
|
|
def setUp(self):
|
|
super(RefundTests, self).setUp()
|
|
|
|
self.course = CourseFactory.create(
|
|
org='testorg', number='run1', display_name='refundable course'
|
|
)
|
|
self.course_id = self.course.location.course_key
|
|
self.client = Client()
|
|
self.admin = UserFactory.create(
|
|
username='test_admin',
|
|
email='test_admin+support@edx.org',
|
|
password='foo'
|
|
)
|
|
self.admin.user_permissions.add(Permission.objects.get(codename='change_courseenrollment'))
|
|
self.client.login(username=self.admin.username, password='foo')
|
|
self.student = UserFactory.create(
|
|
username='student',
|
|
email='student+refund@edx.org'
|
|
)
|
|
self.course_mode = CourseMode.objects.get_or_create(course_id=self.course_id, mode_slug='verified')[0]
|
|
|
|
self.order = None
|
|
self.form_pars = {'course_id': str(self.course_id), 'user': self.student.email}
|
|
|
|
def tearDown(self):
|
|
self.course_mode.delete()
|
|
Order.objects.filter(user=self.student).delete()
|
|
|
|
def _enroll(self, purchase=True):
|
|
# pylint: disable=missing-docstring
|
|
CourseEnrollment.enroll(self.student, self.course_id, self.course_mode.mode_slug)
|
|
if purchase:
|
|
self.order = Order.get_cart_for_user(self.student)
|
|
CertificateItem.add_to_order(self.order, self.course_id, 1, self.course_mode.mode_slug)
|
|
self.order.purchase()
|
|
self.course_mode.expiration_datetime = datetime.datetime(1983, 4, 6)
|
|
self.course_mode.save()
|
|
|
|
def test_support_access(self):
|
|
response = self.client.get('/support/')
|
|
self.assertTrue(response.status_code, 200)
|
|
self.assertContains(response, 'Manual Refund')
|
|
response = self.client.get('/support/refund/')
|
|
self.assertTrue(response.status_code, 200)
|
|
|
|
# users without the permission can't access support
|
|
self.admin.user_permissions.clear()
|
|
response = self.client.get('/support/')
|
|
self.assertTrue(response.status_code, 302)
|
|
|
|
response = self.client.get('/support/refund/')
|
|
self.assertTrue(response.status_code, 302)
|
|
|
|
def test_bad_courseid(self):
|
|
response = self.client.post('/support/refund/', {'course_id': 'foo', 'user': self.student.email})
|
|
self.assertContains(response, 'Invalid course id')
|
|
|
|
def test_bad_user(self):
|
|
response = self.client.post('/support/refund/', {'course_id': str(self.course_id), 'user': 'unknown@foo.com'})
|
|
self.assertContains(response, 'User not found')
|
|
|
|
def test_not_refundable(self):
|
|
self._enroll()
|
|
self.course_mode.expiration_datetime = datetime.datetime(2033, 4, 6)
|
|
self.course_mode.save()
|
|
response = self.client.post('/support/refund/', self.form_pars)
|
|
self.assertContains(response, 'not past the refund window')
|
|
|
|
def test_no_order(self):
|
|
self._enroll(purchase=False)
|
|
response = self.client.post('/support/refund/', self.form_pars)
|
|
self.assertContains(response, 'No order found for %s' % self.student.username)
|
|
|
|
def test_valid_order(self):
|
|
self._enroll()
|
|
response = self.client.post('/support/refund/', self.form_pars)
|
|
self.assertContains(response, "About to refund this order")
|
|
self.assertContains(response, "enrolled")
|
|
self.assertContains(response, "CertificateItem Status")
|
|
|
|
def test_do_refund(self):
|
|
self._enroll()
|
|
pars = self.form_pars
|
|
pars['confirmed'] = 'true'
|
|
response = self.client.post('/support/refund/', pars)
|
|
self.assertTrue(response.status_code, 302)
|
|
response = self.client.get(response.get('location')) # pylint: disable=maybe-no-member
|
|
|
|
self.assertContains(response, "Unenrolled %s from" % self.student)
|
|
self.assertContains(response, "Refunded 1 for order id")
|
|
|
|
self.assertFalse(CourseEnrollment.is_enrolled(self.student, self.course_id))
|