compliance_scan_job.go

package integration

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"text/template"
	"time"

	"github.com/pkg/errors"
	"github.com/stretchr/testify/require"

	"github.com/chef/automate/components/automate-cli/pkg/diagnostics"
	"github.com/chef/automate/components/automate-cli/pkg/diagnostics/lbrequest"
)

const complianceScanJobCreateTemplateStr = `
{
	"name":"{{ .Name }}",
	"type":"exec",
	"tags":[],
	"profiles":[],
	"node_selectors":[{"manager_id":"fake-manager-id"}],
	"recurrence":""
}
`

type complianceScanJobSave struct {
	ID string `json:"id"`
}

// CreateComplianceScanJobDiagnostic create the diagnostic struct for compliance scan job
func CreateComplianceScanJobDiagnostic() diagnostics.Diagnostic {
	tmpl := template.Must(template.New("compliance-scan-job").Parse(complianceScanJobCreateTemplateStr))

	return diagnostics.Diagnostic{
		Name: "compliance-scan-job",
		Tags: diagnostics.Tags{"compliance"},
		Generate: func(tstCtx diagnostics.TestContext) error {
			buf := bytes.NewBuffer([]byte{})
			ts := time.Now()
			name := "integration-diagnostic-" + ts.Format("20060102150405")
			err := tmpl.Execute(buf, struct {
				Name string
			}{
				Name: name,
			})

			if err != nil {
				return err
			}

			reqPath := "/api/v0/compliance/scanner/jobs"
			resp, err := tstCtx.DoLBRequest(
				reqPath,
				lbrequest.WithMethod("POST"),
				lbrequest.WithJSONBody(buf.String()),
			)

			if resp != nil && resp.StatusCode != 200 {
				err = errors.New("Status code not 200")
			}
			if err != nil {
				return errors.Wrapf(err, "Request POST %s failed\nBody:\n%s\nResponse:\n%+v", reqPath, buf.String(), resp)
			}

			defer func() {
				_ = resp.Body.Close()
			}()

			respUnmarshalled := make(map[string]interface{})
			err = json.NewDecoder(resp.Body).Decode(&respUnmarshalled)

			if err != nil {
				return err
			}

			id, ok := respUnmarshalled["id"].(string)

			if !ok {
				return errors.New("Could not find id in response")
			}

			tstCtx.SetValue("compliance-scan-job", complianceScanJobSave{ID: id})
			return err
		},

		Verify: func(tstCtx diagnostics.VerificationTestContext) {
			loaded := complianceScanJobSave{}

			err := tstCtx.GetValue("compliance-scan-job", &loaded)
			require.NoError(tstCtx, err, "Generated context was not found")

			reqPath := fmt.Sprintf("/api/v0/compliance/scanner/jobs/id/%s", loaded.ID)
			resp, err := tstCtx.DoLBRequest(reqPath)
			require.NoError(tstCtx, err, "Failed to make compliance API call")
			defer func() {
				_ = resp.Body.Close()
			}()

			body, _ := ioutil.ReadAll(resp.Body) // nolint: errcheck
			require.Equal(tstCtx, 200, resp.StatusCode, "Failed to GET %s\nBody:\n%s\nResponse:\n%+v", reqPath, body, resp)
		},

		Cleanup: func(tstCtx diagnostics.TestContext) error {
			loaded := complianceScanJobSave{}
			err := tstCtx.GetValue("compliance-scan-job", &loaded)

			if err != nil {
				return errors.Wrap(err, "Generated context was not found")
			}

			reqPath := fmt.Sprintf("/api/v0/compliance/scanner/jobs/id/%s", loaded.ID)
			resp, err := tstCtx.DoLBRequest(
				reqPath,
				lbrequest.WithMethod("DELETE"),
			)

			if err != nil {
				return errors.Wrapf(err, "Failed to DELETE %s", reqPath)
			}

			defer func() {
				_ = resp.Body.Close()
			}()

			if resp.StatusCode != 200 {
				return errors.New("Unexpected status code")
			}

			return nil
		},
	}
}

func init() {
	diagnostics.RegisterDiagnostic(
		CreateComplianceScanJobDiagnostic(),
	)
}