def test_all_present(self): """Ensure that we work properly when all expected fields are present.""" raw_predator_result = {} raw_predator_result['result'] = { 'found': True, 'suspected_project': 'suspected-project', 'suspected_components': ['A', 'B>C'], 'suspected_cls': ['dummy cl'], 'feedback_url': 'https://feedback/', 'error_message': 'error string', 'unused': 'unused', } # Note that this doesn't contain the "unused" field. expected_parsed_result = { 'found': True, 'suspected_project': 'suspected-project', 'suspected_components': ['A', 'B>C'], 'changelists': ['dummy cl'], 'feedback_url': 'https://feedback/', 'error_message': 'error string', } self.assertDictEqual(show._parse_suspected_cls(raw_predator_result), expected_parsed_result)
def test_all_missing(self): """Ensure that we work properly if no expected fields are present.""" raw_predator_result = {'result': {}} expected_parsed_result = { 'found': None, 'suspected_project': None, 'suspected_components': None, 'changelists': None, 'feedback_url': None, 'error_message': None, } self.assertDictEqual(show._parse_suspected_cls(raw_predator_result), expected_parsed_result)
def test_empty(self): """Test when we have no predator result.""" self.assertIsNone(show._parse_suspected_cls(None))