DockerCLI/cli/command/stack/kubernetes/deploy_test.go

package kubernetes

import (
	"errors"
	"testing"

	composev1alpha3 "github.com/docker/compose-on-kubernetes/api/client/clientset/typed/compose/v1alpha3"
	composev1beta1 "github.com/docker/compose-on-kubernetes/api/client/clientset/typed/compose/v1beta1"
	composev1beta2 "github.com/docker/compose-on-kubernetes/api/client/clientset/typed/compose/v1beta2"
	"github.com/docker/compose-on-kubernetes/api/compose/v1alpha3"
	"github.com/docker/compose-on-kubernetes/api/compose/v1beta1"
	"github.com/docker/compose-on-kubernetes/api/compose/v1beta2"
	"gotest.tools/assert"
	kerrors "k8s.io/apimachinery/pkg/api/errors"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	types "k8s.io/apimachinery/pkg/types"
	watch "k8s.io/apimachinery/pkg/watch"
	"k8s.io/client-go/kubernetes/fake"
)

func testStack() Stack {
	return Stack{
		Name:      "test",
		Namespace: "test",
		ComposeFile: `version: "3.3"
services: 
  test:
    image: nginx
secrets:
  test:
    file: testdata/secret
configs:
  test:
    file: testdata/config
`,
		Spec: &v1alpha3.StackSpec{
			Configs: map[string]v1alpha3.ConfigObjConfig{
				"test": {Name: "test", File: "testdata/config"},
			},
			Secrets: map[string]v1alpha3.SecretConfig{
				"test": {Name: "test", File: "testdata/secret"},
			},
		},
	}
}

func TestCreateChildResourcesV1Beta1(t *testing.T) {
	k8sclientSet := fake.NewSimpleClientset()
	stack := testStack()
	configs := k8sclientSet.CoreV1().ConfigMaps("test")
	secrets := k8sclientSet.CoreV1().Secrets("test")
	assert.NilError(t, createResources(
		stack,
		&stackV1Beta1{stacks: &fakeV1beta1Client{}},
		configs,
		secrets))
	c, err := configs.Get("test", metav1.GetOptions{})
	assert.NilError(t, err)
	checkOwnerReferences(t, c.ObjectMeta, "test", v1beta1.SchemeGroupVersion.String())
	s, err := secrets.Get("test", metav1.GetOptions{})
	assert.NilError(t, err)
	checkOwnerReferences(t, s.ObjectMeta, "test", v1beta1.SchemeGroupVersion.String())
}

func checkOwnerReferences(t *testing.T, objMeta metav1.ObjectMeta, stackName, stackVersion string) {
	t.Helper()
	assert.Equal(t, len(objMeta.OwnerReferences), 1)
	assert.Equal(t, objMeta.OwnerReferences[0].Name, stackName)
	assert.Equal(t, objMeta.OwnerReferences[0].Kind, "Stack")
	assert.Equal(t, objMeta.OwnerReferences[0].APIVersion, stackVersion)
}

func TestCreateChildResourcesV1Beta2(t *testing.T) {
	k8sclientSet := fake.NewSimpleClientset()
	stack := testStack()
	configs := k8sclientSet.CoreV1().ConfigMaps("test")
	secrets := k8sclientSet.CoreV1().Secrets("test")
	assert.NilError(t, createResources(
		stack,
		&stackV1Beta2{stacks: &fakeV1beta2Client{}},
		configs,
		secrets))
	c, err := configs.Get("test", metav1.GetOptions{})
	assert.NilError(t, err)
	checkOwnerReferences(t, c.ObjectMeta, "test", v1beta2.SchemeGroupVersion.String())
	s, err := secrets.Get("test", metav1.GetOptions{})
	assert.NilError(t, err)
	checkOwnerReferences(t, s.ObjectMeta, "test", v1beta2.SchemeGroupVersion.String())
}

