65 lines
2.4 KiB
Python
65 lines
2.4 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
||
|
|
# not use this file except in compliance with the License. You may obtain
|
||
|
|
# a copy of the License at
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||
|
|
# License for the specific language governing permissions and limitations
|
||
|
|
# under the License.
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
import uuid
|
||
|
|
|
||
|
|
from kubernetes.client import api_client
|
||
|
|
from kubernetes.client.api import batch_v1_api
|
||
|
|
from kubernetes.e2e_test import base
|
||
|
|
|
||
|
|
|
||
|
|
class TestClientBatch(unittest.TestCase):
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def setUpClass(cls):
|
||
|
|
cls.config = base.get_e2e_configuration()
|
||
|
|
|
||
|
|
def test_job_apis(self):
|
||
|
|
client = api_client.ApiClient(configuration=self.config)
|
||
|
|
api = batch_v1_api.BatchV1Api(client)
|
||
|
|
|
||
|
|
name = 'test-job-' + str(uuid.uuid4())
|
||
|
|
job_manifest = {
|
||
|
|
'kind': 'Job',
|
||
|
|
'spec': {
|
||
|
|
'template':
|
||
|
|
{'spec':
|
||
|
|
{'containers': [
|
||
|
|
{'image': 'busybox',
|
||
|
|
'name': name,
|
||
|
|
'command': ["sh", "-c", "sleep 5"]
|
||
|
|
}],
|
||
|
|
'restartPolicy': 'Never'},
|
||
|
|
'metadata': {'name': name}}},
|
||
|
|
'apiVersion': 'batch/v1',
|
||
|
|
'metadata': {'name': name}}
|
||
|
|
|
||
|
|
create_job_resp = api.create_namespaced_job(
|
||
|
|
body=job_manifest, namespace='default')
|
||
|
|
self.assertEqual(name, create_job_resp.metadata.name)
|
||
|
|
|
||
|
|
resp = api.read_namespaced_job(
|
||
|
|
name=name, namespace='default')
|
||
|
|
self.assertEqual(name, resp.metadata.name)
|
||
|
|
self.assertEqual(name, resp.spec.template.spec.containers[0].name)
|
||
|
|
self.assertEqual('busybox', resp.spec.template.spec.containers[0].image)
|
||
|
|
self.assertEqual('sh', resp.spec.template.spec.containers[0].command[0])
|
||
|
|
self.assertEqual('-c', resp.spec.template.spec.containers[0].command[1])
|
||
|
|
self.assertEqual('sleep 5', resp.spec.template.spec.containers[0].command[2])
|
||
|
|
self.assertEqual('Never', resp.spec.template.spec.restart_policy)
|
||
|
|
|
||
|
|
api.delete_namespaced_job(
|
||
|
|
name=name, namespace='default', propagation_policy='Background')
|