kubernetes-entrypoint/mocks/job.go
Ian Howell bba1bd9d3d This is a code-linting refactor
This refactors a large part of the codebase using the suggestions from
golangci-lint

Change-Id: I2b7735086a64e50f3d5e0b30c225870bddc70935
2019-09-30 14:20:43 -05:00

88 lines
2.1 KiB
Go

package mocks
import (
"fmt"
v1 "k8s.io/api/batch/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
types "k8s.io/apimachinery/pkg/types"
watch "k8s.io/apimachinery/pkg/watch"
v1batch "k8s.io/client-go/kubernetes/typed/batch/v1"
)
const (
SucceedingJobName = "succeed"
FailingJobName = "fail"
SucceedingJobLabel = "succeed"
FailingJobLabel = "fail"
)
type jClient struct {
}
func (j jClient) Get(name string, opts metav1.GetOptions) (*v1.Job, error) {
if name == SucceedingJobName {
return &v1.Job{
Status: v1.JobStatus{Succeeded: 1},
}, nil
}
if name == FailingJobName {
return &v1.Job{
Status: v1.JobStatus{Succeeded: 0},
}, nil
}
return nil, fmt.Errorf("Mock job didnt work")
}
func (j jClient) Create(job *v1.Job) (*v1.Job, error) {
return nil, fmt.Errorf("Not implemented")
}
func (j jClient) Delete(name string, opts *metav1.DeleteOptions) error {
return fmt.Errorf("Not implemented")
}
func (j jClient) DeleteCollection(options *metav1.DeleteOptions, listOptions metav1.ListOptions) error {
return fmt.Errorf("Not implemented")
}
func (j jClient) List(options metav1.ListOptions) (*v1.JobList, error) {
var jobs []v1.Job
switch options.LabelSelector {
case fmt.Sprintf("name=%s", SucceedingJobLabel):
jobs = []v1.Job{NewJob(1)}
case fmt.Sprintf("name=%s", FailingJobLabel):
jobs = []v1.Job{NewJob(1), NewJob(0)}
default:
return nil, fmt.Errorf("Mock job didnt work")
}
return &v1.JobList{
Items: jobs,
}, nil
}
func (j jClient) Update(job *v1.Job) (*v1.Job, error) {
return nil, fmt.Errorf("Not implemented")
}
func (j jClient) UpdateStatus(job *v1.Job) (*v1.Job, error) {
return nil, fmt.Errorf("Not implemented")
}
func (j jClient) Watch(options metav1.ListOptions) (watch.Interface, error) {
return nil, fmt.Errorf("Not implemented")
}
func (j jClient) Patch(name string, pt types.PatchType, data []byte,
subresources ...string) (result *v1.Job, err error) {
return nil, fmt.Errorf("Not implemented")
}
func NewJClient() v1batch.JobInterface {
return jClient{}
}
func NewJob(succeeded int32) v1.Job {
return v1.Job{
Status: v1.JobStatus{Succeeded: succeeded},
}
}