예제 #1
0
    def test_server_response(self, mock_g, mock_n, mock_k, mock_session_post):

        # Mock OpenStack clients and response to POST
        mock_image = {"name": test_image_name, "id": test_image_id}
        mock_gclient = Mock()
        mock_gclient.images = Mock()
        mock_im_list = Mock(return_value=[mock_image])
        mock_gclient.images.list = mock_im_list
        mock_gclient.image_members = Mock()
        mock_gclient.image_members.create = Mock()
        mock_g.return_value = mock_gclient

        mock_flavor = Mock()
        mock_flavor.name = test_flavor_name
        mock_flavor.id = test_flavor_id
        mock_fl_list = Mock(return_value=[mock_flavor])
        mock_nclient = Mock()
        mock_nclient.flavors = Mock()
        mock_nclient.flavors.list = mock_fl_list
        mock_n.return_value = mock_nclient

        mock_response = Mock()
        mock_response.text = test_job_id
        mock_response.status_code = 200
        mock_session_post.return_value = mock_response

        c = ProcClient(test_auth, host=test_host)

        # Failure on server side when trying to validate job
        mock_response_validate = Mock()
        mock_response_validate.text = "Failed to validate job."
        mock_response_validate.status_code = 400
        mock_response_validate.raise_for_status = Mock(side_effect=Exception("Failed."))
        mock_session_post.return_value = mock_response_validate

        with self.assertRaises(Exception):
            c.submit_job(test_job_works, test_image_name, test_flavor_name)
        mock_session_post.assert_called_once_with(test_job_url, params=test_params_works, files=test_files_works)

        mock_session_post.return_value = mock_response
        mock_session_post.reset_mock()

        # glance client needs to be refreshed. First time it tries to obtain
        # an image list it fails. Second time it works.
        mock_gclient.images.list.side_effect = [Exception("Failed"), mock_im_list()]
        mock_gclient.images.list.reset_mock()
        mock_g.reset_mock()
        c.submit_job(test_job_works, test_image_name, test_flavor_name)
        self.assertEqual(mock_gclient.images.list.call_count, 2)
        # only one additional call to glance client constructor since it
        # was originally created with the ProcClient instance
        self.assertEqual(mock_g.call_count, 1)
        mock_session_post.assert_called_once_with(test_job_url, params=test_params_works, files=test_files_works)

        mock_gclient.images.list.side_effect = None
        mock_session_post.reset_mock()

        # nova client needs to be refreshed. First time it tries to obtain
        # a flavor list it fails. Second time it works.
        mock_nclient.flavors.list.side_effect = [Exception("Failed"), mock_fl_list()]
        mock_nclient.flavors.list.reset_mock()
        mock_n.reset_mock()
        c.submit_job(test_job_works, test_image_name, test_flavor_name)
        self.assertEqual(mock_nclient.flavors.list.call_count, 2)
        self.assertEqual(mock_n.call_count, 1)
        mock_session_post.assert_called_once_with(test_job_url, params=test_params_works, files=test_files_works)

        mock_nclient.flavors.list.side_effect = None
        mock_session_post.reset_mock()

        # glance client needs to be refreshed. First time it tries to
        # share an image it fails. Second time it works.
        mock_gclient.image_members.create.side_effect = [Exception("Failed"), Mock()]
        mock_gclient.image_members.create.reset_mock()
        mock_g.reset_mock()
        c.submit_job(test_job_works, test_image_name, test_flavor_name)
        self.assertEqual(mock_gclient.image_members.create.call_count, 2)
        self.assertEqual(mock_g.call_count, 1)
        mock_session_post.assert_called_once_with(test_job_url, params=test_params_works, files=test_files_works)
