constellation/state/internal/test/benchmark_test.go
Thomas Tendyck bd63aa3c6b add license headers
sed -i '1i/*\nCopyright (c) Edgeless Systems GmbH\n\nSPDX-License-Identifier: AGPL-3.0-only\n*/\n' `grep -rL --include='*.go' 'DO NOT EDIT'`
gofumpt -w .
2022-09-05 09:17:25 +02:00

115 lines
2.3 KiB
Go

//go:build integration
/*
Copyright (c) Edgeless Systems GmbH
SPDX-License-Identifier: AGPL-3.0-only
*/
package integration
import (
"fmt"
"math"
"testing"
"github.com/edgelesssys/constellation/internal/logger"
"github.com/edgelesssys/constellation/state/internal/mapper"
"github.com/martinjungblut/go-cryptsetup"
"go.uber.org/zap/zapcore"
)
func BenchmarkMapper(b *testing.B) {
cryptsetup.SetDebugLevel(cryptsetup.CRYPT_LOG_ERROR)
cryptsetup.SetLogCallback(func(_ int, message string) { fmt.Println(message) })
testPath := *diskPath
if testPath == "" {
// no disk specified, use 1GB loopback disk
testPath = devicePath
if err := setup(1); err != nil {
b.Fatal("Failed to setup test environment:", err)
}
defer func() {
if err := teardown(); err != nil {
b.Fatal("failed to delete test disk:", err)
}
}()
}
passphrase := "benchmark"
mapper, err := mapper.New(testPath, logger.New(logger.PlainLog, zapcore.InfoLevel))
if err != nil {
b.Fatal("Failed to create mapper:", err)
}
defer mapper.Close()
if err := mapper.FormatDisk(passphrase); err != nil {
b.Fatal("Failed to format disk:", err)
}
testCases := map[string]struct {
wipeBlockSize int
}{
"16KiB": {
wipeBlockSize: int(math.Pow(2, 14)),
},
"32KiB": {
wipeBlockSize: int(math.Pow(2, 15)),
},
"64KiB": {
wipeBlockSize: int(math.Pow(2, 16)),
},
"128KiB": {
wipeBlockSize: int(math.Pow(2, 17)),
},
"256KiB": {
wipeBlockSize: int(math.Pow(2, 18)),
},
"512KiB": {
wipeBlockSize: int(math.Pow(2, 19)),
},
"1MiB": {
wipeBlockSize: int(math.Pow(2, 20)),
},
"2MiB": {
wipeBlockSize: int(math.Pow(2, 21)),
},
"4MiB": {
wipeBlockSize: int(math.Pow(2, 22)),
},
"8MiB": {
wipeBlockSize: int(math.Pow(2, 23)),
},
"16MiB": {
wipeBlockSize: int(math.Pow(2, 24)),
},
"32MiB": {
wipeBlockSize: int(math.Pow(2, 25)),
},
"64MiB": {
wipeBlockSize: int(math.Pow(2, 26)),
},
"128MiB": {
wipeBlockSize: int(math.Pow(2, 27)),
},
"256MiB": {
wipeBlockSize: int(math.Pow(2, 28)),
},
"512MiB": {
wipeBlockSize: int(math.Pow(2, 29)),
},
}
for name, tc := range testCases {
b.Run(name, func(b *testing.B) {
for i := 0; i < b.N; i++ {
if err := mapper.Wipe(tc.wipeBlockSize); err != nil {
b.Fatal("Failed to wipe disk:", err)
}
}
})
}
}