def test_get_all_profiles(self): session = mock.Mock() session.pbm = mock.Mock() profile_ids = mock.Mock() def invoke_api_side_effect(module, method, *args, **kwargs): self.assertEqual(session.pbm, module) self.assertIn(method, ['PbmQueryProfile', 'PbmRetrieveContent']) self.assertEqual(session.pbm.service_content.profileManager, args[0]) if method == 'PbmQueryProfile': self.assertEqual('STORAGE', kwargs['resourceType'].resourceType) return profile_ids self.assertEqual(profile_ids, kwargs['profileIds']) session.invoke_api.side_effect = invoke_api_side_effect pbm.get_all_profiles(session) self.assertEqual(2, session.invoke_api.call_count)
def test_get_all_profiles_with_no_profiles(self): session = mock.Mock() session.pbm = mock.Mock() session.invoke_api.return_value = [] profiles = pbm.get_all_profiles(session) session.invoke_api.assert_called_once_with( session.pbm, 'PbmQueryProfile', session.pbm.service_content.profileManager, resourceType=session.pbm.client.factory.create()) self.assertEqual([], profiles)