github.com/gnolang/gno@v0.0.0-20240520182011-228e9d0192ce/tm2/pkg/iavl/key_format_test.go (about)

     1  package iavl
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/stretchr/testify/assert"
     7  )
     8  
     9  func TestKeyFormatBytes(t *testing.T) {
    10  	kf := NewKeyFormat(byte('e'), 8, 8, 8)
    11  	assert.Equal(t, []byte{'e', 0, 0, 0, 0, 0, 1, 2, 3}, kf.KeyBytes([]byte{1, 2, 3}))
    12  	assert.Equal(t, []byte{'e', 1, 2, 3, 4, 5, 6, 7, 8}, kf.KeyBytes([]byte{1, 2, 3, 4, 5, 6, 7, 8}))
    13  	assert.Equal(t, []byte{'e', 1, 2, 3, 4, 5, 6, 7, 8, 1, 2, 3, 4, 5, 6, 7, 8, 0, 0, 1, 1, 2, 2, 3, 3},
    14  		kf.KeyBytes([]byte{1, 2, 3, 4, 5, 6, 7, 8}, []byte{1, 2, 3, 4, 5, 6, 7, 8}, []byte{1, 1, 2, 2, 3, 3}))
    15  	assert.Equal(t, []byte{'e'}, kf.KeyBytes())
    16  }
    17  
    18  func TestKeyFormat(t *testing.T) {
    19  	kf := NewKeyFormat(byte('e'), 8, 8, 8)
    20  	key := []byte{'e', 0, 0, 0, 0, 0, 0, 0, 100, 0, 0, 0, 0, 0, 0, 0, 200, 0, 0, 0, 0, 0, 0, 1, 144}
    21  	var a, b, c int64 = 100, 200, 400
    22  	assert.Equal(t, key, kf.Key(a, b, c))
    23  
    24  	ao, bo, co := new(int64), new(int64), new(int64)
    25  	kf.Scan(key, ao, bo, co)
    26  	assert.Equal(t, a, *ao)
    27  	assert.Equal(t, b, *bo)
    28  	assert.Equal(t, c, *co)
    29  
    30  	bs := new([]byte)
    31  	kf.Scan(key, ao, bo, bs)
    32  	assert.Equal(t, a, *ao)
    33  	assert.Equal(t, b, *bo)
    34  	assert.Equal(t, []byte{0, 0, 0, 0, 0, 0, 1, 144}, *bs)
    35  
    36  	assert.Equal(t, []byte{'e', 0, 0, 0, 0, 0, 0, 0, 100, 0, 0, 0, 0, 0, 0, 0, 200}, kf.Key(a, b))
    37  }
    38  
    39  func TestNegativeKeys(t *testing.T) {
    40  	kf := NewKeyFormat(byte('e'), 8, 8)
    41  
    42  	var a, b int64 = -100, -200
    43  	// One's complement plus one
    44  	key := []byte{
    45  		'e',
    46  		0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, byte(0xff + a + 1),
    47  		0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, byte(0xff + b + 1),
    48  	}
    49  	assert.Equal(t, key, kf.Key(a, b))
    50  
    51  	ao, bo := new(int64), new(int64)
    52  	kf.Scan(key, ao, bo)
    53  	assert.Equal(t, a, *ao)
    54  	assert.Equal(t, b, *bo)
    55  }
    56  
    57  func TestOverflow(t *testing.T) {
    58  	kf := NewKeyFormat(byte('o'), 8, 8)
    59  
    60  	var a int64 = 1 << 62
    61  	var b uint64 = 1 << 63
    62  	key := []byte{
    63  		'o',
    64  		0x40, 0, 0, 0, 0, 0, 0, 0,
    65  		0x80, 0, 0, 0, 0, 0, 0, 0,
    66  	}
    67  	assert.Equal(t, key, kf.Key(a, b))
    68  
    69  	ao, bo := new(int64), new(int64)
    70  	kf.Scan(key, ao, bo)
    71  	assert.Equal(t, a, *ao)
    72  	assert.Equal(t, int64(b), *bo)
    73  }