upstream-ory-kratos/cmd/identities/validate_test.go

// Copyright © 2023 Ory Corp
// SPDX-License-Identifier: Apache-2.0

package identities_test

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

	"github.com/ory/kratos/cmd/identities"

	"github.com/spf13/cobra"
	"github.com/stretchr/testify/assert"
)

func TestValidateIdentity(t *testing.T) {
	var testCmd = func() (*cobra.Command, *bytes.Buffer, *bytes.Buffer) {
		out, err := &bytes.Buffer{}, &bytes.Buffer{}
		cmd := &cobra.Command{}
		cmd.SetOut(out)
		cmd.SetErr(err)
		return cmd, out, err
	}

	var testSchemaGetter = func(resp map[string]interface{}) identities.SchemaGetter {
		return func(_ context.Context, _ string) (map[string]interface{}, *http.Response, error) {
			return resp, nil, nil
		}
	}

	for i, tc := range []struct {
		description           string
		payload               string
		expectedRequiredError string
		identitySchema        map[string]interface{}
	}{
		{
			description:           "requires schema_id from swagger schema",
			payload:               "{}",
			expectedRequiredError: "schema_id",
		},
		{
			description:           "requires traits from swagger schema",
			payload:               `{"schema_id": "case2"}`,
			expectedRequiredError: "traits",
		},
		{
			description:           "requires custom trait key",
			payload:               `{"schema_id": "case3", "traits": {}}`,
			expectedRequiredError: "random-property",
			identitySchema: map[string]interface{}{
				"type":     "object",
				"required": []string{"random-property"},
				"properties": map[string]interface{}{
					"random-property": map[string]interface{}{
						"type": "string",
					},
				},
			},
		},
	} {
		t.Run(fmt.Sprintf("case=%d/description=%s", i, tc.description), func(t *testing.T) {
			cmd, stdOut, stdErr := testCmd()

			if tc.identitySchema == nil {
				tc.identitySchema = map[string]interface{}{}
			}
			err := identities.ValidateIdentity(cmd, "test identity", tc.payload, testSchemaGetter(tc.identitySchema))
			assert.Error(t, err, stdOut.String(), stdErr.String())
			assert.Len(t, stdOut.String(), 0, stdErr.String())
			assert.Contains(t, stdErr.String(), "required", stdOut.String())
			assert.Contains(t, stdErr.String(), tc.expectedRequiredError, stdOut.String())
		})
	}
}