| 
 | 1 | +package templates  | 
 | 2 | + | 
 | 3 | +import (  | 
 | 4 | +	"bytes"  | 
 | 5 | +	"context"  | 
 | 6 | +	"errors"  | 
 | 7 | +	"io"  | 
 | 8 | +	"testing"  | 
 | 9 | + | 
 | 10 | +	"github.com/kyma-project/cli.v3/internal/clierror"  | 
 | 11 | +	"github.com/kyma-project/cli.v3/internal/cmd/alpha/templates/types"  | 
 | 12 | +	"github.com/kyma-project/cli.v3/internal/kube/fake"  | 
 | 13 | +	"github.com/spf13/cobra"  | 
 | 14 | +	"github.com/stretchr/testify/require"  | 
 | 15 | +	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"  | 
 | 16 | +)  | 
 | 17 | + | 
 | 18 | +func Test_remove(t *testing.T) {  | 
 | 19 | +	t.Run("build proper command", func(t *testing.T) {  | 
 | 20 | +		cmd := fixDeleteCommand(bytes.NewBuffer([]byte{}), &mockGetter{})  | 
 | 21 | + | 
 | 22 | +		require.Equal(t, "delete", cmd.Use)  | 
 | 23 | +		require.Equal(t, "delete test deploy", cmd.Short)  | 
 | 24 | +		require.Equal(t, "use this to delete test deploy", cmd.Long)  | 
 | 25 | + | 
 | 26 | +		require.NotNil(t, cmd.Flag("name"))  | 
 | 27 | +		require.NotNil(t, cmd.Flag("namespace"))  | 
 | 28 | +	})  | 
 | 29 | + | 
 | 30 | +	t.Run("delete resource", func(t *testing.T) {  | 
 | 31 | +		buf := bytes.NewBuffer([]byte{})  | 
 | 32 | +		fakeClient := &fake.RootlessDynamicClient{}  | 
 | 33 | +		mock := mockGetter{  | 
 | 34 | +			client: &fake.KubeClient{  | 
 | 35 | +				TestRootlessDynamicInterface: fakeClient,  | 
 | 36 | +			},  | 
 | 37 | +		}  | 
 | 38 | + | 
 | 39 | +		cmd := fixDeleteCommand(buf, &mock)  | 
 | 40 | + | 
 | 41 | +		cmd.SetArgs([]string{"--name", "test-deploy", "--namespace", "test-namespace"})  | 
 | 42 | + | 
 | 43 | +		err := cmd.Execute()  | 
 | 44 | +		require.NoError(t, err)  | 
 | 45 | + | 
 | 46 | +		require.Len(t, fakeClient.RemovedObjs, 1)  | 
 | 47 | +		require.Equal(t, fixDeletedUnstructuredDeployment(), fakeClient.RemovedObjs[0])  | 
 | 48 | +	})  | 
 | 49 | + | 
 | 50 | +	t.Run("failed to get client", func(t *testing.T) {  | 
 | 51 | +		buf := bytes.NewBuffer([]byte{})  | 
 | 52 | +		mock := mockGetter{  | 
 | 53 | +			clierror: clierror.New("test error"),  | 
 | 54 | +			client:   nil,  | 
 | 55 | +		}  | 
 | 56 | + | 
 | 57 | +		err := deleteResource(&deleteArgs{  | 
 | 58 | +			out:           buf,  | 
 | 59 | +			ctx:           context.Background(),  | 
 | 60 | +			clientGetter:  &mock,  | 
 | 61 | +			deleteOptions: fixDeleteOptions(),  | 
 | 62 | +		})  | 
 | 63 | +		require.Equal(t, clierror.New("test error"), err)  | 
 | 64 | +	})  | 
 | 65 | + | 
 | 66 | +	t.Run("failed to delete object", func(t *testing.T) {  | 
 | 67 | +		buf := bytes.NewBuffer([]byte{})  | 
 | 68 | +		fakeClient := &fake.RootlessDynamicClient{  | 
 | 69 | +			ReturnRemoveErr: errors.New("test error"),  | 
 | 70 | +		}  | 
 | 71 | +		mock := mockGetter{  | 
 | 72 | +			client: &fake.KubeClient{  | 
 | 73 | +				TestRootlessDynamicInterface: fakeClient,  | 
 | 74 | +			},  | 
 | 75 | +		}  | 
 | 76 | + | 
 | 77 | +		err := deleteResource(&deleteArgs{  | 
 | 78 | +			out:           buf,  | 
 | 79 | +			ctx:           context.Background(),  | 
 | 80 | +			clientGetter:  &mock,  | 
 | 81 | +			deleteOptions: fixDeleteOptions(),  | 
 | 82 | +		})  | 
 | 83 | +		require.Equal(t, clierror.Wrap(errors.New("test error"), clierror.New("failed to delete resource")), err)  | 
 | 84 | +	})  | 
 | 85 | +}  | 
 | 86 | + | 
 | 87 | +func fixDeleteCommand(writer io.Writer, getter KubeClientGetter) *cobra.Command {  | 
 | 88 | +	return buildDeleteCommand(writer, getter, fixDeleteOptions())  | 
 | 89 | +}  | 
 | 90 | + | 
 | 91 | +func fixDeleteOptions() *DeleteOptions {  | 
 | 92 | +	return &DeleteOptions{  | 
 | 93 | +		DeleteCommand: types.DeleteCommand{  | 
 | 94 | +			Description:     "delete test deploy",  | 
 | 95 | +			DescriptionLong: "use this to delete test deploy",  | 
 | 96 | +		},  | 
 | 97 | +		ResourceInfo: types.ResourceInfo{  | 
 | 98 | +			Scope:   types.NamespaceScope,  | 
 | 99 | +			Kind:    "Deployment",  | 
 | 100 | +			Group:   "apps",  | 
 | 101 | +			Version: "v1",  | 
 | 102 | +		},  | 
 | 103 | +	}  | 
 | 104 | +}  | 
 | 105 | + | 
 | 106 | +func fixDeletedUnstructuredDeployment() unstructured.Unstructured {  | 
 | 107 | +	return unstructured.Unstructured{  | 
 | 108 | +		Object: map[string]interface{}{  | 
 | 109 | +			"apiVersion": "apps/v1",  | 
 | 110 | +			"kind":       "Deployment",  | 
 | 111 | +			"metadata": map[string]interface{}{  | 
 | 112 | +				"name":      "test-deploy",  | 
 | 113 | +				"namespace": "test-namespace",  | 
 | 114 | +			},  | 
 | 115 | +		},  | 
 | 116 | +	}  | 
 | 117 | +}  | 
0 commit comments