53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
"""
|
|
Utility methods for unit tests.
|
|
"""
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import filecmp
|
|
|
|
from path import Path as path
|
|
from xblock.reference.user_service import UserService, XBlockUser
|
|
|
|
|
|
def directories_equal(directory1, directory2):
|
|
"""
|
|
Returns True if the 2 directories have equal content, else false.
|
|
"""
|
|
def compare_dirs(dir1, dir2):
|
|
""" Compare directories for equality. """
|
|
comparison = filecmp.dircmp(dir1, dir2)
|
|
if (len(comparison.left_only) > 0) or (len(comparison.right_only) > 0):
|
|
return False
|
|
if (len(comparison.funny_files) > 0) or (len(comparison.diff_files) > 0):
|
|
return False
|
|
for subdir in comparison.subdirs:
|
|
if not compare_dirs(dir1 / subdir, dir2 / subdir):
|
|
return False
|
|
|
|
return True
|
|
|
|
return compare_dirs(path(directory1), path(directory2))
|
|
|
|
|
|
class StubUserService(UserService):
|
|
"""
|
|
Stub UserService for testing the sequence module.
|
|
"""
|
|
|
|
def __init__(self, is_anonymous=False, **kwargs):
|
|
self.is_anonymous = is_anonymous
|
|
super(StubUserService, self).__init__(**kwargs)
|
|
|
|
def get_current_user(self):
|
|
"""
|
|
Implements abstract method for getting the current user.
|
|
"""
|
|
user = XBlockUser()
|
|
if self.is_anonymous:
|
|
user.opt_attrs['edx-platform.username'] = 'anonymous'
|
|
user.opt_attrs['edx-platform.is_authenticated'] = False
|
|
else:
|
|
user.opt_attrs['edx-platform.username'] = 'bilbo'
|
|
return user
|