process_test.go

package client

import (
	"context"
	"net/http"
	"testing"

	"github.com/cloudfoundry/go-cfclient/v3/resource"
	"github.com/cloudfoundry/go-cfclient/v3/testutil"
)

func TestProcesses(t *testing.T) {
	g := testutil.NewObjectJSONGenerator()
	process := g.Process().JSON
	process2 := g.Process().JSON
	process3 := g.Process().JSON
	process4 := g.Process().JSON
	processStats := g.ProcessStats().JSON

	tests := []RouteTest{
		{
			Description: "Get process",
			Route: testutil.MockRoute{
				Method:   "GET",
				Endpoint: "/v3/processes/ec4ff362-60c5-47a0-8246-2a134537c606",
				Output:   g.Single(process),
				Status:   http.StatusOK,
			},
			Expected: process,
			Action: func(c *Client, t *testing.T) (any, error) {
				return c.Processes.Get(context.Background(), "ec4ff362-60c5-47a0-8246-2a134537c606")
			},
		},
		{
			Description: "Get process stats",
			Route: testutil.MockRoute{
				Method:   "GET",
				Endpoint: "/v3/processes/ec4ff362-60c5-47a0-8246-2a134537c606/stats",
				Output:   g.Single(processStats),
				Status:   http.StatusOK,
			},
			Expected: processStats,
			Action: func(c *Client, t *testing.T) (any, error) {
				return c.Processes.GetStats(context.Background(), "ec4ff362-60c5-47a0-8246-2a134537c606")
			},
		},
		{
			Description: "Get app process stats",
			Route: testutil.MockRoute{
				Method:   "GET",
				Endpoint: "/v3/apps/2a550283-9245-493e-af36-5e4b8703f896/processes/web/stats",
				Output:   g.Single(processStats),
				Status:   http.StatusOK,
			},
			Expected: processStats,
			Action: func(c *Client, t *testing.T) (any, error) {
				return c.Processes.GetStatsForApp(context.Background(), "2a550283-9245-493e-af36-5e4b8703f896", "web")
			},
		},
		{
			Description: "List all processes",
			Route: testutil.MockRoute{
				Method:   "GET",
				Endpoint: "/v3/processes",
				Output:   g.Paged([]string{process, process2}, []string{process3, process4}),
				Status:   http.StatusOK,
			},
			Expected: g.Array(process, process2, process3, process4),
			Action: func(c *Client, t *testing.T) (any, error) {
				return c.Processes.ListAll(context.Background(), nil)
			},
		},
		{
			Description: "List all processes for app",
			Route: testutil.MockRoute{
				Method:   "GET",
				Endpoint: "/v3/apps/2a550283-9245-493e-af36-5e4b8703f896/processes",
				Output:   g.Paged([]string{process, process2}, []string{process3, process4}),
				Status:   http.StatusOK,
			},
			Expected: g.Array(process, process2, process3, process4),
			Action: func(c *Client, t *testing.T) (any, error) {
				return c.Processes.ListForAppAll(context.Background(), "2a550283-9245-493e-af36-5e4b8703f896", nil)
			},
		},
		{
			Description: "Scale a process",
			Route: testutil.MockRoute{
				Method:   "POST",
				Endpoint: "/v3/processes/ec4ff362-60c5-47a0-8246-2a134537c606/actions/scale",
				Output:   g.Single(process),
				Status:   http.StatusOK,
				PostForm: `{
					"instances": 5,
					"memory_in_mb": 256,
					"disk_in_mb": 1024,
					"log_rate_limit_in_bytes_per_second": 1024
				  }`,
			},
			Expected: process,
			Action: func(c *Client, t *testing.T) (any, error) {
				r := resource.NewProcessScale().
					WithInstances(5).
					WithMemoryInMB(256).
					WithDiskInMB(1024).
					WithLogRateLimitInBytesPerSecond(1024)
				return c.Processes.Scale(context.Background(), "ec4ff362-60c5-47a0-8246-2a134537c606", r)
			},
		},
		{
			Description: "Update a process health and readiness check",
			Route: testutil.MockRoute{
				Method:   "PATCH",
				Endpoint: "/v3/processes/ec4ff362-60c5-47a0-8246-2a134537c606",
				Output:   g.Single(process),
				Status:   http.StatusOK,
				PostForm: `{
					"command": "rackup",
					"health_check": {
						"type": "http",
						"data": {
							"timeout": 60,
							"invocation_timeout": 5,
							"interval": 10,
							"endpoint": "/health"
						}
					},
					"readiness_health_check": {
						"type": "http",
						"data": {
							"invocation_timeout": 15,
							"interval": 30,
							"endpoint": "/ready"
						}
					}
				}`,
			},
			Expected: process,
			Action: func(c *Client, t *testing.T) (any, error) {
				r := resource.NewProcessUpdate().
					WithCommand("rackup").
					WithHealthCheckType("http").
					WithHealthCheckTimeout(60).
					WithHealthCheckInterval(10).
					WithHealthCheckInvocationTimeout(5).
					WithHealthCheckEndpoint("/health").
					WithReadinessCheckType("http").
					WithReadinessCheckInterval(30).
					WithReadinessCheckInvocationTimeout(15).
					WithReadinessCheckEndpoint("/ready")
				return c.Processes.Update(context.Background(), "ec4ff362-60c5-47a0-8246-2a134537c606", r)
			},
		},
		{
			Description: "Terminate process",
			Route: testutil.MockRoute{
				Method:   "DELETE",
				Endpoint: "/v3/processes/ec4ff362-60c5-47a0-8246-2a134537c606/instances/0",
				Status:   http.StatusAccepted,
			},
			Action: func(c *Client, t *testing.T) (any, error) {
				return nil, c.Processes.Terminate(context.Background(), "ec4ff362-60c5-47a0-8246-2a134537c606", 0)
			},
		},
	}
	ExecuteTests(tests, t)
}