Skip to content

Commit

Permalink
fixed test
Browse files Browse the repository at this point in the history
Signed-off-by: Saravanan Balasubramanian <sarabala1979@gmail.com>
  • Loading branch information
sarabala1979 committed Sep 3, 2021
1 parent 117e85f commit 92153dc
Show file tree
Hide file tree
Showing 3 changed files with 1 addition and 61 deletions.
2 changes: 1 addition & 1 deletion pkg/apis/workflow/v1alpha1/generated.proto

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

4 changes: 0 additions & 4 deletions workflow/artifacts/gcs/gcs.go
Expand Up @@ -294,9 +294,6 @@ func uploadObject(client *storage.Client, bucket, key, localPath string) error {
}

func (g *ArtifactDriver) ListObjects(artifact *wfv1.Artifact) ([]string, error) {
<<<<<<< HEAD
return nil, fmt.Errorf("ListObjects is currently not supported for this artifact type, but it will be in a future version")
=======
var files []string
err := waitutil.Backoff(defaultRetry,
func() (bool, error) {
Expand All @@ -314,5 +311,4 @@ func (g *ArtifactDriver) ListObjects(artifact *wfv1.Artifact) ([]string, error)
return true, nil
})
return files, err
>>>>>>> 30340c427... fix(gcs): backoff bool should be false if error is transient (#6577)
}
56 changes: 0 additions & 56 deletions workflow/controller/workflowpod_test.go
Expand Up @@ -1584,59 +1584,3 @@ func TestPodMetadataWithWorkflowDefaults(t *testing.T) {
assert.Equal(t, "label-value", pod.ObjectMeta.Labels["controller-level-pod-label"])
cancel()
}

func TestPodMetadataWithWorkflowDefaults(t *testing.T) {
cancel, controller := newController()
defer cancel()

wfDefaultAnnotations := make(map[string]string)
wfDefaultAnnotations["controller-level-pod-annotation"] = "annotation-value"
wfDefaultAnnotations["workflow-level-pod-annotation"] = "set-by-controller"
wfDefaultLabels := make(map[string]string)
wfDefaultLabels["controller-level-pod-label"] = "label-value"
wfDefaultLabels["workflow-level-pod-label"] = "set-by-controller"
controller.Config.WorkflowDefaults = &wfv1.Workflow{
Spec: wfv1.WorkflowSpec{
PodMetadata: &wfv1.Metadata{
Annotations: wfDefaultAnnotations,
Labels: wfDefaultLabels,
},
},
}

wf := wfv1.MustUnmarshalWorkflow(helloWorldWf)
ctx := context.Background()
woc := newWorkflowOperationCtx(wf, controller)
err := woc.setExecWorkflow(ctx)
assert.NoError(t, err)
mainCtr := woc.execWf.Spec.Templates[0].Container
pod, _ := woc.createWorkflowPod(ctx, wf.Name, []apiv1.Container{*mainCtr}, &wf.Spec.Templates[0], &createWorkflowPodOpts{})
assert.Equal(t, "annotation-value", pod.ObjectMeta.Annotations["controller-level-pod-annotation"])
assert.Equal(t, "set-by-controller", pod.ObjectMeta.Annotations["workflow-level-pod-annotation"])
assert.Equal(t, "label-value", pod.ObjectMeta.Labels["controller-level-pod-label"])
assert.Equal(t, "set-by-controller", pod.ObjectMeta.Labels["workflow-level-pod-label"])
cancel() // need to cancel to spin up pods with the same name

cancel, controller = newController()
defer cancel()
controller.Config.WorkflowDefaults = &wfv1.Workflow{
Spec: wfv1.WorkflowSpec{
PodMetadata: &wfv1.Metadata{
Annotations: wfDefaultAnnotations,
Labels: wfDefaultLabels,
},
},
}
wf = wfv1.MustUnmarshalWorkflow(wfWithPodMetadata)
ctx = context.Background()
woc = newWorkflowOperationCtx(wf, controller)
err = woc.setExecWorkflow(ctx)
assert.NoError(t, err)
mainCtr = woc.execWf.Spec.Templates[0].Container
pod, _ = woc.createWorkflowPod(ctx, wf.Name, []apiv1.Container{*mainCtr}, &wf.Spec.Templates[0], &createWorkflowPodOpts{})
assert.Equal(t, "foo", pod.ObjectMeta.Annotations["workflow-level-pod-annotation"])
assert.Equal(t, "bar", pod.ObjectMeta.Labels["workflow-level-pod-label"])
assert.Equal(t, "annotation-value", pod.ObjectMeta.Annotations["controller-level-pod-annotation"])
assert.Equal(t, "label-value", pod.ObjectMeta.Labels["controller-level-pod-label"])
cancel()
}

0 comments on commit 92153dc

Please sign in to comment.