def test__get_check_results_unknown(self):
msg = ('This package is not a whitelisted google python package; to '
'whitelist a package, contact the python team.')
expected_self_res = {
'py2': {
'status': main.BadgeStatus.UNKNOWN_PACKAGE,
'details': msg
},
'py3': {
'status': main.BadgeStatus.UNKNOWN_PACKAGE,
'details': msg
},
}
expected_google_res = {
'py2': {
'status': main.BadgeStatus.UNKNOWN_PACKAGE,
'details': {}
},
'py3': {
'status': main.BadgeStatus.UNKNOWN_PACKAGE,
'details': {}
},
}
expected_dep_res = {
'status': main.BadgeStatus.UNKNOWN_PACKAGE,
'details': {},
}
mock_self_res = mock.Mock()
mock_self_res.return_value = expected_self_res
mock_google_res = mock.Mock()
mock_google_res.return_value = expected_google_res
mock_dep_res = mock.Mock()
mock_dep_res.return_value = expected_dep_res
patch_self_res = mock.patch('main._get_self_compatibility_dict',
mock_self_res)
patch_google_res = mock.patch('main._get_pair_compatibility_dict',
mock_google_res)
patch_dep_res = mock.patch('main._get_dependency_dict', mock_dep_res)
with patch_self_res, patch_google_res, patch_dep_res:
self_res, google_res, dep_res = main._get_check_results(
'unknown_package')
status = main._get_badge_status(self_res, google_res, dep_res)
self.assertEqual(self_res, expected_self_res)
self.assertEqual(google_res, expected_google_res)
self.assertEqual(dep_res, expected_dep_res)
self.assertEqual(status, main.BadgeStatus.UNKNOWN_PACKAGE)
def test__get_check_results_internal_error(self):
expected_self_res = {
'py2': {
'status': main.BadgeStatus.INTERNAL_ERROR,
'details': {}
},
'py3': {
'status': main.BadgeStatus.INTERNAL_ERROR,
'details': {}
},
}
expected_google_res = {
'py2': {
'status': main.BadgeStatus.SUCCESS,
'details': {}
},
'py3': {
'status': main.BadgeStatus.SUCCESS,
'details': {}
},
}
expected_dep_res = {
'status': main.BadgeStatus.SUCCESS,
'details': {},
}
mock_self_res = mock.Mock()
mock_self_res.return_value = expected_self_res
mock_google_res = mock.Mock()
mock_google_res.return_value = expected_google_res
mock_dep_res = mock.Mock()
mock_dep_res.return_value = expected_dep_res
patch_self_res = mock.patch('main._get_self_compatibility_dict',
mock_self_res)
patch_google_res = mock.patch('main._get_pair_compatibility_dict',
mock_google_res)
patch_dep_res = mock.patch('main._get_dependency_dict', mock_dep_res)
with patch_self_res, patch_google_res, patch_dep_res:
self_res, google_res, dep_res = main._get_check_results(
'opencensus')
status = main._get_badge_status(self_res, google_res, dep_res)
self.assertEqual(self_res, expected_self_res)
self.assertEqual(google_res, expected_google_res)
self.assertEqual(dep_res, expected_dep_res)
self.assertEqual(status, main.BadgeStatus.INTERNAL_ERROR)