package notify

import "testing"

type benchmarkTransportSecurityMode string

const (
	benchmarkTransportSecurityModernPSK  benchmarkTransportSecurityMode = "modern_psk"
	benchmarkTransportSecurityTrustedRaw benchmarkTransportSecurityMode = "trusted_raw"
)

func benchmarkApplyServerTransportSecurity(tb testing.TB, server *ServerCommon, mode benchmarkTransportSecurityMode) {
	tb.Helper()
	switch mode {
	case benchmarkTransportSecurityModernPSK:
		if err := UseModernPSKServer(server, integrationSharedSecret, integrationModernPSKOptions()); err != nil {
			tb.Fatalf("UseModernPSKServer failed: %v", err)
		}
	case benchmarkTransportSecurityTrustedRaw:
		if err := UsePSKOverExternalTransportServer(server, integrationSharedSecret, integrationModernPSKOptions()); err != nil {
			tb.Fatalf("UsePSKOverExternalTransportServer failed: %v", err)
		}
	default:
		tb.Fatalf("unsupported benchmark transport security mode %q", mode)
	}
}

func benchmarkApplyClientTransportSecurity(tb testing.TB, client *ClientCommon, mode benchmarkTransportSecurityMode) {
	tb.Helper()
	switch mode {
	case benchmarkTransportSecurityModernPSK:
		if err := UseModernPSKClient(client, integrationSharedSecret, integrationModernPSKOptions()); err != nil {
			tb.Fatalf("UseModernPSKClient failed: %v", err)
		}
	case benchmarkTransportSecurityTrustedRaw:
		if err := UsePSKOverExternalTransportClient(client, integrationSharedSecret, integrationModernPSKOptions()); err != nil {
			tb.Fatalf("UsePSKOverExternalTransportClient failed: %v", err)
		}
	default:
		tb.Fatalf("unsupported benchmark transport security mode %q", mode)
	}
}