func TestCreateChildResourcesV1Alpha3(t *testing.T) {
	k8sclientSet := fake.NewSimpleClientset()
	stack := testStack()
	configs := k8sclientSet.CoreV1().ConfigMaps("test")
	secrets := k8sclientSet.CoreV1().Secrets("test")
	assert.NilError(t, createResources(
		stack,
		&stackV1Alpha3{stacks: &fakeV1alpha3Client{}},
		configs,
		secrets))
	c, err := configs.Get("test", metav1.GetOptions{})
	assert.NilError(t, err)
	checkOwnerReferences(t, c.ObjectMeta, "test", v1alpha3.SchemeGroupVersion.String())
	s, err := secrets.Get("test", metav1.GetOptions{})
	assert.NilError(t, err)
	checkOwnerReferences(t, s.ObjectMeta, "test", v1alpha3.SchemeGroupVersion.String())
}

func TestCreateChildResourcesWithStackCreationErrorV1Beta1(t *testing.T) {
	k8sclientSet := fake.NewSimpleClientset()
	stack := testStack()
	configs := k8sclientSet.CoreV1().ConfigMaps("test")
	secrets := k8sclientSet.CoreV1().Secrets("test")
	err := createResources(
		stack,
		&stackV1Beta1{stacks: &fakeV1beta1Client{errorOnCreate: true}},
		configs,
		secrets)
	assert.Error(t, err, "some error")
	_, err = configs.Get("test", metav1.GetOptions{})
	assert.Check(t, kerrors.IsNotFound(err))
	_, err = secrets.Get("test", metav1.GetOptions{})
	assert.Check(t, kerrors.IsNotFound(err))
}

func TestCreateChildResourcesWithStackCreationErrorV1Beta2(t *testing.T) {
	k8sclientSet := fake.NewSimpleClientset()
	stack := testStack()
	configs := k8sclientSet.CoreV1().ConfigMaps("test")
	secrets := k8sclientSet.CoreV1().Secrets("test")
	err := createResources(
		stack,
		&stackV1Beta2{stacks: &fakeV1beta2Client{errorOnCreate: true}},
		configs,
		secrets)
	assert.Error(t, err, "some error")
	_, err = configs.Get("test", metav1.GetOptions{})
	assert.Check(t, kerrors.IsNotFound(err))
	_, err = secrets.Get("test", metav1.GetOptions{})
	assert.Check(t, kerrors.IsNotFound(err))
}

func TestCreateChildResourcesWithStackCreationErrorV1Alpha3(t *testing.T) {
	k8sclientSet := fake.NewSimpleClientset()
	stack := testStack()
	configs := k8sclientSet.CoreV1().ConfigMaps("test")
	secrets := k8sclientSet.CoreV1().Secrets("test")
	err := createResources(
		stack,
		&stackV1Alpha3{stacks: &fakeV1alpha3Client{errorOnCreate: true}},
		configs,
		secrets)
	assert.Error(t, err, "some error")
	_, err = configs.Get("test", metav1.GetOptions{})
	assert.Check(t, kerrors.IsNotFound(err))
	_, err = secrets.Get("test", metav1.GetOptions{})
	assert.Check(t, kerrors.IsNotFound(err))
}

type fakeV1beta1Client struct {
	errorOnCreate bool
}

func (c *fakeV1beta1Client) Create(s *v1beta1.Stack) (*v1beta1.Stack, error) {
	if c.errorOnCreate {
		return nil, errors.New("some error")
	}
	return s, nil
}

func (c *fakeV1beta1Client) Update(*v1beta1.Stack) (*v1beta1.Stack, error) {
	return nil, nil
}

func (c *fakeV1beta1Client) UpdateStatus(*v1beta1.Stack) (*v1beta1.Stack, error) {
	return nil, nil
}

func (c *fakeV1beta1Client) Delete(name string, options *metav1.DeleteOptions) error {
	return nil
}

func (c *fakeV1beta1Client) DeleteCollection(options *metav1.DeleteOptions, listOptions metav1.ListOptions) error {
	return nil
}

func (c *fakeV1beta1Client) Get(name string, options metav1.GetOptions) (*v1beta1.Stack, error) {
	return nil, kerrors.NewNotFound(v1beta1.SchemeGroupVersion.WithResource("stacks").GroupResource(), name)
}

