mirror of
https://github.com/kopia/kopia.git
synced 2026-05-11 00:04:46 -04:00
09415e0c7d
Upgrades go to 1.22 and switches to new-style for loops --------- Co-authored-by: Julio López <1953782+julio-lopez@users.noreply.github.com>
73 lines
1.5 KiB
Go
73 lines
1.5 KiB
Go
package ecc
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/kopia/kopia/internal/gather"
|
|
"github.com/kopia/kopia/repo/encryption"
|
|
)
|
|
|
|
func TestComputeShares(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
testComputeShares(t, 0.1, 254, 2)
|
|
testComputeShares(t, 1, 200, 2)
|
|
testComputeShares(t, 2, 128, 2)
|
|
testComputeShares(t, 10, 128, 12)
|
|
}
|
|
|
|
func testComputeShares(t *testing.T, spaceUsedPercentage float32, expectedRequired, expectedRedundant int) {
|
|
t.Helper()
|
|
|
|
required, redundant := computeShards(spaceUsedPercentage)
|
|
|
|
require.Equal(t, expectedRequired, required)
|
|
require.Equal(t, expectedRedundant, redundant)
|
|
}
|
|
|
|
func testPutAndGet(t *testing.T, opts *Options, originalSize,
|
|
expectedEccSize int, expectedSuccess bool,
|
|
makeChanges func(impl encryption.Encryptor, data []byte),
|
|
) {
|
|
t.Helper()
|
|
|
|
impl, err := CreateAlgorithm(opts)
|
|
require.NoError(t, err)
|
|
|
|
original := make([]byte, originalSize)
|
|
for i := range originalSize {
|
|
original[i] = byte(i%255) + 1
|
|
}
|
|
|
|
output := gather.NewWriteBuffer()
|
|
|
|
err = impl.Encrypt(gather.FromSlice(original), nil, output)
|
|
require.NoError(t, err)
|
|
|
|
result := output.ToByteSlice()
|
|
require.Len(t, result, originalSize+expectedEccSize)
|
|
|
|
makeChanges(impl, result)
|
|
|
|
output = gather.NewWriteBuffer()
|
|
|
|
err = impl.Decrypt(gather.FromSlice(result), nil, output)
|
|
|
|
if expectedSuccess {
|
|
require.NoError(t, err)
|
|
require.Equal(t, original, output.ToByteSlice())
|
|
} else {
|
|
require.Error(t, err)
|
|
}
|
|
}
|
|
|
|
func flipByte(data []byte, i int) {
|
|
if data[i] >= 128 {
|
|
data[i] = 0
|
|
} else {
|
|
data[i] = 255
|
|
}
|
|
}
|