def setUp(self):
# Make some mock subjects.
subjects_utils.subject_enroll(testing.mock_subjects)
# Make a mock git project.
self.mock_repository_setup_dir\
= testing_utils.make_mock_repository_files()
self.mock_repository\
= testing_utils.MockExperimentRepository(
setup_dir = self.mock_repository_setup_dir)
self.mock_repository.initialize()
self.number_of_revisions = 10
self.mock_repository.update(revisions=self.number_of_revisions)
# Make a repository, archives, etc for the experiments.
setup_fields = dict(name = 'mock',
description = 'A mock repository',
date_created = datetime.datetime.now(),
is_active = True,
path = self.mock_repository.path)
self.experiment_repository = archives_models.ExperimentRepository\
.objects.create(**setup_fields)
self.experiment_repository.make_archives()
# Set current version to default.
archives_models.Experiment.objects.set_default_current_version()
def handle(self, *args, **options):
try:
for arg in args:
subjects_bulk_enroll_file = configobj.ConfigObj(arg)
utils.subject_enroll(subjects_bulk_enroll_file)
except:
CommandError('Enroll did not work.')
def setUp(self):
'''
We will use the testing GenericSetup to create the mock
ExperimentRepository, Experiment and ExperimentVersion models.
In addition, we will create mock subjects, and then create a mock
experiment session for an arbitrarily chosen mock subject and mock
experiment. Finally, we create a live experiment session for this
experiment session.
'''
self.generic_setup = testing_utils.GenericSetup()
for attr in ('mock_repository', 'mock_repository_setup_dir'):
setattr(self, attr, getattr(self.generic_setup, attr))
subjects_utils.subject_enroll(testing.mock_subjects)
# FIXME. How about randomly choosing a subject and experiment.
self.subject_name = choice(testing.mock_subjects.keys())
self.subject = subjects_models.Subject.objects.get(
user__username = self.subject_name)
self.experiment_name = 'Rusty'
self.experiment = archives_models.Experiment.objects.get(
class_name = self.experiment_name
)
experiment_session\
= session_models.ExperimentSession.new(self.subject,
self.experiment_name)
self.experiment_session_uid = experiment_session.uid
live_session = presenter_models.LiveExperimentSession.new(
experiment_session)
self.live_session_uid = live_session.uid
def enroll_mock_subjects():
return subject_enroll(mock_subjects)
