43 lines
1.6 KiB
Go
43 lines
1.6 KiB
Go
|
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)
|
||
|
|
}
|
||
|
|
}
|