func (c *fakeV1beta1Client) List(opts metav1.ListOptions) (*v1beta1.StackList, error) {
	return nil, nil
}

func (c *fakeV1beta1Client) Watch(opts metav1.ListOptions) (watch.Interface, error) {
	return nil, nil
}

func (c *fakeV1beta1Client) Patch(name string, pt types.PatchType, data []byte, subresources ...string) (*v1beta1.Stack, error) {
	return nil, nil
}

func (c *fakeV1beta1Client) WithSkipValidation() composev1beta1.StackInterface {
	return c
}

type fakeV1beta2Client struct {
	errorOnCreate bool
}

func (c *fakeV1beta2Client) Create(s *v1beta2.Stack) (*v1beta2.Stack, error) {
	if c.errorOnCreate {
		return nil, errors.New("some error")
	}
	return s, nil
}

func (c *fakeV1beta2Client) Update(*v1beta2.Stack) (*v1beta2.Stack, error) {
	return nil, nil
}

func (c *fakeV1beta2Client) UpdateStatus(*v1beta2.Stack) (*v1beta2.Stack, error) {
	return nil, nil
}

func (c *fakeV1beta2Client) Delete(name string, options *metav1.DeleteOptions) error {
	return nil
}

func (c *fakeV1beta2Client) DeleteCollection(options *metav1.DeleteOptions, listOptions metav1.ListOptions) error {
	return nil
}

func (c *fakeV1beta2Client) Get(name string, options metav1.GetOptions) (*v1beta2.Stack, error) {
	return nil, kerrors.NewNotFound(v1beta1.SchemeGroupVersion.WithResource("stacks").GroupResource(), name)
}

func (c *fakeV1beta2Client) List(opts metav1.ListOptions) (*v1beta2.StackList, error) {
	return nil, nil
}

func (c *fakeV1beta2Client) Watch(opts metav1.ListOptions) (watch.Interface, error) {
	return nil, nil
}

func (c *fakeV1beta2Client) Patch(name string, pt types.PatchType, data []byte, subresources ...string) (*v1beta2.Stack, error) {
	return nil, nil
}

func (c *fakeV1beta2Client) WithSkipValidation() composev1beta2.StackInterface {
	return c
}

type fakeV1alpha3Client struct {
	errorOnCreate bool
}

func (c *fakeV1alpha3Client) Create(s *v1alpha3.Stack) (*v1alpha3.Stack, error) {
	if c.errorOnCreate {
		return nil, errors.New("some error")
	}
	return s, nil
}

func (c *fakeV1alpha3Client) Update(*v1alpha3.Stack) (*v1alpha3.Stack, error) {
	return nil, nil
}

func (c *fakeV1alpha3Client) UpdateStatus(*v1alpha3.Stack) (*v1alpha3.Stack, error) {
	return nil, nil
}

func (c *fakeV1alpha3Client) Delete(name string, options *metav1.DeleteOptions) error {
	return nil
}

func (c *fakeV1alpha3Client) DeleteCollection(options *metav1.DeleteOptions, listOptions metav1.ListOptions) error {
	return nil
}

func (c *fakeV1alpha3Client) Get(name string, options metav1.GetOptions) (*v1alpha3.Stack, error) {
	return nil, kerrors.NewNotFound(v1beta1.SchemeGroupVersion.WithResource("stacks").GroupResource(), name)
}

func (c *fakeV1alpha3Client) List(opts metav1.ListOptions) (*v1alpha3.StackList, error) {
	return nil, nil
}

func (c *fakeV1alpha3Client) Watch(opts metav1.ListOptions) (watch.Interface, error) {
	return nil, nil
}

func (c *fakeV1alpha3Client) Patch(name string, pt types.PatchType, data []byte, subresources ...string) (*v1alpha3.Stack, error) {
	return nil, nil
}

func (c *fakeV1alpha3Client) WithSkipValidation() composev1alpha3.StackInterface {
	return c
}