Commit ea325298 by Nimisha Asthagiri

Fix all modulestore calls to pass in user ids.

parent f5abef88
...@@ -6,9 +6,9 @@ from django.core.management.base import BaseCommand, CommandError, make_option ...@@ -6,9 +6,9 @@ from django.core.management.base import BaseCommand, CommandError, make_option
from django_comment_common.utils import (seed_permissions_roles, from django_comment_common.utils import (seed_permissions_roles,
are_permissions_roles_seeded) are_permissions_roles_seeded)
from xmodule.modulestore.xml_importer import import_from_xml from xmodule.modulestore.xml_importer import import_from_xml
from xmodule.modulestore import ModuleStoreEnum
from xmodule.modulestore.django import modulestore from xmodule.modulestore.django import modulestore
from xmodule.contentstore.django import contentstore from xmodule.contentstore.django import contentstore
from xmodule.modulestore import ModuleStoreEnum
class Command(BaseCommand): class Command(BaseCommand):
...@@ -38,9 +38,10 @@ class Command(BaseCommand): ...@@ -38,9 +38,10 @@ class Command(BaseCommand):
data=data_dir, data=data_dir,
courses=course_dirs, courses=course_dirs,
dis=do_import_static)) dis=do_import_static))
mstore = modulestore()
_, course_items = import_from_xml( _, course_items = import_from_xml(
modulestore(), ModuleStoreEnum.UserID.mgmt_command, data_dir, course_dirs, load_error_modules=False, mstore, ModuleStoreEnum.UserID.mgmt_command, data_dir, course_dirs, load_error_modules=False,
static_content_store=contentstore(), verbose=True, static_content_store=contentstore(), verbose=True,
do_import_static=do_import_static, do_import_static=do_import_static,
create_new_course_if_not_present=True, create_new_course_if_not_present=True,
......
...@@ -187,11 +187,11 @@ class TemplateTests(unittest.TestCase): ...@@ -187,11 +187,11 @@ class TemplateTests(unittest.TestCase):
) )
first_problem.max_attempts = 3 first_problem.max_attempts = 3
first_problem.save() # decache the above into the kvs first_problem.save() # decache the above into the kvs
updated_problem = self.split_store.update_item(first_problem, '**replace_user**') updated_problem = self.split_store.update_item(first_problem, ModuleStoreEnum.UserID.test)
self.assertIsNotNone(updated_problem.previous_version) self.assertIsNotNone(updated_problem.previous_version)
self.assertEqual(updated_problem.previous_version, first_problem.update_version) self.assertEqual(updated_problem.previous_version, first_problem.update_version)
self.assertNotEqual(updated_problem.update_version, first_problem.update_version) self.assertNotEqual(updated_problem.update_version, first_problem.update_version)
updated_loc = self.split_store.delete_item(updated_problem.location, '**replace_user**', 'testbot') updated_loc = self.split_store.delete_item(updated_problem.location, ModuleStoreEnum.UserID.test, 'testbot')
second_problem = persistent_factories.ItemFactory.create( second_problem = persistent_factories.ItemFactory.create(
display_name='problem 2', display_name='problem 2',
......
...@@ -65,7 +65,7 @@ class TestExportGit(CourseTestCase): ...@@ -65,7 +65,7 @@ class TestExportGit(CourseTestCase):
Test failed course export response. Test failed course export response.
""" """
self.course_module.giturl = 'foobar' self.course_module.giturl = 'foobar'
modulestore().update_item(self.course_module, '**replace_user**') modulestore().update_item(self.course_module, self.user.id)
response = self.client.get('{}?action=push'.format(self.test_url)) response = self.client.get('{}?action=push'.format(self.test_url))
self.assertIn('Export Failed:', response.content) self.assertIn('Export Failed:', response.content)
...@@ -75,7 +75,7 @@ class TestExportGit(CourseTestCase): ...@@ -75,7 +75,7 @@ class TestExportGit(CourseTestCase):
Regression test for making sure errors are properly stringified Regression test for making sure errors are properly stringified
""" """
self.course_module.giturl = 'foobar' self.course_module.giturl = 'foobar'
modulestore().update_item(self.course_module, '**replace_user**') modulestore().update_item(self.course_module, self.user.id)
response = self.client.get('{}?action=push'.format(self.test_url)) response = self.client.get('{}?action=push'.format(self.test_url))
self.assertNotIn('django.utils.functional.__proxy__', response.content) self.assertNotIn('django.utils.functional.__proxy__', response.content)
...@@ -98,7 +98,7 @@ class TestExportGit(CourseTestCase): ...@@ -98,7 +98,7 @@ class TestExportGit(CourseTestCase):
self.populate_course() self.populate_course()
self.course_module.giturl = 'file://{}'.format(bare_repo_dir) self.course_module.giturl = 'file://{}'.format(bare_repo_dir)
modulestore().update_item(self.course_module, '**replace_user**') modulestore().update_item(self.course_module, self.user.id)
response = self.client.get('{}?action=push'.format(self.test_url)) response = self.client.get('{}?action=push'.format(self.test_url))
self.assertIn('Export Succeeded', response.content) self.assertIn('Export Succeeded', response.content)
...@@ -33,25 +33,10 @@ class ContentStoreImportTest(ModuleStoreTestCase): ...@@ -33,25 +33,10 @@ class ContentStoreImportTest(ModuleStoreTestCase):
NOTE: refactor using CourseFactory so they do not. NOTE: refactor using CourseFactory so they do not.
""" """
def setUp(self): def setUp(self):
password = super(ContentStoreImportTest, self).setUp()
uname = 'testuser'
email = 'test+courses@edx.org'
password = 'foo'
# Create the use so we can log them in.
self.user = User.objects.create_user(uname, email, password)
# Note that we do not actually need to do anything
# for registration if we directly mark them active.
self.user.is_active = True
# Staff has access to view all courses
self.user.is_staff = True
# Save the data that we've just changed to the db.
self.user.save()
self.client = Client() self.client = Client()
self.client.login(username=uname, password=password) self.client.login(username=self.user.username, password=password)
def tearDown(self): def tearDown(self):
contentstore().drop_database() contentstore().drop_database()
...@@ -66,7 +51,7 @@ class ContentStoreImportTest(ModuleStoreTestCase): ...@@ -66,7 +51,7 @@ class ContentStoreImportTest(ModuleStoreTestCase):
module_store = modulestore() module_store = modulestore()
import_from_xml( import_from_xml(
module_store, module_store,
'**replace_user**', self.user.id,
'common/test/data/', 'common/test/data/',
['test_import_course'], ['test_import_course'],
static_content_store=content_store, static_content_store=content_store,
...@@ -86,7 +71,7 @@ class ContentStoreImportTest(ModuleStoreTestCase): ...@@ -86,7 +71,7 @@ class ContentStoreImportTest(ModuleStoreTestCase):
module_store, __, course = self.load_test_import_course() module_store, __, course = self.load_test_import_course()
__, course_items = import_from_xml( __, course_items = import_from_xml(
module_store, module_store,
'**replace_user**', self.user.id,
'common/test/data', 'common/test/data',
['test_import_course_2'], ['test_import_course_2'],
target_course_id=course.id, target_course_id=course.id,
...@@ -102,7 +87,7 @@ class ContentStoreImportTest(ModuleStoreTestCase): ...@@ -102,7 +87,7 @@ class ContentStoreImportTest(ModuleStoreTestCase):
course_id = SlashSeparatedCourseKey(u'Юникода', u'unicode_course', u'échantillon') course_id = SlashSeparatedCourseKey(u'Юникода', u'unicode_course', u'échantillon')
import_from_xml( import_from_xml(
module_store, module_store,
'**replace_user**', self.user.id,
'common/test/data/', 'common/test/data/',
['2014_Uni'], ['2014_Uni'],
target_course_id=course_id target_course_id=course_id
...@@ -148,7 +133,7 @@ class ContentStoreImportTest(ModuleStoreTestCase): ...@@ -148,7 +133,7 @@ class ContentStoreImportTest(ModuleStoreTestCase):
content_store = contentstore() content_store = contentstore()
module_store = modulestore() module_store = modulestore()
import_from_xml(module_store, '**replace_user**', 'common/test/data/', ['toy'], static_content_store=content_store, do_import_static=False, verbose=True) import_from_xml(module_store, self.user.id, 'common/test/data/', ['toy'], static_content_store=content_store, do_import_static=False, verbose=True)
course = module_store.get_course(SlashSeparatedCourseKey('edX', 'toy', '2012_Fall')) course = module_store.get_course(SlashSeparatedCourseKey('edX', 'toy', '2012_Fall'))
...@@ -159,7 +144,7 @@ class ContentStoreImportTest(ModuleStoreTestCase): ...@@ -159,7 +144,7 @@ class ContentStoreImportTest(ModuleStoreTestCase):
def test_no_static_link_rewrites_on_import(self): def test_no_static_link_rewrites_on_import(self):
module_store = modulestore() module_store = modulestore()
_, courses = import_from_xml(module_store, '**replace_user**', 'common/test/data/', ['toy'], do_import_static=False, verbose=True) _, courses = import_from_xml(module_store, self.user.id, 'common/test/data/', ['toy'], do_import_static=False, verbose=True)
course_key = courses[0].id course_key = courses[0].id
handouts = module_store.get_item(course_key.make_usage_key('course_info', 'handouts')) handouts = module_store.get_item(course_key.make_usage_key('course_info', 'handouts'))
...@@ -178,7 +163,7 @@ class ContentStoreImportTest(ModuleStoreTestCase): ...@@ -178,7 +163,7 @@ class ContentStoreImportTest(ModuleStoreTestCase):
target_course_id = SlashSeparatedCourseKey('testX', 'conditional_copy', 'copy_run') target_course_id = SlashSeparatedCourseKey('testX', 'conditional_copy', 'copy_run')
import_from_xml( import_from_xml(
module_store, module_store,
'**replace_user**', self.user.id,
'common/test/data/', 'common/test/data/',
['conditional'], ['conditional'],
target_course_id=target_course_id target_course_id=target_course_id
...@@ -208,7 +193,7 @@ class ContentStoreImportTest(ModuleStoreTestCase): ...@@ -208,7 +193,7 @@ class ContentStoreImportTest(ModuleStoreTestCase):
target_course_id = SlashSeparatedCourseKey('testX', 'peergrading_copy', 'copy_run') target_course_id = SlashSeparatedCourseKey('testX', 'peergrading_copy', 'copy_run')
import_from_xml( import_from_xml(
module_store, module_store,
'**replace_user**', self.user.id,
'common/test/data/', 'common/test/data/',
['open_ended'], ['open_ended'],
target_course_id=target_course_id target_course_id=target_course_id
...@@ -227,7 +212,7 @@ class ContentStoreImportTest(ModuleStoreTestCase): ...@@ -227,7 +212,7 @@ class ContentStoreImportTest(ModuleStoreTestCase):
target_course_id = SlashSeparatedCourseKey('testX', 'split_test_copy', 'copy_run') target_course_id = SlashSeparatedCourseKey('testX', 'split_test_copy', 'copy_run')
import_from_xml( import_from_xml(
module_store, module_store,
'**replace_user**', self.user.id,
'common/test/data/', 'common/test/data/',
['split_test_module'], ['split_test_module'],
target_course_id=target_course_id target_course_id=target_course_id
......
...@@ -10,7 +10,7 @@ class DraftReorderTestCase(ModuleStoreTestCase): ...@@ -10,7 +10,7 @@ class DraftReorderTestCase(ModuleStoreTestCase):
def test_order(self): def test_order(self):
store = modulestore() store = modulestore()
_, course_items = import_from_xml(store, '**replace_user**', 'common/test/data/', ['import_draft_order']) _, course_items = import_from_xml(store, self.user.id, 'common/test/data/', ['import_draft_order'])
course_key = course_items[0].id course_key = course_items[0].id
sequential = store.get_item(course_key.make_usage_key('sequential', '0f4f7649b10141b0bdc9922dcf94515a')) sequential = store.get_item(course_key.make_usage_key('sequential', '0f4f7649b10141b0bdc9922dcf94515a'))
verticals = sequential.children verticals = sequential.children
......
...@@ -28,9 +28,6 @@ class StubXBlock(XBlock): ...@@ -28,9 +28,6 @@ class StubXBlock(XBlock):
class XBlockImportTest(ModuleStoreTestCase): class XBlockImportTest(ModuleStoreTestCase):
def setUp(self):
self.store = modulestore()
@XBlock.register_temp_plugin(StubXBlock) @XBlock.register_temp_plugin(StubXBlock)
def test_import_public(self): def test_import_public(self):
self._assert_import( self._assert_import(
...@@ -62,7 +59,7 @@ class XBlockImportTest(ModuleStoreTestCase): ...@@ -62,7 +59,7 @@ class XBlockImportTest(ModuleStoreTestCase):
""" """
_, courses = import_from_xml( _, courses = import_from_xml(
self.store, '**replace_user**', 'common/test/data', [course_dir] self.store, self.user.id, 'common/test/data', [course_dir]
) )
xblock_location = courses[0].id.make_usage_key('stubxblock', 'xblock_test') xblock_location = courses[0].id.make_usage_key('stubxblock', 'xblock_test')
......
...@@ -24,23 +24,10 @@ class TestCourseAccess(ModuleStoreTestCase): ...@@ -24,23 +24,10 @@ class TestCourseAccess(ModuleStoreTestCase):
Create a pool of users w/o granting them any permissions Create a pool of users w/o granting them any permissions
""" """
super(TestCourseAccess, self).setUp() user_password = super(TestCourseAccess, self).setUp()
uname = 'testuser'
email = 'test+courses@edx.org'
password = 'foo'
# Create the use so we can log them in.
self.user = User.objects.create_user(uname, email, password)
# Note that we do not actually need to do anything
# for registration if we directly mark them active.
self.user.is_active = True
# Staff has access to view all courses
self.user.is_staff = True
self.user.save()
self.client = AjaxEnabledTestClient() self.client = AjaxEnabledTestClient()
self.client.login(username=uname, password=password) self.client.login(username=self.user.username, password=user_password)
# create a course via the view handler which has a different strategy for permissions than the factory # create a course via the view handler which has a different strategy for permissions than the factory
self.course_key = SlashSeparatedCourseKey('myu', 'mydept.mycourse', 'myrun') self.course_key = SlashSeparatedCourseKey('myu', 'mydept.mycourse', 'myrun')
......
...@@ -9,6 +9,7 @@ from django.test import TestCase ...@@ -9,6 +9,7 @@ from django.test import TestCase
from django.test.utils import override_settings from django.test.utils import override_settings
from contentstore import utils from contentstore import utils
from xmodule.modulestore import ModuleStoreEnum
from xmodule.modulestore.tests.factories import CourseFactory from xmodule.modulestore.tests.factories import CourseFactory
from opaque_keys.edx.locations import SlashSeparatedCourseKey, Location from opaque_keys.edx.locations import SlashSeparatedCourseKey, Location
...@@ -193,7 +194,7 @@ class XBlockVisibilityTestCase(TestCase): ...@@ -193,7 +194,7 @@ class XBlockVisibilityTestCase(TestCase):
"""Tests for xblock visibility for students.""" """Tests for xblock visibility for students."""
def setUp(self): def setUp(self):
self.dummy_user = 123 self.dummy_user = ModuleStoreEnum.UserID.test
self.past = datetime(1970, 1, 1) self.past = datetime(1970, 1, 1)
self.future = datetime.now(UTC) + timedelta(days=1) self.future = datetime.now(UTC) + timedelta(days=1)
......
...@@ -6,8 +6,8 @@ Utilities for contentstore tests ...@@ -6,8 +6,8 @@ Utilities for contentstore tests
import json import json
import re import re
from django.contrib.auth.models import User
from django.test.client import Client from django.test.client import Client
from django.contrib.auth.models import User
from xmodule.contentstore.django import contentstore from xmodule.contentstore.django import contentstore
from xmodule.contentstore.content import StaticContent from xmodule.contentstore.content import StaticContent
...@@ -68,53 +68,32 @@ class AjaxEnabledTestClient(Client): ...@@ -68,53 +68,32 @@ class AjaxEnabledTestClient(Client):
class CourseTestCase(ModuleStoreTestCase): class CourseTestCase(ModuleStoreTestCase):
def setUp(self): def setUp(self):
""" """
These tests need a user in the DB so that the django Test Client These tests need a user in the DB so that the django Test Client can log them in.
can log them in. The test user is created in the ModuleStoreTestCase setUp method.
They inherit from the ModuleStoreTestCase class so that the mongodb collection They inherit from the ModuleStoreTestCase class so that the mongodb collection
will be cleared out before each test case execution and deleted will be cleared out before each test case execution and deleted
afterwards. afterwards.
""" """
uname = 'testuser' user_password = super(CourseTestCase, self).setUp()
email = 'test+courses@edx.org'
password = 'foo'
# Create the user so we can log them in.
self.user = User.objects.create_user(uname, email, password)
# Note that we do not actually need to do anything
# for registration if we directly mark them active.
self.user.is_active = True
# Staff has access to view all courses
self.user.is_staff = True
self.user.save()
self.client = AjaxEnabledTestClient() self.client = AjaxEnabledTestClient()
self.client.login(username=uname, password=password) self.client.login(username=self.user.username, password=user_password)
self.course = CourseFactory.create( self.course = CourseFactory.create(
org='MITx', org='MITx',
number='999', number='999',
display_name='Robot Super Course', display_name='Robot Super Course',
) )
self.store = modulestore()
def create_non_staff_authed_user_client(self, authenticate=True): def create_non_staff_authed_user_client(self, authenticate=True):
""" """
Create a non-staff user, log them in (if authenticate=True), and return the client, user to use for testing. Create a non-staff user, log them in (if authenticate=True), and return the client, user to use for testing.
""" """
uname = 'teststudent' nonstaff, password = super(CourseTestCase, self).create_non_staff_authed_user_client()
password = 'foo'
nonstaff = User.objects.create_user(uname, 'test+student@edx.org', password)
# Note that we do not actually need to do anything
# for registration if we directly mark them active.
nonstaff.is_active = True
nonstaff.is_staff = False
nonstaff.save()
client = Client() client = Client()
if authenticate: if authenticate:
client.login(username=uname, password=password) client.login(username=nonstaff.username, password=password)
return client, nonstaff return client, nonstaff
def populate_course(self): def populate_course(self):
......
...@@ -449,8 +449,8 @@ def component_handler(request, usage_key_string, handler, suffix=''): ...@@ -449,8 +449,8 @@ def component_handler(request, usage_key_string, handler, suffix=''):
log.info("XBlock %s attempted to access missing handler %r", descriptor, handler, exc_info=True) log.info("XBlock %s attempted to access missing handler %r", descriptor, handler, exc_info=True)
raise Http404 raise Http404
# unintentional update to handle any side effects of handle call; so, request user didn't author # unintentional update to handle any side effects of handle call
# the change # could potentially be updating actual course data or simply caching its values
modulestore().update_item(descriptor, None) modulestore().update_item(descriptor, request.user.id)
return webob_to_django_response(resp) return webob_to_django_response(resp)
...@@ -337,6 +337,7 @@ def create_new_course(request): ...@@ -337,6 +337,7 @@ def create_new_course(request):
new_course = modulestore().create_course( new_course = modulestore().create_course(
course_key.org, course_key.org,
course_key.offering, course_key.offering,
request.user.id,
fields=fields, fields=fields,
) )
......
...@@ -140,8 +140,8 @@ def xblock_handler(request, usage_key_string): ...@@ -140,8 +140,8 @@ def xblock_handler(request, usage_key_string):
dest_usage_key = _duplicate_item( dest_usage_key = _duplicate_item(
parent_usage_key, parent_usage_key,
duplicate_source_usage_key, duplicate_source_usage_key,
request.json.get('display_name'),
request.user, request.user,
request.json.get('display_name'),
) )
return JsonResponse({"locator": unicode(dest_usage_key), "courseKey": unicode(dest_usage_key.course_key)}) return JsonResponse({"locator": unicode(dest_usage_key), "courseKey": unicode(dest_usage_key.course_key)})
...@@ -193,8 +193,7 @@ def xblock_view_handler(request, usage_key_string, view_name): ...@@ -193,8 +193,7 @@ def xblock_view_handler(request, usage_key_string, view_name):
log.debug("unable to render studio_view for %r", xblock, exc_info=True) log.debug("unable to render studio_view for %r", xblock, exc_info=True)
fragment = Fragment(render_to_string('html_error.html', {'message': str(exc)})) fragment = Fragment(render_to_string('html_error.html', {'message': str(exc)}))
# change not authored by requestor but by xblocks. store.update_item(xblock, request.user.id)
store.update_item(xblock, None)
elif view_name in (unit_views + container_views): elif view_name in (unit_views + container_views):
is_container_view = (view_name in container_views) is_container_view = (view_name in container_views)
...@@ -432,7 +431,7 @@ def _create_item(request): ...@@ -432,7 +431,7 @@ def _create_item(request):
return JsonResponse({"locator": unicode(created_block.location), "courseKey": unicode(created_block.location.course_key)}) return JsonResponse({"locator": unicode(created_block.location), "courseKey": unicode(created_block.location.course_key)})
def _duplicate_item(parent_usage_key, duplicate_source_usage_key, display_name=None, user=None): def _duplicate_item(parent_usage_key, duplicate_source_usage_key, user, display_name=None):
""" """
Duplicate an existing xblock as a child of the supplied parent_usage_key. Duplicate an existing xblock as a child of the supplied parent_usage_key.
""" """
...@@ -454,6 +453,8 @@ def _duplicate_item(parent_usage_key, duplicate_source_usage_key, display_name=N ...@@ -454,6 +453,8 @@ def _duplicate_item(parent_usage_key, duplicate_source_usage_key, display_name=N
dest_module = store.create_and_save_xmodule( dest_module = store.create_and_save_xmodule(
dest_usage_key, dest_usage_key,
user.id, user.id,
definition_data=source_item.get_explicitly_set_fields_by_scope(Scope.content), definition_data=source_item.get_explicitly_set_fields_by_scope(Scope.content),
metadata=duplicate_metadata, metadata=duplicate_metadata,
...@@ -467,7 +468,7 @@ def _duplicate_item(parent_usage_key, duplicate_source_usage_key, display_name=N ...@@ -467,7 +468,7 @@ def _duplicate_item(parent_usage_key, duplicate_source_usage_key, display_name=N
for child in source_item.children: for child in source_item.children:
dupe = _duplicate_item(dest_module.location, child, user=user) dupe = _duplicate_item(dest_module.location, child, user=user)
dest_module.children.append(dupe) dest_module.children.append(dupe)
store.update_item(dest_module, user.id if user else None) store.update_item(dest_module, user.id)
if not 'detached' in source_item.runtime.load_block_type(category)._class_tags: if not 'detached' in source_item.runtime.load_block_type(category)._class_tags:
parent = store.get_item(parent_usage_key) parent = store.get_item(parent_usage_key)
...@@ -478,7 +479,7 @@ def _duplicate_item(parent_usage_key, duplicate_source_usage_key, display_name=N ...@@ -478,7 +479,7 @@ def _duplicate_item(parent_usage_key, duplicate_source_usage_key, display_name=N
parent.children.insert(source_index + 1, dest_module.location) parent.children.insert(source_index + 1, dest_module.location)
else: else:
parent.children.append(dest_module.location) parent.children.append(dest_module.location)
store.update_item(parent, user.id if user else None) store.update_item(parent, user.id)
return dest_module.location return dest_module.location
......
...@@ -51,7 +51,7 @@ class BasicAssetsTestCase(AssetsTestCase): ...@@ -51,7 +51,7 @@ class BasicAssetsTestCase(AssetsTestCase):
module_store = modulestore() module_store = modulestore()
_, course_items = import_from_xml( _, course_items = import_from_xml(
module_store, module_store,
'**replace_user**', self.user.id,
'common/test/data/', 'common/test/data/',
['toy'], ['toy'],
static_content_store=contentstore(), static_content_store=contentstore(),
...@@ -195,7 +195,7 @@ class LockAssetTestCase(AssetsTestCase): ...@@ -195,7 +195,7 @@ class LockAssetTestCase(AssetsTestCase):
module_store = modulestore() module_store = modulestore()
_, course_items = import_from_xml( _, course_items = import_from_xml(
module_store, module_store,
'**replace_user**', self.user.id,
'common/test/data/', 'common/test/data/',
['toy'], ['toy'],
static_content_store=contentstore(), static_content_store=contentstore(),
......
...@@ -137,7 +137,7 @@ class ContainerPageTestCase(StudioPageTestCase): ...@@ -137,7 +137,7 @@ class ContainerPageTestCase(StudioPageTestCase):
""" """
empty_child_container = ItemFactory.create(parent_location=self.vertical.location, empty_child_container = ItemFactory.create(parent_location=self.vertical.location,
category='split_test', display_name='Split Test') category='split_test', display_name='Split Test')
published_empty_child_container = self.store.publish(empty_child_container.location, '**replace_user**') published_empty_child_container = self.store.publish(empty_child_container.location, self.user.id)
self.validate_preview_html(published_empty_child_container, self.reorderable_child_view, self.validate_preview_html(published_empty_child_container, self.reorderable_child_view,
can_reorder=False, can_edit=False, can_add=False) can_reorder=False, can_edit=False, can_add=False)
......
...@@ -25,10 +25,9 @@ from contentstore.tests.utils import CourseTestCase ...@@ -25,10 +25,9 @@ from contentstore.tests.utils import CourseTestCase
from student.tests.factories import UserFactory from student.tests.factories import UserFactory
from xmodule.capa_module import CapaDescriptor from xmodule.capa_module import CapaDescriptor
from xmodule.modulestore import PublishState from xmodule.modulestore import PublishState
from xmodule.modulestore.django import modulestore
from xmodule.x_module import STUDIO_VIEW, STUDENT_VIEW from xmodule.x_module import STUDIO_VIEW, STUDENT_VIEW
from xblock.exceptions import NoSuchHandlerError from xblock.exceptions import NoSuchHandlerError
from opaque_keys.edx.keys import UsageKey from opaque_keys.edx.keys import UsageKey, CourseKey
from opaque_keys.edx.locations import Location from opaque_keys.edx.locations import Location
from xmodule.partitions.partitions import Group, UserPartition from xmodule.partitions.partitions import Group, UserPartition
...@@ -40,7 +39,6 @@ class ItemTest(CourseTestCase): ...@@ -40,7 +39,6 @@ class ItemTest(CourseTestCase):
self.course_key = self.course.id self.course_key = self.course.id
self.usage_key = self.course.location self.usage_key = self.course.location
self.store = modulestore()
def get_item_from_modulestore(self, usage_key, verify_is_draft=False): def get_item_from_modulestore(self, usage_key, verify_is_draft=False):
""" """
......
...@@ -40,7 +40,7 @@ class UnitPageTestCase(StudioPageTestCase): ...@@ -40,7 +40,7 @@ class UnitPageTestCase(StudioPageTestCase):
""" """
Verify that a public xblock's preview returns the expected HTML. Verify that a public xblock's preview returns the expected HTML.
""" """
published_video = self.store.publish(self.video.location, '**replace_user**') published_video = self.store.publish(self.video.location, self.user.id)
self.validate_preview_html(self.video, STUDENT_VIEW, self.validate_preview_html(self.video, STUDENT_VIEW,
can_edit=True, can_reorder=True, can_add=False) can_edit=True, can_reorder=True, can_add=False)
...@@ -60,7 +60,7 @@ class UnitPageTestCase(StudioPageTestCase): ...@@ -60,7 +60,7 @@ class UnitPageTestCase(StudioPageTestCase):
category='split_test', display_name='Split Test') category='split_test', display_name='Split Test')
ItemFactory.create(parent_location=child_container.location, ItemFactory.create(parent_location=child_container.location,
category='html', display_name='grandchild') category='html', display_name='grandchild')
published_child_container = self.store.publish(child_container.location, '**replace_user**') published_child_container = self.store.publish(child_container.location, self.user.id)
self.validate_preview_html(published_child_container, STUDENT_VIEW, self.validate_preview_html(published_child_container, STUDENT_VIEW,
can_reorder=True, can_edit=True, can_add=False) can_reorder=True, can_edit=True, can_add=False)
......
...@@ -53,7 +53,7 @@ class CourseMetadata(object): ...@@ -53,7 +53,7 @@ class CourseMetadata(object):
return result return result
@classmethod @classmethod
def update_from_json(cls, descriptor, jsondict, filter_tabs=True, user=None): def update_from_json(cls, descriptor, jsondict, user, filter_tabs=True):
""" """
Decode the json into CourseMetadata and save any changed attrs to the db. Decode the json into CourseMetadata and save any changed attrs to the db.
...@@ -84,6 +84,6 @@ class CourseMetadata(object): ...@@ -84,6 +84,6 @@ class CourseMetadata(object):
setattr(descriptor, key, value) setattr(descriptor, key, value)
if len(key_values) > 0: if len(key_values) > 0:
modulestore().update_item(descriptor, user.id if user else None) modulestore().update_item(descriptor, user.id)
return cls.fetch(descriptor) return cls.fetch(descriptor)
...@@ -5,7 +5,6 @@ import copy ...@@ -5,7 +5,6 @@ import copy
import logging import logging
from uuid import uuid4 from uuid import uuid4
from django.contrib.auth.models import User
from django.conf import settings from django.conf import settings
from django.test.client import Client from django.test.client import Client
from django.test.utils import override_settings from django.test.utils import override_settings
...@@ -34,12 +33,16 @@ class ContentStoreToyCourseTest(ModuleStoreTestCase): ...@@ -34,12 +33,16 @@ class ContentStoreToyCourseTest(ModuleStoreTestCase):
""" """
Create user and login. Create user and login.
""" """
self.staff_pwd = super(ContentStoreToyCourseTest, self).setUp()
self.staff_usr = self.user
self.non_staff_usr, self.non_staff_pwd = self.create_non_staff_authed_user_client()
self.client = Client() self.client = Client()
self.contentstore = contentstore() self.contentstore = contentstore()
self.course_key = SlashSeparatedCourseKey('edX', 'toy', '2012_Fall') self.course_key = SlashSeparatedCourseKey('edX', 'toy', '2012_Fall')
import_from_xml(modulestore(), '**replace_user**', 'common/test/data/', ['toy'], import_from_xml(modulestore(), self.user.id, 'common/test/data/', ['toy'],
static_content_store=self.contentstore, verbose=True) static_content_store=self.contentstore, verbose=True)
# A locked asset # A locked asset
...@@ -52,24 +55,6 @@ class ContentStoreToyCourseTest(ModuleStoreTestCase): ...@@ -52,24 +55,6 @@ class ContentStoreToyCourseTest(ModuleStoreTestCase):
self.contentstore.set_attr(self.locked_asset, 'locked', True) self.contentstore.set_attr(self.locked_asset, 'locked', True)
# Create user
self.usr = 'testuser'
self.pwd = 'foo'
email = 'test+courses@edx.org'
self.user = User.objects.create_user(self.usr, email, self.pwd)
self.user.is_active = True
self.user.save()
# Create staff user
self.staff_usr = 'stafftestuser'
self.staff_pwd = 'foo'
staff_email = 'stafftest+courses@edx.org'
self.staff_user = User.objects.create_user(self.staff_usr, staff_email,
self.staff_pwd)
self.staff_user.is_active = True
self.staff_user.is_staff = True
self.staff_user.save()
def tearDown(self): def tearDown(self):
contentstore().drop_database() contentstore().drop_database()
...@@ -97,7 +82,7 @@ class ContentStoreToyCourseTest(ModuleStoreTestCase): ...@@ -97,7 +82,7 @@ class ContentStoreToyCourseTest(ModuleStoreTestCase):
Test that locked assets behave appropriately in case user is logged in Test that locked assets behave appropriately in case user is logged in
in but not registered for the course. in but not registered for the course.
""" """
self.client.login(username=self.usr, password=self.pwd) self.client.login(username=self.non_staff_usr, password=self.non_staff_pwd)
resp = self.client.get(self.url_locked) resp = self.client.get(self.url_locked)
self.assertEqual(resp.status_code, 403) # pylint: disable=E1103 self.assertEqual(resp.status_code, 403) # pylint: disable=E1103
...@@ -106,10 +91,10 @@ class ContentStoreToyCourseTest(ModuleStoreTestCase): ...@@ -106,10 +91,10 @@ class ContentStoreToyCourseTest(ModuleStoreTestCase):
Test that locked assets behave appropriately in case user is logged in Test that locked assets behave appropriately in case user is logged in
and registered for the course. and registered for the course.
""" """
CourseEnrollment.enroll(self.user, self.course_key) CourseEnrollment.enroll(self.non_staff_usr, self.course_key)
self.assertTrue(CourseEnrollment.is_enrolled(self.user, self.course_key)) self.assertTrue(CourseEnrollment.is_enrolled(self.non_staff_usr, self.course_key))
self.client.login(username=self.usr, password=self.pwd) self.client.login(username=self.non_staff_usr, password=self.non_staff_pwd)
resp = self.client.get(self.url_locked) resp = self.client.get(self.url_locked)
self.assertEqual(resp.status_code, 200) # pylint: disable=E1103 self.assertEqual(resp.status_code, 200) # pylint: disable=E1103
......
...@@ -17,7 +17,6 @@ from django.utils.importlib import import_module ...@@ -17,7 +17,6 @@ from django.utils.importlib import import_module
from xmodule.modulestore.tests.factories import CourseFactory from xmodule.modulestore.tests.factories import CourseFactory
from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase, mixed_store_config from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase, mixed_store_config
from xmodule.modulestore.inheritance import own_metadata
from xmodule.modulestore.django import modulestore from xmodule.modulestore.django import modulestore
from opaque_keys.edx.locations import SlashSeparatedCourseKey from opaque_keys.edx.locations import SlashSeparatedCourseKey
...@@ -80,7 +79,7 @@ class ShibSPTest(ModuleStoreTestCase): ...@@ -80,7 +79,7 @@ class ShibSPTest(ModuleStoreTestCase):
request_factory = RequestFactory() request_factory = RequestFactory()
def setUp(self): def setUp(self):
self.store = modulestore() super(ShibSPTest, self).setUp(create_user=False)
@unittest.skipUnless(settings.FEATURES.get('AUTH_USE_SHIB'), "AUTH_USE_SHIB not set") @unittest.skipUnless(settings.FEATURES.get('AUTH_USE_SHIB'), "AUTH_USE_SHIB not set")
def test_exception_shib_login(self): def test_exception_shib_login(self):
...@@ -383,7 +382,7 @@ class ShibSPTest(ModuleStoreTestCase): ...@@ -383,7 +382,7 @@ class ShibSPTest(ModuleStoreTestCase):
for domain in ["", "shib:https://idp.stanford.edu/"]: for domain in ["", "shib:https://idp.stanford.edu/"]:
# set domains # set domains
course.enrollment_domain = domain course.enrollment_domain = domain
self.store.update_item(course, '**replace_user**') self.store.update_item(course, self.user.id)
# setting location to test that GET params get passed through # setting location to test that GET params get passed through
login_request = self.request_factory.get('/course_specific_login/MITx/999/Robot_Super_Course' + login_request = self.request_factory.get('/course_specific_login/MITx/999/Robot_Super_Course' +
...@@ -452,11 +451,11 @@ class ShibSPTest(ModuleStoreTestCase): ...@@ -452,11 +451,11 @@ class ShibSPTest(ModuleStoreTestCase):
# create 2 course, one with limited enrollment one without # create 2 course, one with limited enrollment one without
shib_course = CourseFactory.create(org='Stanford', number='123', display_name='Shib Only') shib_course = CourseFactory.create(org='Stanford', number='123', display_name='Shib Only')
shib_course.enrollment_domain = 'shib:https://idp.stanford.edu/' shib_course.enrollment_domain = 'shib:https://idp.stanford.edu/'
self.store.update_item(shib_course, '**replace_user**') self.store.update_item(shib_course, self.user.id)
open_enroll_course = CourseFactory.create(org='MITx', number='999', display_name='Robot Super Course') open_enroll_course = CourseFactory.create(org='MITx', number='999', display_name='Robot Super Course')
open_enroll_course.enrollment_domain = '' open_enroll_course.enrollment_domain = ''
self.store.update_item(open_enroll_course, '**replace_user**') self.store.update_item(open_enroll_course, self.user.id)
# create 3 kinds of students, external_auth matching shib_course, external_auth not matching, no external auth # create 3 kinds of students, external_auth matching shib_course, external_auth not matching, no external auth
shib_student = UserFactory.create() shib_student = UserFactory.create()
...@@ -522,7 +521,7 @@ class ShibSPTest(ModuleStoreTestCase): ...@@ -522,7 +521,7 @@ class ShibSPTest(ModuleStoreTestCase):
course = CourseFactory.create(org='Stanford', number='123', display_name='Shib Only') course = CourseFactory.create(org='Stanford', number='123', display_name='Shib Only')
course.enrollment_domain = 'shib:https://idp.stanford.edu/' course.enrollment_domain = 'shib:https://idp.stanford.edu/'
self.store.update_item(course, '**replace_user**') self.store.update_item(course, self.user.id)
# use django test client for sessions and url processing # use django test client for sessions and url processing
# no enrollment before trying # no enrollment before trying
......
...@@ -17,7 +17,6 @@ from student.views import _parse_course_id_from_string, _get_course_enrollment_d ...@@ -17,7 +17,6 @@ from student.views import _parse_course_id_from_string, _get_course_enrollment_d
from xmodule.modulestore.tests.factories import CourseFactory from xmodule.modulestore.tests.factories import CourseFactory
from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase, mixed_store_config from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase, mixed_store_config
from xmodule.modulestore.inheritance import own_metadata
from xmodule.modulestore.django import modulestore from xmodule.modulestore.django import modulestore
from external_auth.models import ExternalAuthMap from external_auth.models import ExternalAuthMap
...@@ -346,11 +345,11 @@ class ExternalAuthShibTest(ModuleStoreTestCase): ...@@ -346,11 +345,11 @@ class ExternalAuthShibTest(ModuleStoreTestCase):
Tests how login_user() interacts with ExternalAuth, in particular Shib Tests how login_user() interacts with ExternalAuth, in particular Shib
""" """
def setUp(self): def setUp(self):
self.store = modulestore() super(ExternalAuthShibTest, self).setUp()
self.course = CourseFactory.create(org='Stanford', number='456', display_name='NO SHIB') self.course = CourseFactory.create(org='Stanford', number='456', display_name='NO SHIB')
self.shib_course = CourseFactory.create(org='Stanford', number='123', display_name='Shib Only') self.shib_course = CourseFactory.create(org='Stanford', number='123', display_name='Shib Only')
self.shib_course.enrollment_domain = 'shib:https://idp.stanford.edu/' self.shib_course.enrollment_domain = 'shib:https://idp.stanford.edu/'
self.store.update_item(self.shib_course, '**replace_user**') self.store.update_item(self.shib_course, self.user.id)
self.user_w_map = UserFactory.create(email='withmap@stanford.edu') self.user_w_map = UserFactory.create(email='withmap@stanford.edu')
self.extauth = ExternalAuthMap(external_id='withmap@stanford.edu', self.extauth = ExternalAuthMap(external_id='withmap@stanford.edu',
external_email='withmap@stanford.edu', external_email='withmap@stanford.edu',
......
...@@ -30,8 +30,8 @@ class UrlResetMixin(object): ...@@ -30,8 +30,8 @@ class UrlResetMixin(object):
# Resolve a URL so that the new urlconf gets loaded # Resolve a URL so that the new urlconf gets loaded
resolve('/') resolve('/')
def setUp(self): def setUp(self, **kwargs):
"""Reset django default urlconf before tests and after tests""" """Reset django default urlconf before tests and after tests"""
super(UrlResetMixin, self).setUp() super(UrlResetMixin, self).setUp(**kwargs)
self._reset_urls() self._reset_urls()
self.addCleanup(self._reset_urls) self.addCleanup(self._reset_urls)
...@@ -76,8 +76,8 @@ class ModuleStoreEnum(object): ...@@ -76,8 +76,8 @@ class ModuleStoreEnum(object):
""" """
Values for user ID defaults Values for user ID defaults
""" """
# Note: we use negative values here to (try to) not conflict # Note: we use negative values here to (try to) not collide
# with user identifiers provided by an actual user service. # with user identifiers provided by actual user services.
# user ID to use for all management commands # user ID to use for all management commands
mgmt_command = -1 mgmt_command = -1
...@@ -307,7 +307,7 @@ class ModuleStoreWrite(ModuleStoreRead): ...@@ -307,7 +307,7 @@ class ModuleStoreWrite(ModuleStoreRead):
__metaclass__ = ABCMeta __metaclass__ = ABCMeta
@abstractmethod @abstractmethod
def update_item(self, xblock, user_id=None, allow_not_found=False, force=False): def update_item(self, xblock, user_id, allow_not_found=False, force=False):
""" """
Update the given xblock's persisted repr. Pass the user's unique id which the persistent store Update the given xblock's persisted repr. Pass the user's unique id which the persistent store
should save with the update if it has that ability. should save with the update if it has that ability.
...@@ -323,7 +323,7 @@ class ModuleStoreWrite(ModuleStoreRead): ...@@ -323,7 +323,7 @@ class ModuleStoreWrite(ModuleStoreRead):
pass pass
@abstractmethod @abstractmethod
def delete_item(self, location, user_id=None, **kwargs): def delete_item(self, location, user_id, **kwargs):
""" """
Delete an item and its subtree from persistence. Remove the item from any parents (Note, does not Delete an item and its subtree from persistence. Remove the item from any parents (Note, does not
affect parents from other branches or logical branches; thus, in old mongo, deleting something affect parents from other branches or logical branches; thus, in old mongo, deleting something
...@@ -342,7 +342,7 @@ class ModuleStoreWrite(ModuleStoreRead): ...@@ -342,7 +342,7 @@ class ModuleStoreWrite(ModuleStoreRead):
pass pass
@abstractmethod @abstractmethod
def create_course(self, org, offering, user_id=None, fields=None, **kwargs): def create_course(self, org, offering, user_id, fields=None, **kwargs):
""" """
Creates and returns the course. Creates and returns the course.
...@@ -532,7 +532,7 @@ class ModuleStoreWriteBase(ModuleStoreReadBase, ModuleStoreWrite): ...@@ -532,7 +532,7 @@ class ModuleStoreWriteBase(ModuleStoreReadBase, ModuleStoreWrite):
result[field.scope][field_name] = value result[field.scope][field_name] = value
return result return result
def update_item(self, xblock, user_id=None, allow_not_found=False, force=False): def update_item(self, xblock, user_id, allow_not_found=False, force=False):
""" """
Update the given xblock's persisted repr. Pass the user's unique id which the persistent store Update the given xblock's persisted repr. Pass the user's unique id which the persistent store
should save with the update if it has that ability. should save with the update if it has that ability.
...@@ -547,7 +547,7 @@ class ModuleStoreWriteBase(ModuleStoreReadBase, ModuleStoreWrite): ...@@ -547,7 +547,7 @@ class ModuleStoreWriteBase(ModuleStoreReadBase, ModuleStoreWrite):
""" """
raise NotImplementedError raise NotImplementedError
def delete_item(self, location, user_id=None, force=False): def delete_item(self, location, user_id, force=False):
""" """
Delete an item from persistence. Pass the user's unique id which the persistent store Delete an item from persistence. Pass the user's unique id which the persistent store
should save with the update if it has that ability. should save with the update if it has that ability.
......
...@@ -273,7 +273,7 @@ class MixedModuleStore(ModuleStoreWriteBase): ...@@ -273,7 +273,7 @@ class MixedModuleStore(ModuleStoreWriteBase):
errs.update(store.get_errored_courses()) errs.update(store.get_errored_courses())
return errs return errs
def create_course(self, org, offering, user_id=None, fields=None, **kwargs): def create_course(self, org, offering, user_id, fields=None, **kwargs):
""" """
Creates and returns the course. Creates and returns the course.
...@@ -319,7 +319,7 @@ class MixedModuleStore(ModuleStoreWriteBase): ...@@ -319,7 +319,7 @@ class MixedModuleStore(ModuleStoreWriteBase):
source_course_id, user_id, dest_course_id.org, dest_course_id.offering source_course_id, user_id, dest_course_id.org, dest_course_id.offering
) )
def create_item(self, course_or_parent_loc, category, user_id=None, **kwargs): def create_item(self, course_or_parent_loc, category, user_id, **kwargs):
""" """
Create and return the item. If parent_loc is a specific location v a course id, Create and return the item. If parent_loc is a specific location v a course id,
it installs the new item as a child of the parent (if the parent_loc is a specific it installs the new item as a child of the parent (if the parent_loc is a specific
...@@ -346,7 +346,7 @@ class MixedModuleStore(ModuleStoreWriteBase): ...@@ -346,7 +346,7 @@ class MixedModuleStore(ModuleStoreWriteBase):
if parent_loc is not None and not 'detached' in xblock._class_tags: if parent_loc is not None and not 'detached' in xblock._class_tags:
parent = store.get_item(parent_loc) parent = store.get_item(parent_loc)
parent.children.append(location) parent.children.append(location)
store.update_item(parent) store.update_item(parent, user_id)
elif isinstance(store, SplitMongoModuleStore): elif isinstance(store, SplitMongoModuleStore):
if not isinstance(course_or_parent_loc, (CourseLocator, BlockUsageLocator)): if not isinstance(course_or_parent_loc, (CourseLocator, BlockUsageLocator)):
raise ValueError(u"Cannot create a child of {} in split. Wrong repr.".format(course_or_parent_loc)) raise ValueError(u"Cannot create a child of {} in split. Wrong repr.".format(course_or_parent_loc))
...@@ -371,7 +371,7 @@ class MixedModuleStore(ModuleStoreWriteBase): ...@@ -371,7 +371,7 @@ class MixedModuleStore(ModuleStoreWriteBase):
store = self._verify_modulestore_support(xblock.location, 'update_item') store = self._verify_modulestore_support(xblock.location, 'update_item')
return store.update_item(xblock, user_id, allow_not_found) return store.update_item(xblock, user_id, allow_not_found)
def delete_item(self, location, user_id=None, **kwargs): def delete_item(self, location, user_id, **kwargs):
""" """
Delete the given item from persistence. kwargs allow modulestore specific parameters. Delete the given item from persistence. kwargs allow modulestore specific parameters.
""" """
......
...@@ -841,7 +841,7 @@ class MongoModuleStore(ModuleStoreWriteBase): ...@@ -841,7 +841,7 @@ class MongoModuleStore(ModuleStoreWriteBase):
modules = self._load_items(course_id, list(items)) modules = self._load_items(course_id, list(items))
return modules return modules
def create_course(self, org, offering, user_id=None, fields=None, **kwargs): def create_course(self, org, offering, user_id, fields=None, **kwargs):
""" """
Creates and returns the course. Creates and returns the course.
...@@ -983,7 +983,7 @@ class MongoModuleStore(ModuleStoreWriteBase): ...@@ -983,7 +983,7 @@ class MongoModuleStore(ModuleStoreWriteBase):
self._update_single_item(parent, update) self._update_single_item(parent, update)
self._update_ancestors(parent, update) self._update_ancestors(parent, update)
def update_item(self, xblock, user_id=None, allow_not_found=False, force=False, isPublish=False, def update_item(self, xblock, user_id, allow_not_found=False, force=False, isPublish=False,
is_publish_root=True): is_publish_root=True):
""" """
Update the persisted version of xblock to reflect its current values. Update the persisted version of xblock to reflect its current values.
......
...@@ -431,7 +431,7 @@ class DraftModuleStore(MongoModuleStore): ...@@ -431,7 +431,7 @@ class DraftModuleStore(MongoModuleStore):
# get_item will wrap_draft so don't call it here (otherwise, it would override the is_draft attribute) # get_item will wrap_draft so don't call it here (otherwise, it would override the is_draft attribute)
return self.get_item(location) return self.get_item(location)
def update_item(self, xblock, user_id=None, allow_not_found=False, force=False, isPublish=False): def update_item(self, xblock, user_id, allow_not_found=False, force=False, isPublish=False):
""" """
See superclass doc. See superclass doc.
In addition to the superclass's behavior, this method converts the unit to draft if it's not In addition to the superclass's behavior, this method converts the unit to draft if it's not
......
...@@ -4,8 +4,10 @@ Modulestore configuration for test cases. ...@@ -4,8 +4,10 @@ Modulestore configuration for test cases.
from uuid import uuid4 from uuid import uuid4
from django.test import TestCase from django.test import TestCase
from django.contrib.auth.models import User
from xmodule.modulestore.django import ( from xmodule.modulestore.django import (
modulestore, clear_existing_modulestores, loc_mapper) modulestore, clear_existing_modulestores, loc_mapper
)
from xmodule.modulestore import ModuleStoreEnum from xmodule.modulestore import ModuleStoreEnum
...@@ -126,18 +128,54 @@ class ModuleStoreTestCase(TestCase): ...@@ -126,18 +128,54 @@ class ModuleStoreTestCase(TestCase):
`clear_existing_modulestores()` directly in `clear_existing_modulestores()` directly in
your `setUp()` method. your `setUp()` method.
""" """
def setUp(self, **kwargs):
super(ModuleStoreTestCase, self).setUp()
@staticmethod self.store = modulestore()
def update_course(course):
uname = 'testuser'
email = 'test+courses@edx.org'
password = 'foo'
if kwargs.pop('create_user', True):
# Create the user so we can log them in.
self.user = User.objects.create_user(uname, email, password)
# Note that we do not actually need to do anything
# for registration if we directly mark them active.
self.user.is_active = True
# Staff has access to view all courses
self.user.is_staff = True
self.user.save()
return password
def create_non_staff_authed_user_client(self):
"""
Create a non-staff user, log them in (if authenticate=True), and return the client, user to use for testing.
"""
uname = 'teststudent'
password = 'foo'
nonstaff = User.objects.create_user(uname, 'test+student@edx.org', password)
# Note that we do not actually need to do anything
# for registration if we directly mark them active.
nonstaff.is_active = True
nonstaff.is_staff = False
nonstaff.save()
return nonstaff, password
def update_course(self, course, user_id):
""" """
Updates the version of course in the modulestore Updates the version of course in the modulestore
'course' is an instance of CourseDescriptor for which we want 'course' is an instance of CourseDescriptor for which we want
to update metadata. to update metadata.
""" """
store = modulestore() with self.store.branch_setting(ModuleStoreEnum.Branch.draft_preferred, course.id):
store.update_item(course, '**replace_user**') self.store.update_item(course, user_id)
updated_course = store.get_course(course.id) updated_course = self.store.get_course(course.id)
return updated_course return updated_course
@staticmethod @staticmethod
......
...@@ -2,7 +2,7 @@ from factory import Factory, lazy_attribute_sequence, lazy_attribute ...@@ -2,7 +2,7 @@ from factory import Factory, lazy_attribute_sequence, lazy_attribute
from factory.containers import CyclicDefinitionError from factory.containers import CyclicDefinitionError
from uuid import uuid4 from uuid import uuid4
from xmodule.modulestore import prefer_xmodules from xmodule.modulestore import prefer_xmodules, ModuleStoreEnum
from opaque_keys.edx.locations import Location from opaque_keys.edx.locations import Location
from xblock.core import XBlock from xblock.core import XBlock
from xmodule.tabs import StaticTab from xmodule.tabs import StaticTab
...@@ -52,6 +52,7 @@ class CourseFactory(XModuleFactory): ...@@ -52,6 +52,7 @@ class CourseFactory(XModuleFactory):
store = kwargs.pop('modulestore') store = kwargs.pop('modulestore')
name = kwargs.get('name', kwargs.get('run', Location.clean(kwargs.get('display_name')))) name = kwargs.get('name', kwargs.get('run', Location.clean(kwargs.get('display_name'))))
run = kwargs.get('run', name) run = kwargs.get('run', name)
user_id = kwargs.pop('user_id', ModuleStoreEnum.UserID.test)
location = Location(org, number, run, 'course', name) location = Location(org, number, run, 'course', name)
...@@ -65,7 +66,7 @@ class CourseFactory(XModuleFactory): ...@@ -65,7 +66,7 @@ class CourseFactory(XModuleFactory):
# Save the attributes we just set # Save the attributes we just set
new_course.save() new_course.save()
# Update the data in the mongo datastore # Update the data in the mongo datastore
store.update_item(new_course, '**replace_user**') store.update_item(new_course, user_id)
return new_course return new_course
...@@ -143,7 +144,7 @@ class ItemFactory(XModuleFactory): ...@@ -143,7 +144,7 @@ class ItemFactory(XModuleFactory):
display_name = kwargs.pop('display_name', None) display_name = kwargs.pop('display_name', None)
metadata = kwargs.pop('metadata', {}) metadata = kwargs.pop('metadata', {})
location = kwargs.pop('location') location = kwargs.pop('location')
user_id = kwargs.pop('user_id', 999) user_id = kwargs.pop('user_id', ModuleStoreEnum.UserID.test)
assert isinstance(location, Location) assert isinstance(location, Location)
assert location != parent_location assert location != parent_location
...@@ -175,11 +176,11 @@ class ItemFactory(XModuleFactory): ...@@ -175,11 +176,11 @@ class ItemFactory(XModuleFactory):
# Save the attributes we just set # Save the attributes we just set
module.save() module.save()
store.update_item(module, '**replace_user**') store.update_item(module, user_id)
if 'detached' not in module._class_tags: if 'detached' not in module._class_tags:
parent.children.append(location) parent.children.append(location)
store.update_item(parent, '**replace_user**') store.update_item(parent, user_id)
# VS[compat] cdodge: This is a hack because static_tabs also have references from the course module, so # VS[compat] cdodge: This is a hack because static_tabs also have references from the course module, so
# if we add one then we need to also add it to the policy information (i.e. metadata) # if we add one then we need to also add it to the policy information (i.e. metadata)
...@@ -192,7 +193,7 @@ class ItemFactory(XModuleFactory): ...@@ -192,7 +193,7 @@ class ItemFactory(XModuleFactory):
url_slug=location.name, url_slug=location.name,
) )
) )
store.update_item(course, '**replace_user**') store.update_item(course, user_id)
return store.get_item(location) return store.get_item(location)
......
...@@ -33,7 +33,7 @@ class PersistentCourseFactory(SplitFactory): ...@@ -33,7 +33,7 @@ class PersistentCourseFactory(SplitFactory):
# pylint: disable=W0613 # pylint: disable=W0613
@classmethod @classmethod
def _create(cls, target_class, offering='999', org='testX', user_id='test_user', def _create(cls, target_class, offering='999', org='testX', user_id=ModuleStoreEnum.UserID.test,
master_branch=ModuleStoreEnum.BranchName.draft, **kwargs): master_branch=ModuleStoreEnum.BranchName.draft, **kwargs):
modulestore = kwargs.pop('modulestore') modulestore = kwargs.pop('modulestore')
...@@ -59,7 +59,7 @@ class ItemFactory(SplitFactory): ...@@ -59,7 +59,7 @@ class ItemFactory(SplitFactory):
# pylint: disable=W0613 # pylint: disable=W0613
@classmethod @classmethod
def _create(cls, target_class, parent_location, category='chapter', def _create(cls, target_class, parent_location, category='chapter',
user_id='test_user', block_id=None, definition_locator=None, force=False, user_id=ModuleStoreEnum.UserID.test, block_id=None, definition_locator=None, force=False,
continue_version=False, **kwargs): continue_version=False, **kwargs):
""" """
passes *kwargs* as the new item's field values: passes *kwargs* as the new item's field values:
......
...@@ -116,7 +116,7 @@ class TestMixedModuleStore(LocMapperSetupSansDjango): ...@@ -116,7 +116,7 @@ class TestMixedModuleStore(LocMapperSetupSansDjango):
self.writable_chapter_location = self.store = self.fake_location = self.xml_chapter_location = None self.writable_chapter_location = self.store = self.fake_location = self.xml_chapter_location = None
self.course_locations = [] self.course_locations = []
self.user_id = 0 self.user_id = ModuleStoreEnum.UserID.test
# pylint: disable=invalid-name # pylint: disable=invalid-name
def _create_course(self, default, course_key): def _create_course(self, default, course_key):
...@@ -127,12 +127,12 @@ class TestMixedModuleStore(LocMapperSetupSansDjango): ...@@ -127,12 +127,12 @@ class TestMixedModuleStore(LocMapperSetupSansDjango):
offering = course_key.offering.replace('/', '.') offering = course_key.offering.replace('/', '.')
else: else:
offering = course_key.offering offering = course_key.offering
course = self.store.create_course(course_key.org, offering) course = self.store.create_course(course_key.org, offering, self.user_id)
category = self.writable_chapter_location.category category = self.writable_chapter_location.category
block_id = self.writable_chapter_location.name block_id = self.writable_chapter_location.name
chapter = self.store.create_item( chapter = self.store.create_item(
# don't use course_location as it may not be the repr # don't use course_location as it may not be the repr
course.location, category, location=self.writable_chapter_location, block_id=block_id course.location, category, self.user_id, location=self.writable_chapter_location, block_id=block_id
) )
if isinstance(course.id, CourseLocator): if isinstance(course.id, CourseLocator):
self.course_locations[self.MONGO_COURSEID] = course.location.version_agnostic() self.course_locations[self.MONGO_COURSEID] = course.location.version_agnostic()
...@@ -184,7 +184,7 @@ class TestMixedModuleStore(LocMapperSetupSansDjango): ...@@ -184,7 +184,7 @@ class TestMixedModuleStore(LocMapperSetupSansDjango):
] ]
def create_sub_tree(parent, block_info): def create_sub_tree(parent, block_info):
block = self.store.create_item(parent.location, category=block_info.category, block_id=block_info.display_name) block = self.store.create_item(parent.location, block_info.category, self.user_id, block_id=block_info.display_name)
for tree in block_info.sub_tree: for tree in block_info.sub_tree:
create_sub_tree(block, tree) create_sub_tree(block, tree)
# reload the block to update its children field # reload the block to update its children field
...@@ -306,13 +306,13 @@ class TestMixedModuleStore(LocMapperSetupSansDjango): ...@@ -306,13 +306,13 @@ class TestMixedModuleStore(LocMapperSetupSansDjango):
self.assertFalse(course.show_calculator, "Default changed making test meaningless") self.assertFalse(course.show_calculator, "Default changed making test meaningless")
course.show_calculator = True course.show_calculator = True
with self.assertRaises(NotImplementedError): # ensure it doesn't allow writing with self.assertRaises(NotImplementedError): # ensure it doesn't allow writing
self.store.update_item(course, None) self.store.update_item(course, self.user_id)
# now do it for a r/w db # now do it for a r/w db
course = self.store.get_course(self.course_locations[self.MONGO_COURSEID].course_key) course = self.store.get_course(self.course_locations[self.MONGO_COURSEID].course_key)
# if following raised, then the test is really a noop, change it # if following raised, then the test is really a noop, change it
self.assertFalse(course.show_calculator, "Default changed making test meaningless") self.assertFalse(course.show_calculator, "Default changed making test meaningless")
course.show_calculator = True course.show_calculator = True
self.store.update_item(course, None) self.store.update_item(course, self.user_id)
course = self.store.get_course(self.course_locations[self.MONGO_COURSEID].course_key) course = self.store.get_course(self.course_locations[self.MONGO_COURSEID].course_key)
self.assertTrue(course.show_calculator) self.assertTrue(course.show_calculator)
...@@ -324,8 +324,8 @@ class TestMixedModuleStore(LocMapperSetupSansDjango): ...@@ -324,8 +324,8 @@ class TestMixedModuleStore(LocMapperSetupSansDjango):
self.initdb(default_ms) self.initdb(default_ms)
# r/o try deleting the course (is here to ensure it can't be deleted) # r/o try deleting the course (is here to ensure it can't be deleted)
with self.assertRaises(NotImplementedError): with self.assertRaises(NotImplementedError):
self.store.delete_item(self.xml_chapter_location, 13) self.store.delete_item(self.xml_chapter_location, self.user_id)
self.store.delete_item(self.writable_chapter_location, 9) self.store.delete_item(self.writable_chapter_location, self.user_id)
# verify it's gone # verify it's gone
with self.assertRaises(ItemNotFoundError): with self.assertRaises(ItemNotFoundError):
self.store.get_item(self.writable_chapter_location) self.store.get_item(self.writable_chapter_location)
...@@ -367,7 +367,7 @@ class TestMixedModuleStore(LocMapperSetupSansDjango): ...@@ -367,7 +367,7 @@ class TestMixedModuleStore(LocMapperSetupSansDjango):
xml_store = self.store._get_modulestore_by_type(ModuleStoreEnum.Type.xml) xml_store = self.store._get_modulestore_by_type(ModuleStoreEnum.Type.xml)
# the important thing is not which exception it raises but that it raises an exception # the important thing is not which exception it raises but that it raises an exception
with self.assertRaises(AttributeError): with self.assertRaises(AttributeError):
xml_store.create_course("org", "course/run", 999) xml_store.create_course("org", "course/run", self.user_id)
@ddt.data('draft', 'split') @ddt.data('draft', 'split')
def test_get_course(self, default_ms): def test_get_course(self, default_ms):
...@@ -547,7 +547,7 @@ class TestMixedModuleStore(LocMapperSetupSansDjango): ...@@ -547,7 +547,7 @@ class TestMixedModuleStore(LocMapperSetupSansDjango):
self.initdb(default_ms) self.initdb(default_ms)
# create an orphan # create an orphan
course_id = self.course_locations[self.MONGO_COURSEID].course_key course_id = self.course_locations[self.MONGO_COURSEID].course_key
orphan = self.store.create_item(course_id, 'problem', block_id='orphan') orphan = self.store.create_item(course_id, 'problem', self.user_id, block_id='orphan')
found_orphans = self.store.get_orphans(self.course_locations[self.MONGO_COURSEID].course_key) found_orphans = self.store.get_orphans(self.course_locations[self.MONGO_COURSEID].course_key)
if default_ms == 'split': if default_ms == 'split':
self.assertEqual(found_orphans, [orphan.location.version_agnostic()]) self.assertEqual(found_orphans, [orphan.location.version_agnostic()])
...@@ -561,7 +561,7 @@ class TestMixedModuleStore(LocMapperSetupSansDjango): ...@@ -561,7 +561,7 @@ class TestMixedModuleStore(LocMapperSetupSansDjango):
new location for the child new location for the child
""" """
self.initdb(default_ms) self.initdb(default_ms)
self.store.create_item(self.course_locations[self.MONGO_COURSEID], 'problem', block_id='orphan') self.store.create_item(self.course_locations[self.MONGO_COURSEID], 'problem', self.user_id, block_id='orphan')
orphans = self.store.get_orphans(self.course_locations[self.MONGO_COURSEID].course_key) orphans = self.store.get_orphans(self.course_locations[self.MONGO_COURSEID].course_key)
self.assertEqual(len(orphans), 0, "unexpected orphans: {}".format(orphans)) self.assertEqual(len(orphans), 0, "unexpected orphans: {}".format(orphans))
......
...@@ -5,6 +5,7 @@ Tests for split_migrator ...@@ -5,6 +5,7 @@ Tests for split_migrator
import uuid import uuid
import random import random
import mock import mock
from xmodule.modulestore import ModuleStoreEnum
from xmodule.modulestore.mongo.base import MongoRevisionKey from xmodule.modulestore.mongo.base import MongoRevisionKey
from xmodule.modulestore.loc_mapper_store import LocMapperStore from xmodule.modulestore.loc_mapper_store import LocMapperStore
from xmodule.modulestore.split_migrator import SplitMigrator from xmodule.modulestore.split_migrator import SplitMigrator
...@@ -63,7 +64,7 @@ class TestMigration(SplitWMongoCourseBoostrapper): ...@@ -63,7 +64,7 @@ class TestMigration(SplitWMongoCourseBoostrapper):
self.create_random_units(False, both_vert_loc) self.create_random_units(False, both_vert_loc)
draft_both = self.draft_mongo.get_item(both_vert_loc) draft_both = self.draft_mongo.get_item(both_vert_loc)
draft_both.display_name = 'Both vertical renamed' draft_both.display_name = 'Both vertical renamed'
self.draft_mongo.update_item(draft_both) self.draft_mongo.update_item(draft_both, ModuleStoreEnum.UserID.test)
self.create_random_units(True, both_vert_loc) self.create_random_units(True, both_vert_loc)
# vertical in draft only (x2) # vertical in draft only (x2)
draft_vert_loc = self.old_course_key.make_usage_key('vertical', uuid.uuid4().hex) draft_vert_loc = self.old_course_key.make_usage_key('vertical', uuid.uuid4().hex)
......
...@@ -10,7 +10,6 @@ from xmodule.modulestore.split_mongo.split import SplitMongoModuleStore ...@@ -10,7 +10,6 @@ from xmodule.modulestore.split_mongo.split import SplitMongoModuleStore
from xmodule.modulestore.mongo import MongoModuleStore, DraftMongoModuleStore from xmodule.modulestore.mongo import MongoModuleStore, DraftMongoModuleStore
from xmodule.modulestore.mongo.draft import DIRECT_ONLY_CATEGORIES from xmodule.modulestore.mongo.draft import DIRECT_ONLY_CATEGORIES
from xmodule.modulestore import ModuleStoreEnum from xmodule.modulestore import ModuleStoreEnum
from mock import Mock
class SplitWMongoCourseBoostrapper(unittest.TestCase): class SplitWMongoCourseBoostrapper(unittest.TestCase):
...@@ -138,6 +137,6 @@ class SplitWMongoCourseBoostrapper(unittest.TestCase): ...@@ -138,6 +137,6 @@ class SplitWMongoCourseBoostrapper(unittest.TestCase):
self.split_mongo.create_course( self.split_mongo.create_course(
self.split_course_key.org, self.split_course_key.offering, self.userid, fields=fields, root_block_id='runid' self.split_course_key.org, self.split_course_key.offering, self.userid, fields=fields, root_block_id='runid'
) )
old_course = self.old_mongo.create_course(self.split_course_key.org, 'test_course/runid', fields=fields) old_course = self.old_mongo.create_course(self.split_course_key.org, 'test_course/runid', self.userid, fields=fields)
self.old_course_key = old_course.id self.old_course_key = old_course.id
self.runtime = old_course.runtime self.runtime = old_course.runtime
...@@ -169,7 +169,7 @@ def import_from_xml( ...@@ -169,7 +169,7 @@ def import_from_xml(
# Creates a new course if it doesn't already exist # Creates a new course if it doesn't already exist
if create_new_course_if_not_present and not store.has_course(dest_course_id, ignore_case=True): if create_new_course_if_not_present and not store.has_course(dest_course_id, ignore_case=True):
try: try:
store.create_course(dest_course_id.org, dest_course_id.offering) store.create_course(dest_course_id.org, dest_course_id.offering, user_id)
except InvalidLocationError: except InvalidLocationError:
# course w/ same org and course exists # course w/ same org and course exists
log.debug( log.debug(
......
...@@ -559,7 +559,7 @@ class SplitTestDescriptor(SplitTestFields, SequenceDescriptor, StudioEditableDes ...@@ -559,7 +559,7 @@ class SplitTestDescriptor(SplitTestFields, SequenceDescriptor, StudioEditableDes
changed = True changed = True
if changed: if changed:
# request does not have a user attribute, so pass None for user. # TODO user.id - to be fixed by Publishing team
self.system.modulestore.update_item(self, None) self.system.modulestore.update_item(self, None)
return Response() return Response()
......
...@@ -283,7 +283,7 @@ class VideoDescriptor(VideoFields, VideoStudioViewHandlers, TabsEditingDescripto ...@@ -283,7 +283,7 @@ class VideoDescriptor(VideoFields, VideoStudioViewHandlers, TabsEditingDescripto
Save module with updated metadata to database." Save module with updated metadata to database."
""" """
self.save() self.save()
self.runtime.modulestore.update_item(self, user.id if user else None) self.runtime.modulestore.update_item(self, user.id)
@property @property
def editable_metadata_fields(self): def editable_metadata_fields(self):
......
...@@ -28,12 +28,12 @@ class AnonymousIndexPageTest(ModuleStoreTestCase): ...@@ -28,12 +28,12 @@ class AnonymousIndexPageTest(ModuleStoreTestCase):
Tests that anonymous users can access the '/' page, Need courses with start date Tests that anonymous users can access the '/' page, Need courses with start date
""" """
def setUp(self): def setUp(self):
self.store = modulestore() super(AnonymousIndexPageTest, self).setUp()
self.factory = RequestFactory() self.factory = RequestFactory()
self.course = CourseFactory.create() self.course = CourseFactory.create()
self.course.days_early_for_beta = 5 self.course.days_early_for_beta = 5
self.course.enrollment_start = datetime.datetime.now(UTC) + datetime.timedelta(days=3) self.course.enrollment_start = datetime.datetime.now(UTC) + datetime.timedelta(days=3)
self.store.update_item(self.course, '**replace_user**') self.store.update_item(self.course, self.user.id)
@override_settings(FEATURES=FEATURES_WITH_STARTDATE) @override_settings(FEATURES=FEATURES_WITH_STARTDATE)
def test_none_user_index_access_with_startdate_fails(self): def test_none_user_index_access_with_startdate_fails(self):
......
...@@ -18,6 +18,7 @@ from courseware.tests.modulestore_config import TEST_DATA_XML_MODULESTORE ...@@ -18,6 +18,7 @@ from courseware.tests.modulestore_config import TEST_DATA_XML_MODULESTORE
from courseware.tests.modulestore_config import TEST_DATA_MIXED_MODULESTORE from courseware.tests.modulestore_config import TEST_DATA_MIXED_MODULESTORE
from courseware.tests.modulestore_config import TEST_DATA_MONGO_MODULESTORE from courseware.tests.modulestore_config import TEST_DATA_MONGO_MODULESTORE
from xmodule.modulestore import ModuleStoreEnum
from xmodule.modulestore.django import modulestore from xmodule.modulestore.django import modulestore
from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase
from xmodule.modulestore.tests.factories import CourseFactory from xmodule.modulestore.tests.factories import CourseFactory
...@@ -56,7 +57,7 @@ class CommandsTestBase(TestCase): ...@@ -56,7 +57,7 @@ class CommandsTestBase(TestCase):
courses = store.get_courses() courses = store.get_courses()
# NOTE: if xml store owns these, it won't import them into mongo # NOTE: if xml store owns these, it won't import them into mongo
if SlashSeparatedCourseKey.from_deprecated_string(TEST_COURSE_ID) not in [c.id for c in courses]: if SlashSeparatedCourseKey.from_deprecated_string(TEST_COURSE_ID) not in [c.id for c in courses]:
import_from_xml(store, "**replace_user**", DATA_DIR, ['toy', 'simple']) import_from_xml(store, ModuleStoreEnum.UserID.mgmt_command, DATA_DIR, ['toy', 'simple'])
return [course.id for course in store.get_courses()] return [course.id for course in store.get_courses()]
......
...@@ -132,6 +132,7 @@ class BaseTestXmodule(ModuleStoreTestCase): ...@@ -132,6 +132,7 @@ class BaseTestXmodule(ModuleStoreTestCase):
self.assertTrue(all(self.login_statuses)) self.assertTrue(all(self.login_statuses))
def setUp(self): def setUp(self):
super(BaseTestXmodule, self).setUp()
self.setup_course() self.setup_course()
self.initialize_module(metadata=self.METADATA, data=self.DATA) self.initialize_module(metadata=self.METADATA, data=self.DATA)
......
...@@ -67,7 +67,7 @@ class LoginEnrollmentTestCase(TestCase): ...@@ -67,7 +67,7 @@ class LoginEnrollmentTestCase(TestCase):
self.email = 'foo@test.com' self.email = 'foo@test.com'
self.password = 'bar' self.password = 'bar'
self.username = 'test' self.username = 'test'
self.create_account(self.username, self.user = self.create_account(self.username,
self.email, self.password) self.email, self.password)
self.activate_user(self.email) self.activate_user(self.email)
self.login(self.email, self.password) self.login(self.email, self.password)
...@@ -107,7 +107,9 @@ class LoginEnrollmentTestCase(TestCase): ...@@ -107,7 +107,9 @@ class LoginEnrollmentTestCase(TestCase):
data = json.loads(resp.content) data = json.loads(resp.content)
self.assertEqual(data['success'], True) self.assertEqual(data['success'], True)
# Check both that the user is created, and inactive # Check both that the user is created, and inactive
self.assertFalse(User.objects.get(email=email).is_active) user = User.objects.get(email=email)
self.assertFalse(user.is_active)
return user
def activate_user(self, email): def activate_user(self, email):
""" """
......
...@@ -19,8 +19,6 @@ from django.test.utils import override_settings ...@@ -19,8 +19,6 @@ from django.test.utils import override_settings
from courseware import grades from courseware import grades
from courseware.models import StudentModule from courseware.models import StudentModule
from xmodule.modulestore.django import modulestore
#import factories and parent testcase modules #import factories and parent testcase modules
from xmodule.modulestore.tests.factories import CourseFactory, ItemFactory from xmodule.modulestore.tests.factories import CourseFactory, ItemFactory
from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase
...@@ -46,7 +44,7 @@ class TestSubmittingProblems(ModuleStoreTestCase, LoginEnrollmentTestCase): ...@@ -46,7 +44,7 @@ class TestSubmittingProblems(ModuleStoreTestCase, LoginEnrollmentTestCase):
def setUp(self): def setUp(self):
super(TestSubmittingProblems, self).setUp() super(TestSubmittingProblems, self).setUp(create_user=False)
# Create course # Create course
self.course = CourseFactory.create(display_name=self.COURSE_NAME, number=self.COURSE_SLUG) self.course = CourseFactory.create(display_name=self.COURSE_NAME, number=self.COURSE_SLUG)
assert self.course, "Couldn't load course %r" % self.COURSE_NAME assert self.course, "Couldn't load course %r" % self.COURSE_NAME
...@@ -64,7 +62,7 @@ class TestSubmittingProblems(ModuleStoreTestCase, LoginEnrollmentTestCase): ...@@ -64,7 +62,7 @@ class TestSubmittingProblems(ModuleStoreTestCase, LoginEnrollmentTestCase):
""" """
Re-fetch the course from the database so that the object being dealt with has everything added to it. Re-fetch the course from the database so that the object being dealt with has everything added to it.
""" """
self.course = modulestore().get_course(self.course.id) self.course = self.store.get_course(self.course.id)
def problem_location(self, problem_url_name): def problem_location(self, problem_url_name):
""" """
...@@ -230,8 +228,7 @@ class TestCourseGrader(TestSubmittingProblems): ...@@ -230,8 +228,7 @@ class TestCourseGrader(TestSubmittingProblems):
""" """
self.course.grading_policy = grading_policy self.course.grading_policy = grading_policy
store = modulestore() self.update_course(self.course, self.student_user.id)
store.update_item(self.course, '**replace_user**')
self.refresh_course() self.refresh_course()
def get_grade_summary(self): def get_grade_summary(self):
......
...@@ -411,7 +411,7 @@ class TestTranscriptTranslationGetDispatch(TestVideo): ...@@ -411,7 +411,7 @@ class TestTranscriptTranslationGetDispatch(TestVideo):
self.course.static_asset_path = 'dummy/static' self.course.static_asset_path = 'dummy/static'
self.course.save() self.course.save()
store = modulestore() store = modulestore()
store.update_item(self.course, 'OEoXaMPEzfM') store.update_item(self.course, self.user.id)
# Test youtube style en # Test youtube style en
request = Request.blank('/translation/en?videoId=12345') request = Request.blank('/translation/en?videoId=12345')
......
...@@ -110,6 +110,7 @@ class TestViewAuth(ModuleStoreTestCase, LoginEnrollmentTestCase): ...@@ -110,6 +110,7 @@ class TestViewAuth(ModuleStoreTestCase, LoginEnrollmentTestCase):
return super(TestViewAuth, self).login(user.email, 'test') return super(TestViewAuth, self).login(user.email, 'test')
def setUp(self): def setUp(self):
super(TestViewAuth, self).setUp()
self.course = CourseFactory.create(number='999', display_name='Robot_Super_Course') self.course = CourseFactory.create(number='999', display_name='Robot_Super_Course')
self.courseware_chapter = ItemFactory.create(display_name='courseware') self.courseware_chapter = ItemFactory.create(display_name='courseware')
...@@ -296,8 +297,8 @@ class TestViewAuth(ModuleStoreTestCase, LoginEnrollmentTestCase): ...@@ -296,8 +297,8 @@ class TestViewAuth(ModuleStoreTestCase, LoginEnrollmentTestCase):
tomorrow = now + datetime.timedelta(days=1) tomorrow = now + datetime.timedelta(days=1)
self.course.start = tomorrow self.course.start = tomorrow
self.test_course.start = tomorrow self.test_course.start = tomorrow
self.course = self.update_course(self.course) self.course = self.update_course(self.course, self.user.id)
self.test_course = self.update_course(self.test_course) self.test_course = self.update_course(self.test_course, self.user.id)
self.assertFalse(self.course.has_started()) self.assertFalse(self.course.has_started())
self.assertFalse(self.test_course.has_started()) self.assertFalse(self.test_course.has_started())
...@@ -321,8 +322,8 @@ class TestViewAuth(ModuleStoreTestCase, LoginEnrollmentTestCase): ...@@ -321,8 +322,8 @@ class TestViewAuth(ModuleStoreTestCase, LoginEnrollmentTestCase):
tomorrow = now + datetime.timedelta(days=1) tomorrow = now + datetime.timedelta(days=1)
self.course.start = tomorrow self.course.start = tomorrow
self.test_course.start = tomorrow self.test_course.start = tomorrow
self.course = self.update_course(self.course) self.course = self.update_course(self.course, self.user.id)
self.test_course = self.update_course(self.test_course) self.test_course = self.update_course(self.test_course, self.user.id)
self.login(self.instructor_user) self.login(self.instructor_user)
# Enroll in the classes---can't see courseware otherwise. # Enroll in the classes---can't see courseware otherwise.
...@@ -345,8 +346,8 @@ class TestViewAuth(ModuleStoreTestCase, LoginEnrollmentTestCase): ...@@ -345,8 +346,8 @@ class TestViewAuth(ModuleStoreTestCase, LoginEnrollmentTestCase):
self.course.start = tomorrow self.course.start = tomorrow
self.test_course.start = tomorrow self.test_course.start = tomorrow
self.course = self.update_course(self.course) self.course = self.update_course(self.course, self.user.id)
self.test_course = self.update_course(self.test_course) self.test_course = self.update_course(self.test_course, self.user.id)
self.login(self.global_staff_user) self.login(self.global_staff_user)
self.enroll(self.course, True) self.enroll(self.course, True)
...@@ -373,8 +374,8 @@ class TestViewAuth(ModuleStoreTestCase, LoginEnrollmentTestCase): ...@@ -373,8 +374,8 @@ class TestViewAuth(ModuleStoreTestCase, LoginEnrollmentTestCase):
# test_course course's has # test_course course's has
self.test_course.enrollment_start = yesterday self.test_course.enrollment_start = yesterday
self.test_course.enrollment_end = tomorrow self.test_course.enrollment_end = tomorrow
self.course = self.update_course(self.course) self.course = self.update_course(self.course, self.user.id)
self.test_course = self.update_course(self.test_course) self.test_course = self.update_course(self.test_course, self.user.id)
# First, try with an enrolled student # First, try with an enrolled student
self.login(self.unenrolled_user) self.login(self.unenrolled_user)
......
...@@ -142,9 +142,8 @@ class TestMongoCoursesLoad(ModuleStoreTestCase, PageLoaderTestCase): ...@@ -142,9 +142,8 @@ class TestMongoCoursesLoad(ModuleStoreTestCase, PageLoaderTestCase):
super(TestMongoCoursesLoad, self).setUp() super(TestMongoCoursesLoad, self).setUp()
self.setup_user() self.setup_user()
# Import the toy course into a Mongo-backed modulestore # Import the toy course
self.store = modulestore() import_from_xml(self.store, self.user.id, TEST_DATA_DIR, ['toy'])
import_from_xml(self.store, "**replace_user**", TEST_DATA_DIR, ['toy'])
@mock.patch('xmodule.course_module.requests.get') @mock.patch('xmodule.course_module.requests.get')
def test_toy_textbooks_loads(self, mock_get): def test_toy_textbooks_loads(self, mock_get):
......
...@@ -54,7 +54,7 @@ class SysadminBaseTestCase(ModuleStoreTestCase): ...@@ -54,7 +54,7 @@ class SysadminBaseTestCase(ModuleStoreTestCase):
def setUp(self): def setUp(self):
"""Setup test case by adding primary user.""" """Setup test case by adding primary user."""
super(SysadminBaseTestCase, self).setUp() super(SysadminBaseTestCase, self).setUp(create_user=False)
self.user = UserFactory.create(username='test_user', self.user = UserFactory.create(username='test_user',
email='test_user+sysadmin@edx.org', email='test_user+sysadmin@edx.org',
password='foo') password='foo')
...@@ -129,6 +129,9 @@ class TestSysadmin(SysadminBaseTestCase): ...@@ -129,6 +129,9 @@ class TestSysadmin(SysadminBaseTestCase):
response = self.client.get(reverse(view)) response = self.client.get(reverse(view))
self.assertEqual(response.status_code, 302) self.assertEqual(response.status_code, 302)
self.user.is_staff = False
self.user.save()
logged_in = self.client.login(username=self.user.username, logged_in = self.client.login(username=self.user.username,
password='foo') password='foo')
self.assertTrue(logged_in) self.assertTrue(logged_in)
......
...@@ -41,7 +41,7 @@ class ViewsTestCase(UrlResetMixin, ModuleStoreTestCase, MockRequestSetupMixin): ...@@ -41,7 +41,7 @@ class ViewsTestCase(UrlResetMixin, ModuleStoreTestCase, MockRequestSetupMixin):
# Patching the ENABLE_DISCUSSION_SERVICE value affects the contents of urls.py, # Patching the ENABLE_DISCUSSION_SERVICE value affects the contents of urls.py,
# so we need to call super.setUp() which reloads urls.py (because # so we need to call super.setUp() which reloads urls.py (because
# of the UrlResetMixin) # of the UrlResetMixin)
super(ViewsTestCase, self).setUp() super(ViewsTestCase, self).setUp(create_user=False)
# create a course # create a course
self.course = CourseFactory.create(org='MITx', course='999', self.course = CourseFactory.create(org='MITx', course='999',
......
...@@ -208,7 +208,7 @@ class InstructorTaskModuleTestCase(InstructorTaskCourseTestCase): ...@@ -208,7 +208,7 @@ class InstructorTaskModuleTestCase(InstructorTaskCourseTestCase):
location = InstructorTaskTestCase.problem_location(problem_url_name) location = InstructorTaskTestCase.problem_location(problem_url_name)
item = self.module_store.get_item(location) item = self.module_store.get_item(location)
item.data = problem_xml item.data = problem_xml
self.module_store.update_item(item, '**replace_user**') self.module_store.update_item(item, self.user.id)
def get_student_module(self, username, descriptor): def get_student_module(self, username, descriptor):
"""Get StudentModule object for test course, given the `username` and the problem's `descriptor`.""" """Get StudentModule object for test course, given the `username` and the problem's `descriptor`."""
......
...@@ -105,6 +105,9 @@ class TestRescoringTask(TestIntegrationTask): ...@@ -105,6 +105,9 @@ class TestRescoringTask(TestIntegrationTask):
self.create_student('u4') self.create_student('u4')
self.logout() self.logout()
# set up test user for performing test operations
self.setup_user()
def render_problem(self, username, problem_url_name): def render_problem(self, username, problem_url_name):
""" """
Use ajax interface to request html for a problem. Use ajax interface to request html for a problem.
...@@ -294,7 +297,7 @@ class TestRescoringTask(TestIntegrationTask): ...@@ -294,7 +297,7 @@ class TestRescoringTask(TestIntegrationTask):
InstructorTaskModuleTestCase.problem_location(problem_url_name) InstructorTaskModuleTestCase.problem_location(problem_url_name)
) )
descriptor.data = problem_xml descriptor.data = problem_xml
self.module_store.update_item(descriptor, '**replace_user**') self.module_store.update_item(descriptor, self.user.id)
else: else:
# Use "per-student" rerandomization so that check-problem can be called more than once. # Use "per-student" rerandomization so that check-problem can be called more than once.
# Using "always" means we cannot check a problem twice, but we want to call once to get the # Using "always" means we cannot check a problem twice, but we want to call once to get the
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment