def test_empty_answers(self):
string_answer_id = "string-answer"
checkbox_answer_id = "checkbox-answer"
radio_answer_id = "radio-answer"
self.schema.get_answer_ids_for_question.return_value = [
string_answer_id,
checkbox_answer_id,
radio_answer_id,
]
form_data = {
string_answer_id: "",
checkbox_answer_id: [],
radio_answer_id: None,
}
form = MagicMock(spec=QuestionnaireForm, data=form_data)
self.current_question = {
"answers": [
{"id": "string-answer"},
{"id": "checkbox-answer"},
{"id": "radio-answer"},
]
}
self.questionnaire_store_updater = QuestionnaireStoreUpdater(
self.location, self.schema, self.questionnaire_store, self.current_question
)
self.questionnaire_store_updater.update_answers(form)
assert self.answer_store.add_or_update.call_count == 0
def test_remove_completed_relationship_locations_for_list_name_no_locations(self):
self.progress_store.add_completed_location(
"section",
Location(section_id="section", block_id="test-relationship-collector"),
)
initial_progress_store = self.progress_store.serialise()
list_store = fake_list_store()
questionnaire_store = MagicMock(
spec=QuestionnaireStore,
completed_blocks=[],
answer_store=self.answer_store,
list_store=list_store,
progress_store=self.progress_store,
)
questionnaire_store_updater = QuestionnaireStoreUpdater(
self.location, self.schema, questionnaire_store, self.current_question
)
patch_method = "app.questionnaire.questionnaire_store_updater.QuestionnaireStoreUpdater._get_relationship_collectors_by_list_name"
with patch(patch_method) as patched:
patched.return_value = None
questionnaire_store_updater.remove_completed_relationship_locations_for_list_name(
"test-relationship-collector"
)
self.assertEqual(self.progress_store.serialise(), initial_progress_store)
def test_remove_primary_person(self):
answer_store = AnswerStore(existing_answers=[
{
"answer_id": "test1",
"value": 1,
"list_item_id": "abcdef"
},
{
"answer_id": "test2",
"value": 2,
"list_item_id": "abcdef"
},
{
"answer_id": "test3",
"value": 3,
"list_item_id": "xyzabc"
},
])
list_store = fake_list_store()
questionnaire_store = MagicMock(
spec=QuestionnaireStore,
completed_blocks=[],
answer_store=answer_store,
list_store=list_store,
progress_store=ProgressStore(),
)
questionnaire_store_updater = QuestionnaireStoreUpdater(
self.location, self.schema, questionnaire_store,
self.current_question)
questionnaire_store_updater.remove_primary_person("people")
def test_remove_all_answers_with_list_item_id(self):
answer_store = AnswerStore(existing_answers=[
{
"answer_id": "test1",
"value": 1,
"list_item_id": "abcdef"
},
{
"answer_id": "test2",
"value": 2,
"list_item_id": "abcdef"
},
{
"answer_id": "test3",
"value": 3,
"list_item_id": "uvwxyz"
},
])
questionnaire_store = MagicMock(
spec=QuestionnaireStore,
completed_blocks=[],
answer_store=answer_store,
list_store=MagicMock(spec=ListStore),
progress_store=ProgressStore(),
)
self.questionnaire_store_updater = QuestionnaireStoreUpdater(
self.location, self.schema, questionnaire_store,
self.current_question)
self.questionnaire_store_updater.remove_list_item_and_answers(
"abc", "abcdef")
assert len(answer_store) == 1
assert answer_store.get_answer("test3", "uvwxyz")
def test_default_answers_are_not_saved(self):
answer_id = "answer"
default_value = 0
self.schema.get_answer_ids_for_question.return_value = [answer_id]
self.schema.get_answers_by_answer_id.return_value = [{
"default":
default_value
}]
# No answer given so will use schema defined default
form_data = MultiDict({answer_id: None})
self.current_question = {
"answers": [{
"id": "answer",
"default": default_value
}]
}
self.questionnaire_store_updater = QuestionnaireStoreUpdater(
self.location, self.schema, self.questionnaire_store,
self.current_question)
self.questionnaire_store_updater.update_answers(form_data)
assert self.answer_store.add_or_update.call_count == 0
def test_add_primary_person(self):
list_store = fake_list_store()
questionnaire_store = MagicMock(
spec=QuestionnaireStore,
completed_blocks=[],
answer_store=self.answer_store,
list_store=list_store,
progress_store=ProgressStore(),
)
questionnaire_store_updater = QuestionnaireStoreUpdater(
self.location, self.schema, questionnaire_store, self.current_question
)
questionnaire_store_updater.add_primary_person("people")
def questionnaire_store_updater(self):
return QuestionnaireStoreUpdater(
self._current_location,
self._schema,
self._questionnaire_store,
self.block.get("question"),
)
def questionnaire_store_updater(self):
if not self._questionnaire_store_updater:
self._questionnaire_store_updater = QuestionnaireStoreUpdater(
self._current_location,
self._schema,
self._questionnaire_store,
self.rendered_block.get("question"),
)
return self._questionnaire_store_updater
def test_update_relationship_question_completeness_no_relationship_collectors(self):
list_store = fake_list_store()
questionnaire_store = MagicMock(
spec=QuestionnaireStore,
completed_blocks=[],
answer_store=self.answer_store,
list_store=list_store,
progress_store=self.progress_store,
)
questionnaire_store_updater = QuestionnaireStoreUpdater(
self.location, self.schema, questionnaire_store, self.current_question
)
patch_method = "app.questionnaire.questionnaire_store_updater.QuestionnaireStoreUpdater._get_relationship_collectors_by_list_name"
with patch(patch_method) as patched:
patched.return_value = None
self.assertIsNone(
questionnaire_store_updater.update_relationship_question_completeness(
"test-relationship-collector"
)
)
def test_save_empty_answer_removes_existing_answer(self):
answer_id = "answer"
answer_value = "1000"
list_item_id = "abc123"
location = Location(
section_id="section-foo",
block_id="block-bar",
list_name="people",
list_item_id=list_item_id,
)
self.schema.get_answer_ids_for_question.return_value = [answer_id]
form_data = MultiDict({answer_id: answer_value})
self.current_question = self.schema.get_block(
location.block_id)["question"]
self.questionnaire_store_updater = QuestionnaireStoreUpdater(
location, self.schema, self.questionnaire_store,
self.current_question)
self.questionnaire_store_updater.update_answers(form_data)
assert self.answer_store.add_or_update.call_count == 1
created_answer = self.answer_store.add_or_update.call_args[0][0]
assert created_answer.__dict__ == {
"answer_id": answer_id,
"list_item_id": "abc123",
"value": answer_value,
}
form_data = MultiDict({answer_id: ""})
self.questionnaire_store_updater.update_answers(form_data)
assert self.answer_store.remove_answer.call_count == 1
answer_key = self.answer_store.remove_answer.call_args[0]
assert answer_key == (answer_id, list_item_id)
def test_save_answers_with_form_data(self):
answer_id = "answer"
answer_value = "1000"
self.schema.get_answer_ids_for_question.return_value = [answer_id]
form_data = {answer_id: answer_value}
self.current_question = self.schema.get_block(
self.location.block_id)["question"]
self.questionnaire_store_updater = QuestionnaireStoreUpdater(
self.location, self.schema, self.questionnaire_store,
self.current_question)
self.questionnaire_store_updater.update_answers(form_data)
assert self.answer_store.add_or_update.call_count == 1
created_answer = self.answer_store.add_or_update.call_args[0][0]
assert created_answer.__dict__ == {
"answer_id": answer_id,
"list_item_id": None,
"value": answer_value,
}
class TestQuestionnaireStoreUpdater(unittest.TestCase):
def setUp(self):
super().setUp()
self.location = Location(section_id="section-foo", block_id="block-bar")
self.schema = MagicMock(spec=QuestionnaireSchema)
self.answer_store = MagicMock(spec=AnswerStore)
self.progress_store = MagicMock(spec=ProgressStore)
self.progress_store.locations = list()
self.list_store = MagicMock(spec=ListStore)
self.questionnaire_store = MagicMock(
spec=QuestionnaireStore,
completed_blocks=[],
answer_store=self.answer_store,
list_store=self.list_store,
progress_store=self.progress_store,
)
self.metadata = MagicMock()
self.questionnaire_store_updater = None
self.current_question = None
def test_save_answers_with_form_data(self):
answer_id = "answer"
answer_value = "1000"
self.schema.get_answer_ids_for_question.return_value = [answer_id]
form = MagicMock(spec=QuestionnaireForm, data={answer_id: answer_value})
self.current_question = self.schema.get_block(self.location.block_id)[
"question"
]
self.questionnaire_store_updater = QuestionnaireStoreUpdater(
self.location, self.schema, self.questionnaire_store, self.current_question
)
self.questionnaire_store_updater.update_answers(form)
assert self.answer_store.add_or_update.call_count == 1
created_answer = self.answer_store.add_or_update.call_args[0][0]
assert created_answer.__dict__ == {
"answer_id": answer_id,
"list_item_id": None,
"value": answer_value,
}
def test_save_empty_answer_removes_existing_answer(self):
answer_id = "answer"
answer_value = "1000"
list_item_id = "abc123"
location = Location(
section_id="section-foo",
block_id="block-bar",
list_name="people",
list_item_id=list_item_id,
)
self.schema.get_answer_ids_for_question.return_value = [answer_id]
form = MagicMock(spec=QuestionnaireForm, data={answer_id: answer_value})
self.current_question = self.schema.get_block(location.block_id)["question"]
self.questionnaire_store_updater = QuestionnaireStoreUpdater(
location, self.schema, self.questionnaire_store, self.current_question
)
self.questionnaire_store_updater.update_answers(form)
assert self.answer_store.add_or_update.call_count == 1
created_answer = self.answer_store.add_or_update.call_args[0][0]
assert created_answer.__dict__ == {
"answer_id": answer_id,
"list_item_id": "abc123",
"value": answer_value,
}
form = MagicMock(spec=QuestionnaireForm, data={answer_id: ""})
self.questionnaire_store_updater.update_answers(form)
assert self.answer_store.remove_answer.call_count == 1
answer_key = self.answer_store.remove_answer.call_args[0]
assert answer_key == (answer_id, list_item_id)
def test_default_answers_are_not_saved(self):
answer_id = "answer"
default_value = 0
self.schema.get_answer_ids_for_question.return_value = [answer_id]
self.schema.get_answers_by_answer_id.return_value = [{"default": default_value}]
# No answer given so will use schema defined default
form_data = {answer_id: None}
form = MagicMock(spec=QuestionnaireForm, data=form_data)
self.current_question = {
"answers": [{"id": "answer", "default": default_value}]
}
self.questionnaire_store_updater = QuestionnaireStoreUpdater(
self.location, self.schema, self.questionnaire_store, self.current_question
)
self.questionnaire_store_updater.update_answers(form)
assert self.answer_store.add_or_update.call_count == 0
def test_empty_answers(self):
string_answer_id = "string-answer"
checkbox_answer_id = "checkbox-answer"
radio_answer_id = "radio-answer"
self.schema.get_answer_ids_for_question.return_value = [
string_answer_id,
checkbox_answer_id,
radio_answer_id,
]
form_data = {
string_answer_id: "",
checkbox_answer_id: [],
radio_answer_id: None,
}
form = MagicMock(spec=QuestionnaireForm, data=form_data)
self.current_question = {
"answers": [
{"id": "string-answer"},
{"id": "checkbox-answer"},
{"id": "radio-answer"},
]
}
self.questionnaire_store_updater = QuestionnaireStoreUpdater(
self.location, self.schema, self.questionnaire_store, self.current_question
)
self.questionnaire_store_updater.update_answers(form)
assert self.answer_store.add_or_update.call_count == 0
def test_remove_all_answers_with_list_item_id(self):
answer_store = AnswerStore(
existing_answers=[
{"answer_id": "test1", "value": 1, "list_item_id": "abcdef"},
{"answer_id": "test2", "value": 2, "list_item_id": "abcdef"},
{"answer_id": "test3", "value": 3, "list_item_id": "uvwxyz"},
]
)
questionnaire_store = MagicMock(
spec=QuestionnaireStore,
completed_blocks=[],
answer_store=answer_store,
list_store=MagicMock(spec=ListStore),
progress_store=ProgressStore(),
)
self.questionnaire_store_updater = QuestionnaireStoreUpdater(
self.location, self.schema, questionnaire_store, self.current_question
)
self.questionnaire_store_updater.remove_list_item_and_answers("abc", "abcdef")
assert len(answer_store) == 1
assert answer_store.get_answer("test3", "uvwxyz")
def test_remove_primary_person(self):
answer_store = AnswerStore(
existing_answers=[
{"answer_id": "test1", "value": 1, "list_item_id": "abcdef"},
{"answer_id": "test2", "value": 2, "list_item_id": "abcdef"},
{"answer_id": "test3", "value": 3, "list_item_id": "xyzabc"},
]
)
list_store = fake_list_store()
questionnaire_store = MagicMock(
spec=QuestionnaireStore,
completed_blocks=[],
answer_store=answer_store,
list_store=list_store,
progress_store=ProgressStore(),
)
questionnaire_store_updater = QuestionnaireStoreUpdater(
self.location, self.schema, questionnaire_store, self.current_question
)
questionnaire_store_updater.remove_primary_person("people")
def test_add_primary_person(self):
list_store = fake_list_store()
questionnaire_store = MagicMock(
spec=QuestionnaireStore,
completed_blocks=[],
answer_store=self.answer_store,
list_store=list_store,
progress_store=ProgressStore(),
)
questionnaire_store_updater = QuestionnaireStoreUpdater(
self.location, self.schema, questionnaire_store, self.current_question
)
questionnaire_store_updater.add_primary_person("people")
def test_remove_completed_relationship_locations_for_list_name(self):
list_store = fake_list_store()
self.progress_store.add_completed_location(
"section",
Location(section_id="section", block_id="test-relationship-collector"),
)
questionnaire_store = MagicMock(
spec=QuestionnaireStore,
completed_blocks=[],
answer_store=self.answer_store,
list_store=list_store,
progress_store=self.progress_store,
)
questionnaire_store_updater = QuestionnaireStoreUpdater(
self.location, self.schema, questionnaire_store, self.current_question
)
patch_method = "app.questionnaire.questionnaire_store_updater.QuestionnaireStoreUpdater._get_relationship_collectors_by_list_name"
with patch(patch_method) as patched:
patched.return_value = [{"id": "test-relationship-collector"}]
questionnaire_store_updater.remove_completed_relationship_locations_for_list_name(
"test-relationship-collector"
)
completed = self.progress_store.serialise()
self.assertEqual(len(completed), 0)
def test_remove_completed_relationship_locations_for_list_name_no_locations(self):
self.progress_store.add_completed_location(
"section",
Location(section_id="section", block_id="test-relationship-collector"),
)
initial_progress_store = self.progress_store.serialise()
list_store = fake_list_store()
questionnaire_store = MagicMock(
spec=QuestionnaireStore,
completed_blocks=[],
answer_store=self.answer_store,
list_store=list_store,
progress_store=self.progress_store,
)
questionnaire_store_updater = QuestionnaireStoreUpdater(
self.location, self.schema, questionnaire_store, self.current_question
)
patch_method = "app.questionnaire.questionnaire_store_updater.QuestionnaireStoreUpdater._get_relationship_collectors_by_list_name"
with patch(patch_method) as patched:
patched.return_value = None
questionnaire_store_updater.remove_completed_relationship_locations_for_list_name(
"test-relationship-collector"
)
self.assertEqual(self.progress_store.serialise(), initial_progress_store)
def test_update_relationship_question_completeness_no_relationship_collectors(self):
list_store = fake_list_store()
questionnaire_store = MagicMock(
spec=QuestionnaireStore,
completed_blocks=[],
answer_store=self.answer_store,
list_store=list_store,
progress_store=self.progress_store,
)
questionnaire_store_updater = QuestionnaireStoreUpdater(
self.location, self.schema, questionnaire_store, self.current_question
)
patch_method = "app.questionnaire.questionnaire_store_updater.QuestionnaireStoreUpdater._get_relationship_collectors_by_list_name"
with patch(patch_method) as patched:
patched.return_value = None
self.assertIsNone(
questionnaire_store_updater.update_relationship_question_completeness(
"test-relationship-collector"
)
)
def test_update_same_name_items(self):
answer_store = AnswerStore(existing_answers=[
{
"answer_id": "first-name",
"value": "Joe",
"list_item_id": "abcdef"
},
{
"answer_id": "middle-name",
"value": "Brian",
"list_item_id": "abcdef",
},
{
"answer_id": "last-name",
"value": "Bloggs",
"list_item_id": "abcdef"
},
{
"answer_id": "first-name",
"value": "Joe",
"list_item_id": "ghijkl"
},
{
"answer_id": "middle-name",
"value": "Roger",
"list_item_id": "ghijkl",
},
{
"answer_id": "last-name",
"value": "Bloggs",
"list_item_id": "ghijkl"
},
{
"answer_id": "first-name",
"value": "Martha",
"list_item_id": "xyzabc",
},
{
"answer_id": "last-name",
"value": "Bloggs",
"list_item_id": "xyzabc"
},
])
list_store = fake_list_store()
questionnaire_store = MagicMock(
spec=QuestionnaireStore,
completed_blocks=[],
answer_store=answer_store,
list_store=list_store,
progress_store=ProgressStore(),
)
questionnaire_store_updater = QuestionnaireStoreUpdater(
self.location, self.schema, questionnaire_store,
self.current_question)
questionnaire_store_updater.update_same_name_items(
"people", ["first-name", "last-name"])
assert "abcdef" in list_store["people"].same_name_items
assert "ghijkl" in list_store["people"].same_name_items