예제 #2
0
    def test_server_response(self, mock_g, mock_n, mock_k, mock_session_post):

        # Mock OpenStack clients and response to POST
        mock_image = {'name': test_image_name, 'id': test_image_id}
        mock_gclient = Mock()
        mock_gclient.images = Mock()
        mock_im_list = Mock(return_value=[mock_image])
        mock_gclient.images.list = mock_im_list
        mock_gclient.image_members = Mock()
        mock_gclient.image_members.create = Mock()
        mock_g.return_value = mock_gclient

        mock_flavor = Mock()
        mock_flavor.name = test_flavor_name
        mock_flavor.id = test_flavor_id
        mock_fl_list = Mock(return_value=[mock_flavor])
        mock_nclient = Mock()
        mock_nclient.flavors = Mock()
        mock_nclient.flavors.list = mock_fl_list
        mock_n.return_value = mock_nclient

        mock_response = Mock()
        mock_response.text = test_job_id
        mock_response.status_code = 200
        mock_session_post.return_value = mock_response

        c = ProcClient(test_auth, host=test_host)

        # Failure on server side when trying to validate job
        mock_response_validate = Mock()
        mock_response_validate.text = 'Failed to validate job.'
        mock_response_validate.status_code = 400
        mock_response_validate.raise_for_status = Mock(
            side_effect=Exception('Failed.'))
        mock_session_post.return_value = mock_response_validate

        with self.assertRaises(Exception):
            c.submit_job(test_job_works, test_image_name, test_flavor_name)
        mock_session_post.assert_called_once_with(test_job_url,
                                                  params=test_params_works,
                                                  files=test_files_works)

        mock_session_post.return_value = mock_response
        mock_session_post.reset_mock()

        # glance client needs to be refreshed. First time it tries to obtain
        # an image list it fails. Second time it works.
        mock_gclient.images.list.side_effect = [
            Exception('Failed'), mock_im_list()
        ]
        mock_gclient.images.list.reset_mock()
        mock_g.reset_mock()
        c.submit_job(test_job_works, test_image_name, test_flavor_name)
        self.assertEqual(mock_gclient.images.list.call_count, 2)
        # only one additional call to glance client constructor since it
        # was originally created with the ProcClient instance
        self.assertEqual(mock_g.call_count, 1)
        mock_session_post.assert_called_once_with(test_job_url,
                                                  params=test_params_works,
                                                  files=test_files_works)

        mock_gclient.images.list.side_effect = None
        mock_session_post.reset_mock()

        # nova client needs to be refreshed. First time it tries to obtain
        # a flavor list it fails. Second time it works.
        mock_nclient.flavors.list.side_effect = [
            Exception('Failed'), mock_fl_list()
        ]
        mock_nclient.flavors.list.reset_mock()
        mock_n.reset_mock()
        c.submit_job(test_job_works, test_image_name, test_flavor_name)
        self.assertEqual(mock_nclient.flavors.list.call_count, 2)
        self.assertEqual(mock_n.call_count, 1)
        mock_session_post.assert_called_once_with(test_job_url,
                                                  params=test_params_works,
                                                  files=test_files_works)

        mock_nclient.flavors.list.side_effect = None
        mock_session_post.reset_mock()

        # glance client needs to be refreshed. First time it tries to
        # share an image it fails. Second time it works.
        mock_gclient.image_members.create.side_effect = [
            Exception('Failed'), Mock()
        ]
        mock_gclient.image_members.create.reset_mock()
        mock_g.reset_mock()
        c.submit_job(test_job_works, test_image_name, test_flavor_name)
        self.assertEqual(mock_gclient.image_members.create.call_count, 2)
        self.assertEqual(mock_g.call_count, 1)
        mock_session_post.assert_called_once_with(test_job_url,
                                                  params=test_params_works,
                                                  files=test_files_works)
