def testQuestionChoiceInsert(self):
survey_id = connection.execute(
survey_table.select().where(
survey_table.c.survey_title == 'test_title'
)).first().survey_id
seq_number = get_free_sequence_number(connection, survey_id)
stmt = question_insert(hint=None, allow_multiple=None,
logic={
'required': False,
'allow_other': False,
'allow_dont_know': False
},
sequence_number=seq_number,
question_title='test choice',
type_constraint_name='multiple_choice',
question_to_sequence_number=-1,
survey_id=survey_id)
question_id = connection.execute(stmt).inserted_primary_key[0]
c_stmt = question_choice_insert(question_id=question_id,
choice='test choice',
choice_number=1,
type_constraint_name='multiple_choice',
question_sequence_number=seq_number,
allow_multiple=False,
survey_id=survey_id)
choice_id = connection.execute(c_stmt).inserted_primary_key[0]
cond = question_choice_table.c.question_id == question_id
self.assertEqual(connection.execute(
question_choice_table.select().where(cond)
).first().question_choice_id, choice_id)
def _create_choices(connection: Connection,
values: dict,
question_id: str,
submission_map: dict,
existing_question_id: str=None) -> Iterator:
"""
Create the choices of a survey question. If this is an update to an
existing survey, it will also copy over answers to the questions.
:param connection: the SQLAlchemy Connection object for the transaction
:param values: the dictionary of values associated with the question
:param question_id: the UUID of the question
:param submission_map: a dictionary mapping old submission_id to new
:param existing_question_id: the UUID of the existing question (if this is
an update)
:return: an iterable of the resultant choice fields
"""
choices = values['choices']
new_choices, updates = _determine_choices(connection, existing_question_id,
choices)
for number, choice in enumerate(new_choices):
choice_dict = {
'question_id': question_id,
'survey_id': values['survey_id'],
'choice': choice,
'choice_number': number,
'type_constraint_name': values['type_constraint_name'],
'question_sequence_number': values['sequence_number'],
'allow_multiple': values['allow_multiple']}
executable = question_choice_insert(**choice_dict)
exc = [('unique_choice_names', RepeatedChoiceError(choice))]
result = execute_with_exceptions(connection, executable, exc)
result_ipk = result.inserted_primary_key
question_choice_id = result_ipk[0]
if choice in updates:
question_fields = {'question_id': question_id,
'type_constraint_name': result_ipk[2],
'sequence_number': result_ipk[3],
'allow_multiple': result_ipk[4],
'survey_id': values['survey_id']}
for answer in get_answer_choices_for_choice_id(connection,
updates[choice]):
answer_values = question_fields.copy()
new_submission_id = submission_map[answer.submission_id]
answer_values['question_choice_id'] = question_choice_id
answer_values['submission_id'] = new_submission_id
answer_metadata = answer.answer_choice_metadata
answer_values['answer_choice_metadata'] = answer_metadata
connection.execute(answer_choice_insert(**answer_values))
yield question_choice_id
