def test_serialise_relationship_answers(self): location = Location("household-relationships", 0, "relationships") field_data = ['Husband or Wife', 'Son or daughter', 'Unrelated'] actual_answers = serialise_relationship_answers( location, 'who-is-related', field_data) expected_answers = [{ 'group_id': "household-relationships", 'group_instance': 0, 'block_id': 'relationships', 'answer_id': 'who-is-related', 'answer_instance': 0, 'value': 'Husband or Wife' }, { 'group_id': "household-relationships", 'group_instance': 0, 'block_id': 'relationships', 'answer_id': 'who-is-related', 'answer_instance': 1, 'value': 'Son or daughter' }, { 'group_id': "household-relationships", 'group_instance': 0, 'block_id': 'relationships', 'answer_id': 'who-is-related', 'answer_instance': 2, 'value': 'Unrelated' }] for answer in actual_answers: self.assertIn(answer.__dict__, expected_answers)
def test_serialise_relationship_answers_second_group(self): field_data = ['Husband or Wife', 'Son or daughter', 'Unrelated'] actual_answers = serialise_relationship_answers( 'who-is-related', field_data, 1, 'group-1') expected_answers = [{ 'group_instance_id': 'group-1', 'group_instance': 1, 'answer_id': 'who-is-related', 'answer_instance': 0, 'value': 'Husband or Wife' }, { 'group_instance_id': 'group-1', 'group_instance': 1, 'answer_id': 'who-is-related', 'answer_instance': 1, 'value': 'Son or daughter' }, { 'group_instance_id': 'group-1', 'group_instance': 1, 'answer_id': 'who-is-related', 'answer_instance': 2, 'value': 'Unrelated' }] for answer in actual_answers: self.assertIn(answer.__dict__, expected_answers)