caddy/modules/caddytls/leaffolderloader_test.go

package caddytls

import (
	"context"
	"encoding/pem"
	"os"
	"strings"
	"testing"

	"github.com/caddyserver/caddy/v2"
)

func TestLeafFolderLoader(t *testing.T) {
	fl := LeafFolderLoader{Folders: []string{"../../caddytest"}}
	fl.Provision(caddy.Context{Context: context.Background()})

	out, err := fl.LoadLeafCertificates()
	if err != nil {
		t.Errorf("Leaf certs folder loading test failed: %v", err)
	}
	if len(out) != 1 {
		t.Errorf("Error loading leaf cert in memory struct")
		return
	}
	pemBytes := pem.EncodeToMemory(&pem.Block{Type: "CERTIFICATE", Bytes: out[0].Raw})
	pemFileBytes, err := os.ReadFile("../../caddytest/leafcert.pem")
	if err != nil {
		t.Errorf("Unable to read the example certificate from the file")
	}

	// Remove /r because windows.
	pemFileString := strings.ReplaceAll(string(pemFileBytes), "\r\n", "\n")

	if string(pemBytes) != pemFileString {
		t.Errorf("Leaf Certificate Folder Loader: Failed to load the correct certificate")
	}
}