提交 26d2a61d 编写于 作者: L liqingping

feat: update CleanPodPolicyALL to CleanPodPolicyAll

上级 70468e89
......@@ -65,8 +65,8 @@ const (
// CleanPodPolicyRunning means deleting all running pods of the job after completed
CleanPodPolicyRunning CleanPodPolicy = "Running"
// CleanPodPolicyALL means deleting all pods of the job after completed
CleanPodPolicyALL CleanPodPolicy = "ALL"
// CleanPodPolicyAll means deleting all pods of the job after completed
CleanPodPolicyAll CleanPodPolicy = "All"
// CleanPodPolicyNone means never deleting any pods of the job after completed
CleanPodPolicyNone CleanPodPolicy = "None"
......
......@@ -59,10 +59,10 @@ func (r *DIJob) ValidateCreate() error {
dijoblog.Info("validate create", "name", r.Name)
// TODO(user): fill in your validation logic upon object creation.
if r.Spec.CleanPodPolicy != CleanPodPolicyALL && r.Spec.CleanPodPolicy != CleanPodPolicyNone &&
if r.Spec.CleanPodPolicy != CleanPodPolicyAll && r.Spec.CleanPodPolicy != CleanPodPolicyNone &&
r.Spec.CleanPodPolicy != CleanPodPolicyRunning {
return fmt.Errorf("Invalid CleanPodPolicy %s, expected in [%s, %s, %s]",
r.Spec.CleanPodPolicy, CleanPodPolicyNone, CleanPodPolicyRunning, CleanPodPolicyALL)
r.Spec.CleanPodPolicy, CleanPodPolicyNone, CleanPodPolicyRunning, CleanPodPolicyAll)
}
return nil
}
......@@ -72,10 +72,10 @@ func (r *DIJob) ValidateUpdate(old runtime.Object) error {
dijoblog.Info("validate update", "name", r.Name)
// TODO(user): fill in your validation logic upon object update.
if r.Spec.CleanPodPolicy != CleanPodPolicyALL && r.Spec.CleanPodPolicy != CleanPodPolicyNone &&
if r.Spec.CleanPodPolicy != CleanPodPolicyAll && r.Spec.CleanPodPolicy != CleanPodPolicyNone &&
r.Spec.CleanPodPolicy != CleanPodPolicyRunning {
return fmt.Errorf("Invalid CleanPodPolicy %s, expected in [%s, %s, %s]",
r.Spec.CleanPodPolicy, CleanPodPolicyNone, CleanPodPolicyRunning, CleanPodPolicyALL)
r.Spec.CleanPodPolicy, CleanPodPolicyNone, CleanPodPolicyRunning, CleanPodPolicyAll)
}
return nil
}
......
......@@ -23,7 +23,7 @@ var _ = Describe("Webhook test", func() {
}
testCases := []testCase{
{cleanPodPolicy: CleanPodPolicyRunning, expectCleanPodPolicy: CleanPodPolicyRunning},
{cleanPodPolicy: CleanPodPolicyALL, expectCleanPodPolicy: CleanPodPolicyALL},
{cleanPodPolicy: CleanPodPolicyAll, expectCleanPodPolicy: CleanPodPolicyAll},
{cleanPodPolicy: CleanPodPolicyNone, expectCleanPodPolicy: CleanPodPolicyNone},
{cleanPodPolicy: CleanPodPolicy(""), expectCleanPodPolicy: CleanPodPolicyRunning},
{cleanPodPolicy: CleanPodPolicy("hello"), expectCleanPodPolicy: CleanPodPolicy("will be refused by webhook")},
......@@ -42,7 +42,7 @@ var _ = Describe("Webhook test", func() {
err = k8sClient.Create(ctx, job, &client.CreateOptions{})
if err != nil {
if c.cleanPodPolicy != CleanPodPolicyRunning && c.cleanPodPolicy != CleanPodPolicyNone &&
c.cleanPodPolicy != CleanPodPolicyALL {
c.cleanPodPolicy != CleanPodPolicyAll {
Expect(err.Error()).To(ContainSubstring("Invalid CleanPodPolicy"))
continue
} else {
......@@ -69,7 +69,7 @@ var _ = Describe("Webhook test", func() {
}
testCases := []testCase{
{cleanPodPolicy: CleanPodPolicyRunning, expectCleanPodPolicy: CleanPodPolicyRunning},
{cleanPodPolicy: CleanPodPolicyALL, expectCleanPodPolicy: CleanPodPolicyALL},
{cleanPodPolicy: CleanPodPolicyAll, expectCleanPodPolicy: CleanPodPolicyAll},
{cleanPodPolicy: CleanPodPolicyNone, expectCleanPodPolicy: CleanPodPolicyNone},
{cleanPodPolicy: CleanPodPolicy(""), expectCleanPodPolicy: CleanPodPolicyRunning},
{cleanPodPolicy: CleanPodPolicy("hello"), expectCleanPodPolicy: CleanPodPolicy("will be refused by webhook")},
......@@ -90,7 +90,7 @@ var _ = Describe("Webhook test", func() {
err = k8sClient.Update(ctx, job, &client.UpdateOptions{})
if err != nil {
if c.cleanPodPolicy != CleanPodPolicyRunning && c.cleanPodPolicy != CleanPodPolicyNone &&
c.cleanPodPolicy != CleanPodPolicyALL {
c.cleanPodPolicy != CleanPodPolicyAll {
Expect(err.Error()).To(ContainSubstring("Invalid CleanPodPolicy"))
continue
} else {
......
......@@ -9,7 +9,7 @@ spec:
spec:
containers:
- name: di-container
image: registry.sensetime.com/cloudnative4ai/ding:v0.1.0-75c41277
image: registry.sensetime.com/cloudnative4ai/ding:v0.1.1-3b98cb70
imagePullPolicy: Always
env:
- name: PYTHONUNBUFFERED
......
......@@ -19,7 +19,7 @@ spec:
spec:
containers:
- name: di-container
image: registry.sensetime.com/cloudnative4ai/ding:v0.1.0-75c41277
image: registry.sensetime.com/cloudnative4ai/ding:v0.1.1-3b98cb70
imagePullPolicy: Always
env:
- name: PYTHONUNBUFFERED
......@@ -162,7 +162,7 @@ spec:
spec:
containers:
- name: di-container
image: registry.sensetime.com/cloudnative4ai/ding:v0.1.0-75c41277
image: registry.sensetime.com/cloudnative4ai/ding:v0.1.1-3b98cb70
imagePullPolicy: Always
env:
- name: PYTHONUNBUFFERED
......@@ -189,7 +189,7 @@ spec:
spec:
containers:
- name: di-container
image: registry.sensetime.com/cloudnative4ai/ding:v0.1.0-75c41277
image: registry.sensetime.com/cloudnative4ai/ding:v0.1.1-3b98cb70
imagePullPolicy: Always
env:
- name: PYTHONUNBUFFERED
......
......@@ -19,7 +19,7 @@ spec:
spec:
containers:
- name: di-container
image: registry.sensetime.com/cloudnative4ai/ding:v0.1.0-75c41277
image: registry.sensetime.com/cloudnative4ai/ding:v0.1.1-3b98cb70
imagePullPolicy: Always
env:
- name: PYTHONUNBUFFERED
......@@ -170,7 +170,7 @@ spec:
spec:
containers:
- name: di-container
image: registry.sensetime.com/cloudnative4ai/ding:v0.1.0-75c41277
image: registry.sensetime.com/cloudnative4ai/ding:v0.1.1-3b98cb70
imagePullPolicy: Always
env:
- name: PYTHONUNBUFFERED
......@@ -200,7 +200,7 @@ spec:
spec:
containers:
- name: di-container
image: registry.sensetime.com/cloudnative4ai/ding:v0.1.0-75c41277
image: registry.sensetime.com/cloudnative4ai/ding:v0.1.1-3b98cb70
imagePullPolicy: Always
env:
- name: PYTHONUNBUFFERED
......
......@@ -139,7 +139,7 @@ func (r *DIJobReconciler) deletePodsAndServices(ctx context.Context, job *div1al
}
}
if job.Spec.CleanPodPolicy != div1alpha1.CleanPodPolicyALL &&
if job.Spec.CleanPodPolicy != div1alpha1.CleanPodPolicyAll &&
job.Spec.CleanPodPolicy != div1alpha1.CleanPodPolicyRunning {
return nil
}
......
......@@ -20,7 +20,7 @@ var _ = Describe("DIJob Specification", func() {
Context("When creating a DIJob with different CleanPodPolicy", func() {
It("Should execute different pods deletion policy with different CleanPodPolicy", func() {
cleanPodPolicies := []div1alpha1.CleanPodPolicy{
div1alpha1.CleanPodPolicyALL,
div1alpha1.CleanPodPolicyAll,
div1alpha1.CleanPodPolicyRunning,
div1alpha1.CleanPodPolicyNone,
}
......@@ -116,7 +116,7 @@ var _ = Describe("DIJob Specification", func() {
By("Checking all the pods and services are deleted")
switch policy {
case div1alpha1.CleanPodPolicyALL:
case div1alpha1.CleanPodPolicyAll:
Eventually(func() int {
pods, err := diutil.ListPods(ctx, k8sClient, &dijob)
if err != nil {
......
......@@ -408,22 +408,6 @@ var _ = Describe("Server Test", func() {
Expect(portCount).Should(Equal(c.expectedDDPLPorts))
}
By("Send request on DELETE /v1alpha1/replicas")
var dln int = 1
dreq := commontypes.DIJobRequest{
Namespace: job.Namespace,
Coordinator: coorname,
Learners: commontypes.ResourceQuantity{
Replicas: dln,
},
}
drbody, err := json.Marshal(dreq)
Expect(err).NotTo(HaveOccurred())
ddiresp, err := sendRequest(http.MethodDelete, drbody, rurl, http.StatusOK, true)
Expect(err).NotTo(HaveOccurred())
Expect(len(ddiresp.Learners)).Should(Equal(dln))
err = testutil.CleanUpJob(ctx, k8sClient, job.DeepCopy())
Expect(err).NotTo(HaveOccurred())
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册