def test_questionnaire_store_updates_storage(self):
# Given
expected = get_basic_input()
store = QuestionnaireStore(self.storage)
store.set_metadata(expected["METADATA"])
store.answer_store = AnswerStore(expected["ANSWERS"])
store.collection_metadata = expected["COLLECTION_METADATA"]
store.progress_store = ProgressStore(expected["PROGRESS"])
# When
store.save() # See setUp - populates self.output_data
# Then
self.assertEqual(expected, json.loads(self.output_data))
def fake_questionnaire_store(fake_metadata, fake_collection_metadata):
user_answer = Answer(answer_id="GHI", value=0, list_item_id=None)
storage = MagicMock()
storage.get_user_data = MagicMock(return_value=("{}", 1))
storage.add_or_update = MagicMock()
store = QuestionnaireStore(storage)
store.answer_store = AnswerStore()
store.answer_store.add_or_update(user_answer)
store.metadata = fake_metadata
store.collection_metadata = fake_collection_metadata
return store
def test_questionnaire_store_errors_on_invalid_object(self):
# Given
class NotSerializable:
pass
non_serializable_metadata = {"test": NotSerializable()}
expected = get_basic_input()
store = QuestionnaireStore(self.storage)
store.set_metadata(non_serializable_metadata)
store.collection_metadata = expected["COLLECTION_METADATA"]
store.answer_store = AnswerStore(expected["ANSWERS"])
store.progress_store = ProgressStore(expected["PROGRESS"])
# When / Then
self.assertRaises(TypeError, store.save)
def test_questionnaire_store_deletes(self):
# Given
expected = get_basic_input()
store = QuestionnaireStore(self.storage)
store.set_metadata(expected["METADATA"])
store.collection_metadata = expected["COLLECTION_METADATA"]
store.answer_store = AnswerStore(expected["ANSWERS"])
store.progress_store = ProgressStore(expected["PROGRESS"])
# When
store.delete() # See setUp - populates self.output_data
# Then
self.assertNotIn("a-test-section", store.progress_store)
self.assertEqual(store.metadata.copy(), {})
self.assertEqual(len(store.answer_store), 0)
self.assertEqual(store.collection_metadata, {})