snapshot of https://github.com/ory/kratos.git (master, 2026-09-15; blobs > 131072B omitted)

cmd/identities/import_test.go

@@ -0,0 +1,150 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package identities_test
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"os"
+	"testing"
+
+	"github.com/ory/kratos/identity"
+
+	"github.com/ory/kratos/cmd/identities"
+
+	"github.com/gofrs/uuid"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"github.com/tidwall/gjson"
+
+	"github.com/ory/kratos/driver/config"
+	kratos "github.com/ory/kratos/pkg/httpclient"
+)
+
+func TestImportCmd(t *testing.T) {
+	reg, cmd := setup(t, identities.NewImportIdentitiesCmd)
+
+	t.Run("case=imports a new identity from file", func(t *testing.T) {
+		i := kratos.CreateIdentityBody{
+			SchemaId: config.DefaultIdentityTraitsSchemaID,
+			Traits:   map[string]interface{}{},
+		}
+		ij, err := json.Marshal(i)
+		require.NoError(t, err)
+		f, err := os.CreateTemp("", "")
+		require.NoError(t, err)
+		_, err = f.Write(ij)
+		require.NoError(t, err)
+		require.NoError(t, f.Close())
+
+		stdOut := cmd.ExecNoErr(t, f.Name())
+
+		id, err := uuid.FromString(gjson.Get(stdOut, "id").String())
+		require.NoError(t, err)
+		_, err = reg.Persister().GetIdentity(context.Background(), id, identity.ExpandNothing)
+		assert.NoError(t, err)
+	})
+
+	t.Run("case=imports multiple identities from single file", func(t *testing.T) {
+		i := []kratos.CreateIdentityBody{
+			{
+				SchemaId: config.DefaultIdentityTraitsSchemaID,
+				Traits:   map[string]interface{}{},
+			},
+			{
+				SchemaId: config.DefaultIdentityTraitsSchemaID,
+				Traits:   map[string]interface{}{},
+			},
+		}
+		ij, err := json.Marshal(i)
+		require.NoError(t, err)
+		f, err := os.CreateTemp("", "")
+		require.NoError(t, err)
+		_, err = f.Write(ij)
+		require.NoError(t, err)
+		require.NoError(t, f.Close())
+
+		stdOut := cmd.ExecNoErr(t, f.Name())
+
+		id, err := uuid.FromString(gjson.Get(stdOut, "0.id").String())
+		require.NoError(t, err)
+		_, err = reg.Persister().GetIdentity(context.Background(), id, identity.ExpandNothing)
+		assert.NoError(t, err)
+
+		id, err = uuid.FromString(gjson.Get(stdOut, "1.id").String())
+		require.NoError(t, err)
+		_, err = reg.Persister().GetIdentity(context.Background(), id, identity.ExpandNothing)
+		assert.NoError(t, err)
+	})
+
+	t.Run("case=imports a new identity from STD_IN", func(t *testing.T) {
+		i := []kratos.CreateIdentityBody{
+			{
+				SchemaId: config.DefaultIdentityTraitsSchemaID,
+				Traits:   map[string]interface{}{},
+			},
+			{
+				SchemaId: config.DefaultIdentityTraitsSchemaID,
+				Traits:   map[string]interface{}{},
+			},
+		}
+		ij, err := json.Marshal(i)
+		require.NoError(t, err)
+
+		stdOut, stdErr, err := cmd.Exec(bytes.NewBuffer(ij))
+		require.NoError(t, err, "%s %s", stdOut, stdErr)
+
+		id, err := uuid.FromString(gjson.Get(stdOut, "0.id").String())
+		require.NoError(t, err)
+		_, err = reg.Persister().GetIdentity(context.Background(), id, identity.ExpandNothing)
+		assert.NoError(t, err)
+
+		id, err = uuid.FromString(gjson.Get(stdOut, "1.id").String())
+		require.NoError(t, err)
+		_, err = reg.Persister().GetIdentity(context.Background(), id, identity.ExpandNothing)
+		assert.NoError(t, err)
+	})
+
+	t.Run("case=imports multiple identities from STD_IN", func(t *testing.T) {
+		i := kratos.CreateIdentityBody{
+			SchemaId: config.DefaultIdentityTraitsSchemaID,
+			Traits:   map[string]interface{}{},
+		}
+		ij, err := json.Marshal(i)
+		require.NoError(t, err)
+
+		stdOut, stdErr, err := cmd.Exec(bytes.NewBuffer(ij))
+		require.NoError(t, err, "%s %s", stdOut, stdErr)
+
+		id, err := uuid.FromString(gjson.Get(stdOut, "id").String())
+		require.NoError(t, err)
+		_, err = reg.Persister().GetIdentity(context.Background(), id, identity.ExpandNothing)
+		assert.NoError(t, err)
+	})
+
+	t.Run("case=imports an identity with an external ID", func(t *testing.T) {
+		externalID := "customer-12345"
+		i := kratos.CreateIdentityBody{
+			SchemaId:   config.DefaultIdentityTraitsSchemaID,
+			Traits:     map[string]interface{}{},
+			ExternalId: kratos.PtrString(externalID),
+		}
+		ij, err := json.Marshal(i)
+		require.NoError(t, err)
+
+		stdOut, stdErr, err := cmd.Exec(bytes.NewBuffer(ij))
+		require.NoError(t, err, "%s %s", stdOut, stdErr)
+
+		// the external ID must be reflected in the command output
+		assert.Equal(t, externalID, gjson.Get(stdOut, "external_id").String())
+
+		// and it must be persisted on the identity
+		id, err := uuid.FromString(gjson.Get(stdOut, "id").String())
+		require.NoError(t, err)
+		stored, err := reg.Persister().GetIdentity(context.Background(), id, identity.ExpandNothing)
+		require.NoError(t, err)
+		assert.Equal(t, externalID, stored.ExternalID.String())
+	})
+}