예제 #3
0
    def test_submit_job(self, mock_g, mock_n, mock_k, mock_session_post):

        # Mock OpenStack clients and response to POST
        mock_image = {"name": test_image_name, "id": test_image_id}
        mock_gclient = Mock()
        mock_gclient.images = Mock()
        mock_gclient.images.list = Mock(return_value=[mock_image])
        mock_g.return_value = mock_gclient

        mock_flavor = Mock()
        mock_flavor.name = test_flavor_name
        mock_flavor.id = test_flavor_id
        mock_nclient = Mock()
        mock_nclient.flavors = Mock()
        mock_nclient.flavors.list = Mock(return_value=[mock_flavor])
        mock_n.return_value = mock_nclient

        mock_response = Mock()
        mock_response.text = test_job_id
        mock_response.status_code = 200
        mock_session_post.return_value = mock_response

        c = ProcClient(test_auth, host=test_host)

        # good submission using image and flavor names
        c.submit_job(test_job_works, test_image_name, test_flavor_name)
        mock_session_post.assert_called_once_with(test_job_url, params=test_params_works, files=test_files_works)

        mock_session_post.reset_mock()

        # good submission using image and flavor UUIDs
        c.submit_job(test_job_works, test_image_id, test_flavor_id)
        mock_session_post.assert_called_once_with(test_job_url, params=test_params_works, files=test_files_works)

        mock_session_post.reset_mock()

        # With nopost we should not do a POST
        c.submit_job(test_job_works, test_image_name, test_flavor_name, nopost=True)
        self.assertEqual(mock_session_post.call_count, 0)

        # failed submission due to bad image name
        with self.assertRaises(ValueError):
            c.submit_job(test_job_works, "bad_image", test_flavor_name)
        self.assertEqual(mock_session_post.call_count, 0)

        # failed submission due to bad flavor name
        with self.assertRaises(ValueError):
            c.submit_job(test_job_works, test_image_name, "yucky")
        self.assertEqual(mock_session_post.call_count, 0)

        # failed submission because no Executable line in job file
        with self.assertRaises(ValueError):
            c.submit_job(test_job_fails, test_image_name, test_flavor_name)
        self.assertEqual(mock_session_post.call_count, 0)

        # works without Executable by specifying jobscriptonvm
        c.submit_job(test_job_fails, test_image_name, test_flavor_name, jobscriptonvm=test_script_on_vm)
        mock_session_post.assert_called_once_with(test_job_url, params=test_params_on_vm, files=test_files_on_vm)

        mock_session_post.reset_mock()
예제 #4
0
    def test_submit_job(self, mock_g, mock_n, mock_k, mock_session_post):

        # Mock OpenStack clients and response to POST
        mock_image = {'name': test_image_name, 'id': test_image_id}
        mock_gclient = Mock()
        mock_gclient.images = Mock()
        mock_gclient.images.list = Mock(return_value=[mock_image])
        mock_g.return_value = mock_gclient

        mock_flavor = Mock()
        mock_flavor.name = test_flavor_name
        mock_flavor.id = test_flavor_id
        mock_nclient = Mock()
        mock_nclient.flavors = Mock()
        mock_nclient.flavors.list = Mock(return_value=[mock_flavor])
        mock_n.return_value = mock_nclient

        mock_response = Mock()
        mock_response.text = test_job_id
        mock_response.status_code = 200
        mock_session_post.return_value = mock_response

        c = ProcClient(test_auth, host=test_host)

        # good submission using image and flavor names
        c.submit_job(test_job_works, test_image_name, test_flavor_name)
        mock_session_post.assert_called_once_with(test_job_url,
                                                  params=test_params_works,
                                                  files=test_files_works)

        mock_session_post.reset_mock()

        # good submission using image and flavor UUIDs
        c.submit_job(test_job_works, test_image_id, test_flavor_id)
        mock_session_post.assert_called_once_with(test_job_url,
                                                  params=test_params_works,
                                                  files=test_files_works)

        mock_session_post.reset_mock()

        # With nopost we should not do a POST
        c.submit_job(test_job_works,
                     test_image_name,
                     test_flavor_name,
                     nopost=True)
        self.assertEqual(mock_session_post.call_count, 0)

        # failed submission due to bad image name
        with self.assertRaises(ValueError):
            c.submit_job(test_job_works, 'bad_image', test_flavor_name)
        self.assertEqual(mock_session_post.call_count, 0)

        # failed submission due to bad flavor name
        with self.assertRaises(ValueError):
            c.submit_job(test_job_works, test_image_name, 'yucky')
        self.assertEqual(mock_session_post.call_count, 0)

        # failed submission because no Executable line in job file
        with self.assertRaises(ValueError):
            c.submit_job(test_job_fails, test_image_name, test_flavor_name)
        self.assertEqual(mock_session_post.call_count, 0)

        # works without Executable by specifying jobscriptonvm
        c.submit_job(test_job_fails,
                     test_image_name,
                     test_flavor_name,
                     jobscriptonvm=test_script_on_vm)
        mock_session_post.assert_called_once_with(test_job_url,
                                                  params=test_params_on_vm,
                                                  files=test_files_on_vm)

        mock_session_post.reset_mock()