71 lines
3.0 KiB
Go
71 lines
3.0 KiB
Go
package emojihash
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/status-im/status-go/protocol/identity"
|
|
)
|
|
|
|
func TestGenerateFor(t *testing.T) {
|
|
checker := func(pubkey string, expected *[](string)) {
|
|
emojihash, err := GenerateFor(pubkey)
|
|
require.NoError(t, err)
|
|
if !reflect.DeepEqual(emojihash, *expected) {
|
|
t.Fatalf("invalid emojihash %v != %v", emojihash, *expected)
|
|
}
|
|
}
|
|
|
|
checker("0x04e25da6994ea2dc4ac70727e07eca153ae92bf7609db7befb7ebdceaad348f4fc55bbe90abf9501176301db5aa103fc0eb3bc3750272a26c424a10887db2a7ea8",
|
|
&[](string){"👦🏽", "🦹🏻", "👶🏿", "🛁", "🌁", "🙌🏻", "🙇🏽♂️", "🙌🏾", "🤥", "🐛", "👩🏽🔧", "🔧", "⚙️", "🧒🏽"})
|
|
|
|
checker("0x0400000000000000000000000000000000000000000000000000000000000000014218F20AE6C646B363DB68605822FB14264CA8D2587FDD6FBC750D587E76A7EE",
|
|
&[](string){"😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀"})
|
|
|
|
checker("0x04000000000000000000000000000000000000000010000000000000000000000033600332D373318ECC2F212A30A5750D2EAC827B6A32B33D326CCF369B12B1BE",
|
|
&[](string){"😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀", (emojisAlphabet)[1]})
|
|
|
|
checker("0x040000000000000000000000000000000000000000200000000000000000000000353050BFE33B724E60A0C600FBA565A9B62217B1BD35BF9848F2AB847C598B30",
|
|
&[](string){"😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀", "😀", (emojisAlphabet)[2]})
|
|
}
|
|
|
|
func TestEmojiHashOfInvalidKey(t *testing.T) {
|
|
checker := func(pubkey string) {
|
|
_, err := GenerateFor(pubkey)
|
|
require.Error(t, err)
|
|
}
|
|
checker("abc")
|
|
checker("0x01")
|
|
checker("0x01e25da6994ea2dc4ac70727e07eca153ae92bf7609db7befb7ebdceaad348f4fc55bbe90abf9501176301db5aa103fc0eb3bc3750272a26c424a10887db2a7ea8")
|
|
checker("0x04425da6994ea2dc4ac70727e07eca153ae92bf7609db7befb7ebdceaad348f4fc55bbe90abf9501176301db5aa103fc0eb3bc3750272a26c424a10887db2a7ea8")
|
|
}
|
|
|
|
func TestToEmojiHash(t *testing.T) {
|
|
alphabet := [](string){"😇", "🤐", "🥵", "🙊", "🤌"}
|
|
|
|
checker := func(valueStr string, hashLen int, expected *[](string)) {
|
|
value := identity.ToBigInt(t, valueStr)
|
|
res, err := toEmojiHash(value, hashLen, &alphabet)
|
|
require.NoError(t, err)
|
|
if !reflect.DeepEqual(res, *expected) {
|
|
t.Fatalf("invalid emojihash conversion %v != %v", res, *expected)
|
|
}
|
|
}
|
|
|
|
checker("777", 5, &[](string){"🤐", "🤐", "🤐", "😇", "🥵"})
|
|
checker("777", 0, &[](string){"🤐", "🤐", "🤐", "😇", "🥵"})
|
|
checker("777", 10, &[](string){"😇", "😇", "😇", "😇", "😇", "🤐", "🤐", "🤐", "😇", "🥵"})
|
|
|
|
// 20bytes of data described by 14 emojis requires at least 2757 length alphabet
|
|
alphabet = make([](string), 2757)
|
|
val := identity.ToBigInt(t, "0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF") // 20 bytes
|
|
_, err := toEmojiHash(val, 14, &alphabet)
|
|
require.NoError(t, err)
|
|
|
|
alphabet = make([](string), 2757-1)
|
|
_, err = toEmojiHash(val, 14, &alphabet)
|
|
require.Error(t, err)
|
|
}
|