def handle(self, *args, **options):
file_name = options['load_surveys'][0]
absolute_file_name = os.path.join(str(root), file_name)
with open(absolute_file_name, 'r') as survey_to_load:
loader = SurveyLoader(survey_to_load.read())
loader.load_survey()
def load_survey_test(self):
loader = SurveyLoader(sample_survey)
loader.load_survey()
all_surveys = Survey.objects.all()
all_questions = Question.objects.all()
assert len(all_surveys) == 1
assert len(all_questions) == 5
assert all_surveys.first().title == 'About bears'
with pytest.raises(KeyError):
loader = SurveyLoader(broken_survey)
loader.load_survey()