add tests

Author: phm07

internal/cmd/base/delete_test.go

@@ -41,8 +41,16 @@ func TestDelete(t *testing.T) {
 			Args:   []string{"delete", "123"},
 			ExpOut: "Fetching fake resource\nDeleting fake resource\nFake resource 123 deleted\n",
 		},
+		"no flags multiple": {
+			Args: []string{"delete", "123", "456", "789"},
+			ExpOut: "Fetching fake resource\nDeleting fake resource\nFake resource 123 deleted\nFetching fake resource\n" +
+				"Deleting fake resource\nFake resource 456 deleted\nFetching fake resource\nDeleting fake resource\nFake resource 789 deleted\n",
+		},
 		"quiet": {
 			Args: []string{"delete", "123", "--quiet"},
 		},
+		"quiet multiple": {
+			Args: []string{"delete", "123", "456", "789", "--quiet"},
+		},
 	})
 }

internal/cmd/certificate/delete_test.go

@@ -1,6 +1,8 @@
 package certificate_test
 
 import (
+	"fmt"
+	"strings"
 	"testing"
 
 	"github.com/golang/mock/gomock"
@@ -38,3 +40,47 @@ func TestDelete(t *testing.T) {
 	assert.Empty(t, errOut)
 	assert.Equal(t, expOut, out)
 }
+
+func TestDeleteMultiple(t *testing.T) {
+	fx := testutil.NewFixture(t)
+	defer fx.Finish()
+
+	cmd := certificate.DeleteCmd.CobraCommand(fx.State())
+	fx.ExpectEnsureToken()
+
+	certs := []*hcloud.Certificate{
+		{
+			ID:   123,
+			Name: "test1",
+		},
+		{
+			ID:   456,
+			Name: "test2",
+		},
+		{
+			ID:   789,
+			Name: "test3",
+		},
+	}
+
+	expOutBuilder := strings.Builder{}
+
+	var names []string
+	for _, cert := range certs {
+		names = append(names, cert.Name)
+		expOutBuilder.WriteString(fmt.Sprintf("certificate %s deleted\n", cert.Name))
+		fx.Client.CertificateClient.EXPECT().
+			Get(gomock.Any(), cert.Name).
+			Return(cert, nil, nil)
+		fx.Client.CertificateClient.EXPECT().
+			Delete(gomock.Any(), cert).
+			Return(nil, nil)
+	}
+
+	out, errOut, err := fx.Run(cmd, names)
+	expOut := expOutBuilder.String()
+
+	assert.NoError(t, err)
+	assert.Empty(t, errOut)
+	assert.Equal(t, expOut, out)
+}

internal/cmd/firewall/delete_test.go

@@ -1,6 +1,8 @@
 package firewall_test
 
 import (
+	"fmt"
+	"strings"
 	"testing"
 
 	"github.com/golang/mock/gomock"
@@ -38,3 +40,47 @@ func TestDelete(t *testing.T) {
 	assert.Empty(t, errOut)
 	assert.Equal(t, expOut, out)
 }
+
+func TestDeleteMultiple(t *testing.T) {
+	fx := testutil.NewFixture(t)
+	defer fx.Finish()
+
+	cmd := firewall.DeleteCmd.CobraCommand(fx.State())
+	fx.ExpectEnsureToken()
+
+	firewalls := []*hcloud.Firewall{
+		{
+			ID:   123,
+			Name: "test1",
+		},
+		{
+			ID:   456,
+			Name: "test2",
+		},
+		{
+			ID:   789,
+			Name: "test3",
+		},
+	}
+
+	expOutBuilder := strings.Builder{}
+
+	var names []string
+	for _, fw := range firewalls {
+		names = append(names, fw.Name)
+		expOutBuilder.WriteString(fmt.Sprintf("firewall %s deleted\n", fw.Name))
+		fx.Client.FirewallClient.EXPECT().
+			Get(gomock.Any(), fw.Name).
+			Return(fw, nil, nil)
+		fx.Client.FirewallClient.EXPECT().
+			Delete(gomock.Any(), fw).
+			Return(nil, nil)
+	}
+
+	out, errOut, err := fx.Run(cmd, names)
+	expOut := expOutBuilder.String()
+
+	assert.NoError(t, err)
+	assert.Empty(t, errOut)
+	assert.Equal(t, expOut, out)
+}

internal/cmd/floatingip/delete_test.go

@@ -1,6 +1,8 @@
 package floatingip_test
 
 import (
+	"fmt"
+	"strings"
 	"testing"
 
 	"github.com/golang/mock/gomock"
@@ -38,3 +40,47 @@ func TestDelete(t *testing.T) {
 	assert.Empty(t, errOut)
 	assert.Equal(t, expOut, out)
 }
+
+func TestDeleteMultiple(t *testing.T) {
+	fx := testutil.NewFixture(t)
+	defer fx.Finish()
+
+	cmd := floatingip.DeleteCmd.CobraCommand(fx.State())
+	fx.ExpectEnsureToken()
+
+	ips := []*hcloud.FloatingIP{
+		{
+			ID:   123,
+			Name: "test1",
+		},
+		{
+			ID:   456,
+			Name: "test2",
+		},
+		{
+			ID:   789,
+			Name: "test3",
+		},
+	}
+
+	expOutBuilder := strings.Builder{}
+
+	var names []string
+	for _, ip := range ips {
+		names = append(names, ip.Name)
+		expOutBuilder.WriteString(fmt.Sprintf("Floating IP %s deleted\n", ip.Name))
+		fx.Client.FloatingIPClient.EXPECT().
+			Get(gomock.Any(), ip.Name).
+			Return(ip, nil, nil)
+		fx.Client.FloatingIPClient.EXPECT().
+			Delete(gomock.Any(), ip).
+			Return(nil, nil)
+	}
+
+	out, errOut, err := fx.Run(cmd, names)
+	expOut := expOutBuilder.String()
+
+	assert.NoError(t, err)
+	assert.Empty(t, errOut)
+	assert.Equal(t, expOut, out)
+}

internal/cmd/image/delete_test.go

@@ -1,6 +1,8 @@
 package image_test
 
 import (
+	"fmt"
+	"strings"
 	"testing"
 
 	"github.com/golang/mock/gomock"
@@ -38,3 +40,47 @@ func TestDelete(t *testing.T) {
 	assert.Empty(t, errOut)
 	assert.Equal(t, expOut, out)
 }
+
+func TestDeleteMultiple(t *testing.T) {
+	fx := testutil.NewFixture(t)
+	defer fx.Finish()
+
+	cmd := image.DeleteCmd.CobraCommand(fx.State())
+	fx.ExpectEnsureToken()
+
+	images := []*hcloud.Image{
+		{
+			ID:   123,
+			Name: "test1",
+		},
+		{
+			ID:   456,
+			Name: "test2",
+		},
+		{
+			ID:   789,
+			Name: "test3",
+		},
+	}
+
+	expOutBuilder := strings.Builder{}
+
+	var names []string
+	for _, img := range images {
+		names = append(names, img.Name)
+		expOutBuilder.WriteString(fmt.Sprintf("image %s deleted\n", img.Name))
+		fx.Client.ImageClient.EXPECT().
+			Get(gomock.Any(), img.Name).
+			Return(img, nil, nil)
+		fx.Client.ImageClient.EXPECT().
+			Delete(gomock.Any(), img).
+			Return(nil, nil)
+	}
+
+	out, errOut, err := fx.Run(cmd, names)
+	expOut := expOutBuilder.String()
+
+	assert.NoError(t, err)
+	assert.Empty(t, errOut)
+	assert.Equal(t, expOut, out)
+}

internal/cmd/loadbalancer/delete_test.go

@@ -1,6 +1,8 @@
 package loadbalancer_test
 
 import (
+	"fmt"
+	"strings"
 	"testing"
 
 	"github.com/golang/mock/gomock"
@@ -38,3 +40,47 @@ func TestDelete(t *testing.T) {
 	assert.Empty(t, errOut)
 	assert.Equal(t, expOut, out)
 }
+
+func TestDeleteMultiple(t *testing.T) {
+	fx := testutil.NewFixture(t)
+	defer fx.Finish()
+
+	cmd := loadbalancer.DeleteCmd.CobraCommand(fx.State())
+	fx.ExpectEnsureToken()
+
+	loadBalancers := []*hcloud.LoadBalancer{
+		{
+			ID:   123,
+			Name: "test1",
+		},
+		{
+			ID:   456,
+			Name: "test2",
+		},
+		{
+			ID:   789,
+			Name: "test3",
+		},
+	}
+
+	expOutBuilder := strings.Builder{}
+
+	var names []string
+	for _, lb := range loadBalancers {
+		names = append(names, lb.Name)
+		expOutBuilder.WriteString(fmt.Sprintf("Load Balancer %s deleted\n", lb.Name))
+		fx.Client.LoadBalancerClient.EXPECT().
+			Get(gomock.Any(), lb.Name).
+			Return(lb, nil, nil)
+		fx.Client.LoadBalancerClient.EXPECT().
+			Delete(gomock.Any(), lb).
+			Return(nil, nil)
+	}
+
+	out, errOut, err := fx.Run(cmd, names)
+	expOut := expOutBuilder.String()
+
+	assert.NoError(t, err)
+	assert.Empty(t, errOut)
+	assert.Equal(t, expOut, out)
+}

internal/cmd/network/delete_test.go

@@ -1,6 +1,8 @@
 package network_test
 
 import (
+	"fmt"
+	"strings"
 	"testing"
 
 	"github.com/golang/mock/gomock"
@@ -38,3 +40,47 @@ func TestDelete(t *testing.T) {
 	assert.Empty(t, errOut)
 	assert.Equal(t, expOut, out)
 }
+
+func TestDeleteMultiple(t *testing.T) {
+	fx := testutil.NewFixture(t)
+	defer fx.Finish()
+
+	cmd := network.DeleteCmd.CobraCommand(fx.State())
+	fx.ExpectEnsureToken()
+
+	networks := []*hcloud.Network{
+		{
+			ID:   123,
+			Name: "test1",
+		},
+		{
+			ID:   456,
+			Name: "test2",
+		},
+		{
+			ID:   789,
+			Name: "test3",
+		},
+	}
+
+	expOutBuilder := strings.Builder{}
+
+	var names []string
+	for _, net := range networks {
+		names = append(names, net.Name)
+		expOutBuilder.WriteString(fmt.Sprintf("Network %s deleted\n", net.Name))
+		fx.Client.NetworkClient.EXPECT().
+			Get(gomock.Any(), net.Name).
+			Return(net, nil, nil)
+		fx.Client.NetworkClient.EXPECT().
+			Delete(gomock.Any(), net).
+			Return(nil, nil)
+	}
+
+	out, errOut, err := fx.Run(cmd, names)
+	expOut := expOutBuilder.String()
+
+	assert.NoError(t, err)
+	assert.Empty(t, errOut)
+	assert.Equal(t, expOut, out)
+}