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

courier/http_test.go

@@ -0,0 +1,169 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package courier_test
+
+import (
+	"encoding/base64"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/gofrs/uuid"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"github.com/tidwall/gjson"
+
+	"github.com/ory/kratos/courier/template"
+	"github.com/ory/kratos/courier/template/email"
+	"github.com/ory/kratos/driver/config"
+	"github.com/ory/kratos/pkg"
+	"github.com/ory/x/configx"
+)
+
+func TestQueueHTTPEmail(t *testing.T) {
+	type sendEmailRequestBody struct {
+		IdentityID       string `json:"identity_id"`
+		IdentityEmail    string `json:"identity_email"`
+		Recipient        string `json:"recipient"`
+		TemplateType     string `json:"template_type"`
+		To               string `json:"to"`
+		RecoveryCode     string `json:"recovery_code"`
+		RecoveryURL      string `json:"recovery_url"`
+		VerificationURL  string `json:"verification_url"`
+		VerificationCode string `json:"verification_code"`
+		Body             string `json:"body"`
+		HTMLBody         string `json:"html_body"`
+		Subject          string `json:"subject"`
+	}
+
+	expectedEmail := []*email.TestStubModel{
+		{
+			To:       "test-2@test.com",
+			Subject:  "test-mailer-subject-1",
+			Body:     "test-mailer-body-1",
+			HTMLBody: "<html><body>test-mailer-body-html-1</body></html>",
+		},
+		{
+			To:      "test-2@test.com",
+			Subject: "test-mailer-subject-2",
+			Body:    "test-mailer-body-2",
+		},
+	}
+
+	actual := make(chan sendEmailRequestBody, len(expectedEmail))
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		rb, err := io.ReadAll(r.Body)
+		require.NoError(t, err)
+
+		var body sendEmailRequestBody
+
+		err = json.Unmarshal(rb, &body)
+		require.NoError(t, err)
+
+		assert.NotEmpty(t, r.Header["Authorization"])
+		assert.Equal(t, "Basic bWU6MTIzNDU=", r.Header["Authorization"][0])
+
+		actual <- body
+	}))
+	t.Cleanup(srv.Close)
+
+	requestConfig := fmt.Sprintf(`{
+		"url": "%s",
+		"method": "POST",
+		"auth": {
+			"type": "basic_auth",
+			"config": {
+				"user":     "me",
+				"password": "12345"
+			}
+		},
+		"body": "file://./stub/request.config.mailer.jsonnet"
+	}`, srv.URL)
+
+	_, reg := pkg.NewFastRegistryWithMocks(t, configx.WithValues(map[string]any{
+		config.ViperKeyCourierDeliveryStrategy:  "http",
+		config.ViperKeyCourierHTTPRequestConfig: requestConfig,
+		config.ViperKeyCourierSMTPURL:           "http://foo.url",
+	}))
+
+	courier, err := reg.Courier(t.Context())
+	require.NoError(t, err)
+
+	for _, message := range expectedEmail {
+		id, err := courier.QueueEmail(t.Context(), email.NewTestStub(message))
+		require.NoError(t, err)
+		require.NotEqual(t, uuid.Nil, id)
+	}
+
+	require.NoError(t, courier.DispatchQueue(t.Context()))
+	close(actual)
+
+	require.Len(t, actual, len(expectedEmail))
+
+	i := 0
+	for message := range actual {
+		expected := expectedEmail[i]
+
+		assert.Equal(t, expected.To, message.To)
+		assert.Equal(t, expected.Body, message.Body)
+		assert.Equal(t, expected.HTMLBody, message.HTMLBody)
+		assert.Equal(t, expected.Subject, message.Subject)
+
+		i++
+	}
+}
+
+func TestQueueHTTPEmailWithOAuth2LoginRequest(t *testing.T) {
+	actual := make(chan []byte, 1)
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		rb, err := io.ReadAll(r.Body)
+		require.NoError(t, err)
+		actual <- rb
+	}))
+	t.Cleanup(srv.Close)
+
+	requestConfig := fmt.Sprintf(`{
+		"url": "%s",
+		"method": "POST",
+		"body": "base64://%s"
+	}`, srv.URL, base64.StdEncoding.EncodeToString([]byte("function(ctx) ctx")))
+
+	_, reg := pkg.NewFastRegistryWithMocks(t, configx.WithValues(map[string]any{
+		config.ViperKeyCourierDeliveryStrategy:  "http",
+		config.ViperKeyCourierHTTPRequestConfig: requestConfig,
+		config.ViperKeyCourierSMTPURL:           "http://foo.url",
+	}))
+
+	c, err := reg.Courier(t.Context())
+	require.NoError(t, err)
+
+	_, err = c.QueueEmail(t.Context(), email.NewLoginCodeValid(reg, &email.LoginCodeValidModel{
+		To:        "test@test.com",
+		LoginCode: "123456",
+		OAuth2LoginRequest: &template.OAuth2LoginRequest{
+			Challenge: "test-challenge",
+			Client: template.OAuth2Client{
+				ClientID:   "test-client-id",
+				ClientName: "Test Client",
+				ClientURI:  "https://client.example.com",
+				LogoURI:    "https://client.example.com/logo.png",
+				Metadata:   map[string]any{"brand": "acme"},
+			},
+		},
+	}))
+	require.NoError(t, err)
+
+	require.NoError(t, c.DispatchQueue(t.Context()))
+
+	body := <-actual
+	assert.Equal(t, "test-challenge", gjson.GetBytes(body, "template_data.oauth2_login_request.challenge").String())
+	assert.Equal(t, "test-client-id", gjson.GetBytes(body, "template_data.oauth2_login_request.client.client_id").String())
+	assert.Equal(t, "Test Client", gjson.GetBytes(body, "template_data.oauth2_login_request.client.client_name").String())
+	assert.Equal(t, "https://client.example.com", gjson.GetBytes(body, "template_data.oauth2_login_request.client.client_uri").String())
+	assert.Equal(t, "https://client.example.com/logo.png", gjson.GetBytes(body, "template_data.oauth2_login_request.client.logo_uri").String())
+	assert.Equal(t, "acme", gjson.GetBytes(body, "template_data.oauth2_login_request.client.metadata.brand").String())
+}