status-im/status-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)
}