return tests + mock api call

Author: fabriciojs

commands/cloud_deploy_destroy_test.go

@@ -1,73 +1,65 @@
 package commands
 
-// import (
-// 	"errors"
-// 	"kool-dev/kool/core/environment"
-// 	"kool-dev/kool/core/shell"
-// 	"kool-dev/kool/services/cloud/api"
-// 	"strings"
-// 	"testing"
-// )
-
-// func TestNewDeployDestroyCommand(t *testing.T) {
-// 	destroy := NewKoolDeployDestroy()
-// 	cmd := NewDeployDestroyCommand(destroy)
-// 	if cmd.Use != "destroy" {
-// 		t.Errorf("bad command use: %s", cmd.Use)
-// 	}
-
-// 	if _, ok := destroy.env.(*environment.DefaultEnvStorage); !ok {
-// 		t.Error("unexpected default env on destroy")
-// 	}
-// }
-
-// type fakeDestroyCall struct {
-// 	api.DefaultEndpoint
-
-// 	err  error
-// 	resp *api.DestroyResponse
-// }
-
-// func (d *fakeDestroyCall) Call() (*api.DestroyResponse, error) {
-// 	return d.resp, d.err
-// }
-
-// func TestDeployDestroyExec(t *testing.T) {
-// 	destroy := &KoolDeployDestroy{
-// 		*(newDefaultKoolService().Fake()),
-// 		environment.NewFakeEnvStorage(),
-// 		&fakeDestroyCall{
-// 			DefaultEndpoint: *api.NewDefaultEndpoint(""),
-// 		},
-// 	}
-
-// 	destroy.env.Set("KOOL_API_TOKEN", "fake token")
-// 	destroy.env.Set("KOOL_API_URL", "fake-url")
-
-// 	args := []string{}
-
-// 	if err := destroy.Execute(args); !strings.Contains(err.Error(), "missing deploy domain") {
-// 		t.Errorf("unexpected error - expected missing deploy domain, got: %v", err)
-// 	}
-
-// 	destroy.env.Set("KOOL_DEPLOY_DOMAIN", "domain.com")
-
-// 	destroy.apiDestroy.(*fakeDestroyCall).err = errors.New("failed call")
-
-// 	if err := destroy.Execute(args); !strings.Contains(err.Error(), "failed call") {
-// 		t.Errorf("unexpected error - expected failed call, got: %v", err)
-// 	}
-
-// 	destroy.apiDestroy.(*fakeDestroyCall).err = nil
-// 	resp := new(api.DestroyResponse)
-// 	resp.Environment.ID = 100
-// 	destroy.apiDestroy.(*fakeDestroyCall).resp = resp
-
-// 	if err := destroy.Execute(args); err != nil {
-// 		t.Errorf("unexpected error, got: %v", err)
-// 	}
-
-// 	if !strings.Contains(destroy.shell.(*shell.FakeShell).SuccessOutput[0].(string), "ID: 100") {
-// 		t.Errorf("did not get success message")
-// 	}
-// }
+import (
+	"errors"
+	"kool-dev/kool/core/environment"
+	"kool-dev/kool/core/shell"
+	"kool-dev/kool/services/cloud/api"
+	"strings"
+	"testing"
+)
+
+func TestNewDeployDestroyCommand(t *testing.T) {
+	destroy := NewKoolDeployDestroy()
+	cmd := NewDeployDestroyCommand(destroy)
+	if cmd.Use != "destroy" {
+		t.Errorf("bad command use: %s", cmd.Use)
+	}
+
+	if _, ok := destroy.env.(*environment.DefaultEnvStorage); !ok {
+		t.Error("unexpected default env on destroy")
+	}
+}
+
+func TestDeployDestroyExec(t *testing.T) {
+	destroy := &KoolDeployDestroy{
+		*(newDefaultKoolService().Fake()),
+		environment.NewFakeEnvStorage(),
+		*api.NewDeployDestroy(),
+	}
+
+	destroy.env.Set("KOOL_API_TOKEN", "fake token")
+	destroy.env.Set("KOOL_API_URL", "fake-url")
+
+	args := []string{}
+
+	if err := destroy.Execute(args); !strings.Contains(err.Error(), "missing deploy domain") {
+		t.Errorf("unexpected error - expected missing deploy domain, got: %v", err)
+	}
+
+	destroy.env.Set("KOOL_DEPLOY_DOMAIN", "domain.com")
+
+	destroy.apiDestroy.Endpoint.(*api.DefaultEndpoint).Fake()
+	destroy.apiDestroy.Endpoint.(*api.DefaultEndpoint).MockErr(errors.New("failed call"))
+
+	if err := destroy.Execute(args); !strings.Contains(err.Error(), "failed call") {
+		t.Errorf("unexpected error - expected failed call, got: %v", err)
+	}
+
+	destroy.apiDestroy.Endpoint.(*api.DefaultEndpoint).MockErr(nil)
+	destroy.apiDestroy.Endpoint.(*api.DefaultEndpoint).MockResp(&api.DeployDestroyResponse{
+		Environment: struct {
+			ID int `json:"id"`
+		}{
+			ID: 100,
+		},
+	})
+
+	if err := destroy.Execute(args); err != nil {
+		t.Errorf("unexpected error, got: %v", err)
+	}
+
+	if !strings.Contains(destroy.shell.(*shell.FakeShell).SuccessOutput[0].(string), "ID: 100") {
+		t.Errorf("did not get success message")
+	}
+}

services/cloud/api/endpoint.go

@@ -48,6 +48,10 @@ type DefaultEndpoint struct {
 
 	postBodyBuff bytes.Buffer
 	postBodyFmtr *multipart.Writer
+
+	fake     bool
+	mockErr  error
+	mockResp any
 }
 
 // NewDefaultEndpoint creates an Endpoint with given method
@@ -150,6 +154,16 @@ func (e *DefaultEndpoint) DoCall() (err error) {
 		verbose = e.env.IsTrue("KOOL_VERBOSE")
 	)
 
+	if e.fake {
+		// fake call
+		err = e.mockErr
+		if e.mockResp != nil {
+			raw, _ := json.Marshal(e.mockResp)
+			json.Unmarshal(raw, e.response)
+		}
+		return
+	}
+
 	if e.postBodyFmtr != nil {
 		e.SetContentType(e.postBodyFmtr.FormDataContentType())
 		e.postBodyFmtr.Close()
@@ -254,3 +268,15 @@ func (e *DefaultEndpoint) initPostBody() {
 		e.postBodyFmtr = multipart.NewWriter(&e.postBodyBuff)
 	}
 }
+
+func (e *DefaultEndpoint) Fake() {
+	e.fake = true
+}
+
+func (e *DefaultEndpoint) MockErr(err error) {
+	e.mockErr = err
+}
+
+func (e *DefaultEndpoint) MockResp(i any) {
+	e.mockResp = i
+}