示例#1
0
    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)
示例#2
0
    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)
示例#3
0
 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)
示例#4
0
 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)