storj/satellite/orders/encryptionkey_test.go
Egon Elbre 735dc6e163 satellite/orders: add encryption keys
This adds EncryptionKey definition that can be used as a flag.
These order.EncryptionKey-s will be used to encrypt data in
order limits.

This helps to avoid storing lots of transient data in the
main database.

This code doesn't yet contain encryption itself.

Change-Id: I2efae102a89b851d33342a0106f8d8b3f35119bb
2020-06-30 15:03:14 +00:00

138 lines
4.5 KiB
Go

// Copyright (C) 2020 Storj Labs, Inc.
// See LICENSE for copying information.
package orders_test
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"storj.io/common/storj"
"storj.io/storj/satellite/orders"
)
func TestEncryptionKey_Set_Valid(t *testing.T) {
type Test struct {
Hex string
Key orders.EncryptionKey
}
tests := []Test{
{
Hex: `0100000000000000=0100000000000000000000000000000000000000000000000000000000000000`,
Key: orders.EncryptionKey{
ID: orders.EncryptionKeyID{0x01},
Key: storj.Key{0x01},
},
},
{
Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF`,
Key: orders.EncryptionKey{
ID: orders.EncryptionKeyID{0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0xFF},
Key: storj.Key{
0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88,
0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88,
0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88,
0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0xFF,
},
},
},
}
for _, test := range tests {
var got orders.EncryptionKey
err := got.Set(test.Hex)
assert.NoError(t, err, test.Hex)
assert.Equal(t, test.Key, got, test.Hex)
}
}
func TestEncryptionKey_Set_Invalid(t *testing.T) {
type Test struct {
Hex string
}
tests := []Test{
{Hex: ``},
{Hex: `=`},
{Hex: `01=`},
{Hex: `=01`},
{Hex: `1=1`},
{Hex: `=1122334455667788112233445566778811223344556677881122334455667788`},
{Hex: `112233445566778=1122334455667788112233445566778811223344556677881122334455667788`},
{Hex: `1122334455667788=`},
{Hex: `1122334455667788=112233445566778811223344556677881122334455667788112233445566778`},
{Hex: `11223344556677QQ=11223344556677881122334455667788112233445566778811223344556677QQ`},
}
for _, test := range tests {
var got orders.EncryptionKey
err := got.Set(test.Hex)
assert.Error(t, err, test.Hex)
}
}
func TestEncryptionKeys_Set_Valid(t *testing.T) {
var keys orders.EncryptionKeys
err := keys.Set(`11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,0100000000000000=0100000000000000000000000000000000000000000000000000000000000000`)
require.NoError(t, err)
first := orders.EncryptionKey{
ID: orders.EncryptionKeyID{0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0xFF},
Key: storj.Key{
0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88,
0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88,
0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88,
0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0xFF,
},
}
second := orders.EncryptionKey{
ID: orders.EncryptionKeyID{0x01},
Key: storj.Key{0x01},
}
assert.Equal(t, first, keys.Default)
assert.EqualValues(t, []orders.EncryptionKey{first, second}, keys.List)
assert.Equal(t, first.Key, keys.KeyByID[first.ID])
assert.Equal(t, second.Key, keys.KeyByID[second.ID])
}
func TestEncryptionKeys_Set_Invalid(t *testing.T) {
type Test struct {
Hex string
}
tests := []Test{
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,=`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,01=`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,=01`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,1=1`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,=1122334455667788112233445566778811223344556677881122334455667788`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,112233445566778=1122334455667788112233445566778811223344556677881122334455667788`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,1122334455667788=`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,1122334455667788=112233445566778811223344556677881122334455667788112233445566778`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,11223344556677QQ=11223344556677881122334455667788112233445566778811223344556677QQ`},
}
for _, test := range tests {
var got orders.EncryptionKeys
err := got.Set(test.Hex)
assert.Error(t, err, test.Hex)
}
}