1095 lines
29 KiB
Go
1095 lines
29 KiB
Go
package protocol
|
||
|
||
import (
|
||
"fmt"
|
||
"reflect"
|
||
"strings"
|
||
"testing"
|
||
|
||
"github.com/stretchr/testify/require"
|
||
|
||
"github.com/status-im/status-go/eth-node/crypto"
|
||
"github.com/status-im/status-go/logutils"
|
||
)
|
||
|
||
func TestRePosRegex(t *testing.T) {
|
||
testCases := []struct {
|
||
input string
|
||
expected bool
|
||
}{
|
||
{"@", true},
|
||
{"~", true},
|
||
{"\\", true},
|
||
{"*", true},
|
||
{"_", true},
|
||
{"\n", true},
|
||
{"`", true},
|
||
{"a", false},
|
||
{"#", false},
|
||
}
|
||
|
||
for _, tc := range testCases {
|
||
actual := specialCharsRegex.MatchString(tc.input)
|
||
if actual != tc.expected {
|
||
t.Errorf("Unexpected match result for input '%s': expected=%v, actual=%v", tc.input, tc.expected, actual)
|
||
}
|
||
}
|
||
}
|
||
|
||
func TestRePos(t *testing.T) {
|
||
// Test case 1: Empty string
|
||
s1 := ""
|
||
var want1 []specialCharLocation
|
||
if got1 := rePos(s1); !reflect.DeepEqual(got1, want1) {
|
||
t.Errorf("rePos(%q) = %v, want %v", s1, got1, want1)
|
||
}
|
||
|
||
// Test case 2: Single match
|
||
s2 := "test@string"
|
||
want2 := []specialCharLocation{{4, "@"}}
|
||
if got2 := rePos(s2); !reflect.DeepEqual(got2, want2) {
|
||
t.Errorf("rePos(%q) = %v, want %v", s2, got2, want2)
|
||
}
|
||
|
||
// Test case 3: Multiple matches
|
||
s3 := "this is a test string@with multiple@matches"
|
||
want3 := []specialCharLocation{{21, "@"}, {35, "@"}}
|
||
if got3 := rePos(s3); !reflect.DeepEqual(got3, want3) {
|
||
t.Errorf("rePos(%q) = %v, want %v", s3, got3, want3)
|
||
}
|
||
|
||
// Test case 4: No matches
|
||
s4 := "this is a test string with no matches"
|
||
var want4 []specialCharLocation
|
||
if got4 := rePos(s4); !reflect.DeepEqual(got4, want4) {
|
||
t.Errorf("rePos(%q) = %v, want %v", s4, got4, want4)
|
||
}
|
||
|
||
// Test case 5: Matches at the beginning and end
|
||
s5 := "@this is a test string@"
|
||
want5 := []specialCharLocation{{0, "@"}, {22, "@"}}
|
||
if got5 := rePos(s5); !reflect.DeepEqual(got5, want5) {
|
||
t.Errorf("rePos(%q) = %v, want %v", s5, got5, want5)
|
||
}
|
||
|
||
// Test case 6: special characters
|
||
s6 := "Привет @testm1 "
|
||
want6 := []specialCharLocation{{7, "@"}}
|
||
if got6 := rePos(s6); !reflect.DeepEqual(got6, want6) {
|
||
t.Errorf("rePos(%q) = %v, want %v", s6, got6, want6)
|
||
}
|
||
|
||
}
|
||
|
||
func TestReplaceMentions(t *testing.T) {
|
||
users := map[string]*MentionableUser{
|
||
"0xpk1": {
|
||
Contact: &Contact{
|
||
ID: "0xpk1",
|
||
LocalNickname: "User Number One",
|
||
},
|
||
},
|
||
"0xpk2": {
|
||
Contact: &Contact{
|
||
ID: "0xpk2",
|
||
LocalNickname: "user2",
|
||
ENSVerified: true,
|
||
EnsName: "User Number Two",
|
||
},
|
||
},
|
||
"0xpk3": {
|
||
Contact: &Contact{
|
||
ID: "0xpk3",
|
||
LocalNickname: "user3",
|
||
ENSVerified: true,
|
||
EnsName: "User Number Three",
|
||
},
|
||
},
|
||
"0xpk4": {
|
||
Contact: &Contact{
|
||
ID: "0xpk4",
|
||
EnsName: "ens-user-4.eth",
|
||
ENSVerified: true,
|
||
DisplayName: "display-name-user-4",
|
||
LocalNickname: "primary-name-user-4",
|
||
},
|
||
},
|
||
"0xpk5": {
|
||
Contact: &Contact{
|
||
ID: "0xpk5",
|
||
LocalNickname: "User Number",
|
||
},
|
||
},
|
||
"0xpk6": {
|
||
Contact: &Contact{
|
||
ID: "0xpk6",
|
||
LocalNickname: "特别字符",
|
||
DisplayName: "特别 字符",
|
||
},
|
||
},
|
||
}
|
||
|
||
tests := []struct {
|
||
name string
|
||
text string
|
||
expected string
|
||
}{
|
||
{"empty string", "", ""},
|
||
{"no text", "", ""},
|
||
{"incomlepte mention 1", "@", "@"},
|
||
{"incomplete mention 2", "@r", "@r"},
|
||
{"no mentions", "foo bar @buzz kek @foo", "foo bar @buzz kek @foo"},
|
||
{"starts with mention", "@User Number One", "@0xpk1"},
|
||
{"starts with mention, comma after mention", "@User Number One,", "@0xpk1,"},
|
||
{"starts with mention but no space after", "@User NumberOnefoo", "@User NumberOnefoo"},
|
||
{"starts with mention, some text after mention", "@User Number One foo", "@0xpk1 foo"},
|
||
{"starts with some text, then mention", "text @User Number One", "text @0xpk1"},
|
||
{"starts with some text, then mention, then more text", "text @User Number One foo", "text @0xpk1 foo"},
|
||
{"no space before mention", "text@User Number One", "text@0xpk1"},
|
||
{"two different mentions", "@User Number One @User Number two", "@0xpk1 @0xpk2"},
|
||
{"two different mentions, separated with comma", "@User Number One,@User Number two", "@0xpk1,@0xpk2"},
|
||
{"two different mentions inside text", "foo@User Number One bar @User Number two baz", "foo@0xpk1 bar @0xpk2 baz"},
|
||
{"ens mention", "@user2", "@0xpk2"},
|
||
{"multiple mentions", strings.Repeat("@User Number One @User Number two ", 1000), strings.Repeat("@0xpk1 @0xpk2 ", 1000)},
|
||
|
||
{"single * case 1", "*@user2*", "*@user2*"},
|
||
{"single * case 2", "*@user2 *", "*@0xpk2 *"},
|
||
{"single * case 3", "a*@user2*", "a*@user2*"},
|
||
{"single * case 4", "*@user2 foo*foo", "*@0xpk2 foo*foo"},
|
||
{"single * case 5", "a *@user2*", "a *@user2*"},
|
||
{"single * case 6", "*@user2 foo*", "*@user2 foo*"},
|
||
{"single * case 7", "@user2 *@user2 foo* @user2", "@0xpk2 *@user2 foo* @0xpk2"},
|
||
{"single * case 8", "*@user2 foo**@user2 foo*", "*@user2 foo**@user2 foo*"},
|
||
{"single * case 9", "*@user2 foo***@user2 foo* @user2", "*@user2 foo***@user2 foo* @0xpk2"},
|
||
|
||
{"double * case 1", "**@user2**", "**@user2**"},
|
||
{"double * case 2", "**@user2 **", "**@0xpk2 **"},
|
||
{"double * case 3", "a**@user2**", "a**@user2**"},
|
||
{"double * case 4", "**@user2 foo**foo", "**@user2 foo**foo"},
|
||
{"double * case 5", "a **@user2**", "a **@user2**"},
|
||
{"double * case 6", "**@user2 foo**", "**@user2 foo**"},
|
||
{"double * case 7", "@user2 **@user2 foo** @user2", "@0xpk2 **@user2 foo** @0xpk2"},
|
||
{"double * case 8", "**@user2 foo****@user2 foo**", "**@user2 foo****@user2 foo**"},
|
||
{"double * case 9", "**@user2 foo*****@user2 foo** @user2", "**@user2 foo*****@user2 foo** @0xpk2"},
|
||
|
||
{"tripple * case 1", "***@user2 foo***@user2 foo*", "***@user2 foo***@0xpk2 foo*"},
|
||
{"tripple ~ case 1", "~~~@user2 foo~~~@user2 foo~", "~~~@user2 foo~~~@user2 foo~"},
|
||
|
||
{"quote case 1", ">@user2", ">@user2"},
|
||
{"quote case 2", "\n>@user2", "\n>@user2"},
|
||
{"quote case 3", "\n> @user2 \n \n @user2", "\n> @user2 \n \n @0xpk2"},
|
||
{"quote case 4", ">@user2\n\n>@user2", ">@user2\n\n>@user2"},
|
||
{"quote case 5", "***hey\n\n>@user2\n\n@user2 foo***", "***hey\n\n>@user2\n\n@0xpk2 foo***"},
|
||
|
||
{"code case 1", "` @user2 `", "` @user2 `"},
|
||
{"code case 2", "` @user2 `", "` @user2 `"},
|
||
{"code case 3", "``` @user2 ```", "``` @user2 ```"},
|
||
{"code case 4", "` ` @user2 ``", "` ` @0xpk2 ``"},
|
||
|
||
{"double @", "@ @user2", "@ @0xpk2"},
|
||
|
||
{"user name contains dash", "@display-name-user-4 ", "@0xpk4 "},
|
||
|
||
{"username or nickname of one is a substring of another case 1", "@User Number One @User Number", "@0xpk1 @0xpk5"},
|
||
{"username or nickname of one is a substring of another case 2", "@User Number @User Number One ", "@0xpk5 @0xpk1 "},
|
||
|
||
{"special chars in username case1", "@特别字符", "@0xpk6"},
|
||
{"special chars in username case2", "@特别字符 ", "@0xpk6 "},
|
||
{"special chars in username case3", " @特别 字符 ", " @0xpk6 "},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
got := ReplaceMentions(tt.text, users)
|
||
if got != tt.expected {
|
||
t.Errorf("testing %q, ReplaceMentions(%q) got %q, expected %q", tt.name, tt.text, got, tt.expected)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
func TestGetAtSignIdxs(t *testing.T) {
|
||
tests := []struct {
|
||
name string
|
||
text string
|
||
start int
|
||
want []int
|
||
}{
|
||
{
|
||
name: "no @ sign",
|
||
text: "hello world",
|
||
start: 0,
|
||
want: []int{},
|
||
},
|
||
{
|
||
name: "single @ sign",
|
||
text: "hello @world",
|
||
start: 0,
|
||
want: []int{6},
|
||
},
|
||
{
|
||
name: "multiple @ signs",
|
||
text: "@hello @world @again",
|
||
start: 0,
|
||
want: []int{0, 7, 14},
|
||
},
|
||
{
|
||
name: "start after first @ sign",
|
||
text: "hello @world",
|
||
start: 6,
|
||
want: []int{12},
|
||
},
|
||
{
|
||
name: "start after second @ sign",
|
||
text: "hello @world @again",
|
||
start: 8,
|
||
want: []int{14, 21},
|
||
},
|
||
{
|
||
name: "start after last @ sign",
|
||
text: "hello @world @again",
|
||
start: 15,
|
||
want: []int{21, 28},
|
||
},
|
||
}
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
got := getAtSignIdxs(tt.text, tt.start)
|
||
if !reflect.DeepEqual(got, tt.want) {
|
||
t.Errorf("getAtSignIdxs() = %v, want %v", got, tt.want)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
func TestToInfo(t *testing.T) {
|
||
newText := " "
|
||
t.Run("toInfo base case", func(t *testing.T) {
|
||
expected := &MentionState{
|
||
AtSignIdx: 2,
|
||
AtIdxs: []*AtIndexEntry{
|
||
{
|
||
Checked: true,
|
||
Mentioned: true,
|
||
From: 2,
|
||
To: 17,
|
||
},
|
||
},
|
||
MentionEnd: 19,
|
||
PreviousText: "",
|
||
NewText: newText,
|
||
Start: 18,
|
||
End: 18,
|
||
}
|
||
|
||
inputSegments := []InputSegment{
|
||
{Type: Text, Value: "H."},
|
||
{Type: Mention, Value: "@helpinghand.eth"},
|
||
{Type: Text, Value: " "},
|
||
}
|
||
|
||
actual := toInfo(inputSegments)
|
||
|
||
if !reflect.DeepEqual(expected.AtIdxs, actual.AtIdxs) {
|
||
t.Errorf("Expected AtIdxs: %#v, but got: %#v", expected.AtIdxs, actual.AtIdxs)
|
||
}
|
||
|
||
expected.AtIdxs = nil
|
||
actual.AtIdxs = nil
|
||
|
||
if !reflect.DeepEqual(expected, actual) {
|
||
t.Errorf("Expected %#v, but got %#v", expected, actual)
|
||
}
|
||
})
|
||
}
|
||
|
||
func TestToInputField(t *testing.T) {
|
||
testCases := []struct {
|
||
name string
|
||
input string
|
||
expected []InputSegment
|
||
}{
|
||
{
|
||
"only text",
|
||
"parse-text",
|
||
[]InputSegment{{Type: Text, Value: "parse-text"}},
|
||
},
|
||
{
|
||
"in the middle",
|
||
"hey @0x04fbce10971e1cd7253b98c7b7e54de3729ca57ce41a2bfb0d1c4e0a26f72c4b6913c3487fa1b4bb86125770f1743fb4459da05c1cbe31d938814cfaf36e252073 he",
|
||
[]InputSegment{
|
||
{Type: Text, Value: "hey "},
|
||
{Type: Mention, Value: "0x04fbce10971e1cd7253b98c7b7e54de3729ca57ce41a2bfb0d1c4e0a26f72c4b6913c3487fa1b4bb86125770f1743fb4459da05c1cbe31d938814cfaf36e252073"},
|
||
{Type: Text, Value: " he"},
|
||
},
|
||
},
|
||
{
|
||
"at the beginning",
|
||
"@0x04fbce10971e1cd7253b98c7b7e54de3729ca57ce41a2bfb0d1c4e0a26f72c4b6913c3487fa1b4bb86125770f1743fb4459da05c1cbe31d938814cfaf36e252073 he",
|
||
[]InputSegment{
|
||
{Type: Mention, Value: "0x04fbce10971e1cd7253b98c7b7e54de3729ca57ce41a2bfb0d1c4e0a26f72c4b6913c3487fa1b4bb86125770f1743fb4459da05c1cbe31d938814cfaf36e252073"},
|
||
{Type: Text, Value: " he"},
|
||
},
|
||
},
|
||
{
|
||
"at the end",
|
||
"hey @0x04fbce10971e1cd7253b98c7b7e54de3729ca57ce41a2bfb0d1c4e0a26f72c4b6913c3487fa1b4bb86125770f1743fb4459da05c1cbe31d938814cfaf36e252073",
|
||
[]InputSegment{
|
||
{Type: Text, Value: "hey "},
|
||
{Type: Mention, Value: "0x04fbce10971e1cd7253b98c7b7e54de3729ca57ce41a2bfb0d1c4e0a26f72c4b6913c3487fa1b4bb86125770f1743fb4459da05c1cbe31d938814cfaf36e252073"},
|
||
},
|
||
},
|
||
{
|
||
"invalid",
|
||
"invalid @0x04fBce10971e1cd7253b98c7b7e54de3729ca57ce41a2bfb0d1c4e0a26f72c4b6913c3487fa1b4bb86125770f1743fb4459da05c1cbe31d938814cfaf36e252073",
|
||
[]InputSegment{
|
||
{Type: Text, Value: "invalid @0x04fBce10971e1cd7253b98c7b7e54de3729ca57ce41a2bfb0d1c4e0a26f72c4b6913c3487fa1b4bb86125770f1743fb4459da05c1cbe31d938814cfaf36e252073"},
|
||
},
|
||
},
|
||
}
|
||
|
||
for _, tc := range testCases {
|
||
t.Run(tc.name, func(t *testing.T) {
|
||
result := toInputField(tc.input)
|
||
if !reflect.DeepEqual(result, tc.expected) {
|
||
t.Errorf("Expected: %v, got: %v", tc.expected, result)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
func TestSubs(t *testing.T) {
|
||
testCases := []struct {
|
||
name string
|
||
input string
|
||
start int
|
||
end int
|
||
expected string
|
||
}{
|
||
{
|
||
name: "Normal case",
|
||
input: "Hello, world!",
|
||
start: 0,
|
||
end: 5,
|
||
expected: "Hello",
|
||
},
|
||
{
|
||
name: "Start index out of range (negative)",
|
||
input: "Hello, world!",
|
||
start: -5,
|
||
end: 5,
|
||
expected: "Hello",
|
||
},
|
||
{
|
||
name: "End index out of range",
|
||
input: "Hello, world!",
|
||
start: 7,
|
||
end: 50,
|
||
expected: "world!",
|
||
},
|
||
{
|
||
name: "Start index greater than end index",
|
||
input: "Hello, world!",
|
||
start: 10,
|
||
end: 5,
|
||
expected: ", wor",
|
||
},
|
||
{
|
||
name: "Both indices out of range",
|
||
input: "Hello, world!",
|
||
start: -5,
|
||
end: 50,
|
||
expected: "Hello, world!",
|
||
},
|
||
{
|
||
name: "Start index negative, end index out of range",
|
||
input: "Hello, world!",
|
||
start: -10,
|
||
end: 15,
|
||
expected: "Hello, world!",
|
||
},
|
||
{
|
||
name: "Start index negative, end index within range",
|
||
input: "Hello, world!",
|
||
start: -10,
|
||
end: 5,
|
||
expected: "Hello",
|
||
},
|
||
{
|
||
name: "Start index negative, end index negative",
|
||
input: "Hello, world!",
|
||
start: -10,
|
||
end: -5,
|
||
expected: "",
|
||
},
|
||
|
||
{
|
||
name: "Start index zero, end index zero",
|
||
input: "Hello, world!",
|
||
start: 0,
|
||
end: 0,
|
||
expected: "",
|
||
},
|
||
{
|
||
name: "Start index positive, end index zero",
|
||
input: "Hello, world!",
|
||
start: 3,
|
||
end: 0,
|
||
expected: "Hel",
|
||
},
|
||
{
|
||
name: "Start index equal to input length",
|
||
input: "Hello, world!",
|
||
start: 13,
|
||
end: 15,
|
||
expected: "",
|
||
},
|
||
{
|
||
name: "End index negative",
|
||
input: "Hello, world!",
|
||
start: 5,
|
||
end: -5,
|
||
expected: "Hello",
|
||
},
|
||
{
|
||
name: "Start and end indices equal and negative",
|
||
input: "Hello, world!",
|
||
start: -3,
|
||
end: -3,
|
||
expected: "",
|
||
},
|
||
{
|
||
name: "Start index greater than input length",
|
||
input: "Hello, world!",
|
||
start: 15,
|
||
end: 20,
|
||
expected: "",
|
||
},
|
||
{
|
||
name: "End index equal to input length",
|
||
input: "Hello, world!",
|
||
start: 0,
|
||
end: 13,
|
||
expected: "Hello, world!",
|
||
},
|
||
}
|
||
|
||
for _, tc := range testCases {
|
||
t.Run(tc.name, func(t *testing.T) {
|
||
actual := subs(tc.input, tc.start, tc.end)
|
||
if actual != tc.expected {
|
||
t.Errorf("Test case '%s': expected '%s', got '%s'", tc.name, tc.expected, actual)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
func TestLastIndexOf(t *testing.T) {
|
||
atSignIdx := lastIndexOfAtSign("@", 0)
|
||
require.Equal(t, 0, atSignIdx)
|
||
|
||
atSignIdx = lastIndexOfAtSign("@@", 1)
|
||
require.Equal(t, 1, atSignIdx)
|
||
|
||
//at-sign-idx 0 text @t searched-text t start 2 end 2 new-text
|
||
atSignIdx = lastIndexOfAtSign("@t", 2)
|
||
require.Equal(t, 0, atSignIdx)
|
||
|
||
atSignIdx = lastIndexOfAtSign("at", 3)
|
||
require.Equal(t, -1, atSignIdx)
|
||
}
|
||
|
||
func TestDiffText(t *testing.T) {
|
||
testCases := []struct {
|
||
oldText string
|
||
newText string
|
||
expected *TextDiff
|
||
}{
|
||
{
|
||
oldText: "",
|
||
newText: "A",
|
||
expected: &TextDiff{
|
||
start: 0,
|
||
end: 0,
|
||
previousText: "",
|
||
newText: "A",
|
||
operation: textOperationAdd,
|
||
},
|
||
},
|
||
{
|
||
oldText: "A",
|
||
newText: "Ab",
|
||
expected: &TextDiff{
|
||
start: 1,
|
||
end: 1,
|
||
previousText: "A",
|
||
newText: "b",
|
||
operation: textOperationAdd,
|
||
},
|
||
},
|
||
{
|
||
oldText: "Ab",
|
||
newText: "Abc",
|
||
expected: &TextDiff{
|
||
start: 2,
|
||
end: 2,
|
||
previousText: "Ab",
|
||
newText: "c",
|
||
operation: textOperationAdd,
|
||
},
|
||
},
|
||
{
|
||
oldText: "Ab",
|
||
newText: "cAb",
|
||
expected: &TextDiff{
|
||
start: 0,
|
||
end: 0,
|
||
previousText: "Ab",
|
||
newText: "c",
|
||
operation: textOperationAdd,
|
||
},
|
||
},
|
||
{
|
||
oldText: "Ac",
|
||
newText: "Adc",
|
||
expected: &TextDiff{
|
||
start: 1,
|
||
end: 1,
|
||
previousText: "Ac",
|
||
newText: "d",
|
||
operation: textOperationAdd,
|
||
},
|
||
},
|
||
{
|
||
oldText: "Adc",
|
||
newText: "Ad ee c",
|
||
expected: &TextDiff{
|
||
start: 2,
|
||
end: 2,
|
||
previousText: "Adc",
|
||
newText: " ee ",
|
||
operation: textOperationAdd,
|
||
},
|
||
},
|
||
{
|
||
oldText: "Ad ee c",
|
||
newText: "A fff d ee c",
|
||
expected: &TextDiff{
|
||
start: 1,
|
||
end: 1,
|
||
previousText: "Ad ee c",
|
||
newText: " fff ",
|
||
operation: textOperationAdd,
|
||
},
|
||
},
|
||
{
|
||
oldText: "Abc",
|
||
newText: "Ac",
|
||
expected: &TextDiff{
|
||
start: 1,
|
||
end: 1,
|
||
previousText: "Abc",
|
||
newText: "",
|
||
operation: textOperationDelete,
|
||
},
|
||
},
|
||
{
|
||
oldText: "Abcd",
|
||
newText: "Ab",
|
||
expected: &TextDiff{
|
||
start: 2,
|
||
end: 3,
|
||
previousText: "Abcd",
|
||
newText: "",
|
||
operation: textOperationDelete,
|
||
},
|
||
},
|
||
{
|
||
oldText: "Abcd",
|
||
newText: "bcd",
|
||
expected: &TextDiff{
|
||
start: 0,
|
||
end: 0,
|
||
previousText: "Abcd",
|
||
newText: "",
|
||
operation: textOperationDelete,
|
||
},
|
||
},
|
||
{
|
||
oldText: "Abcd你好",
|
||
newText: "Abcd你",
|
||
expected: &TextDiff{
|
||
start: 5,
|
||
end: 5,
|
||
previousText: "Abcd你好",
|
||
newText: "",
|
||
operation: textOperationDelete,
|
||
},
|
||
},
|
||
{
|
||
oldText: "Abcd你好",
|
||
newText: "Abcd",
|
||
expected: &TextDiff{
|
||
start: 4,
|
||
end: 5,
|
||
previousText: "Abcd你好",
|
||
newText: "",
|
||
operation: textOperationDelete,
|
||
},
|
||
},
|
||
{
|
||
oldText: "A fff d ee c",
|
||
newText: " fff d ee c",
|
||
expected: &TextDiff{
|
||
start: 0,
|
||
end: 0,
|
||
previousText: "A fff d ee c",
|
||
newText: "",
|
||
operation: textOperationDelete,
|
||
},
|
||
},
|
||
{
|
||
oldText: " fff d ee c",
|
||
newText: " fffee c",
|
||
expected: &TextDiff{
|
||
start: 4,
|
||
end: 6,
|
||
previousText: " fff d ee c",
|
||
newText: "",
|
||
operation: textOperationDelete,
|
||
},
|
||
},
|
||
{
|
||
oldText: "abc",
|
||
newText: "abc",
|
||
expected: nil,
|
||
},
|
||
{
|
||
oldText: "abc",
|
||
newText: "ghij",
|
||
expected: &TextDiff{
|
||
start: 0,
|
||
end: 2,
|
||
previousText: "abc",
|
||
newText: "ghij",
|
||
operation: textOperationReplace,
|
||
},
|
||
},
|
||
{
|
||
oldText: "abc",
|
||
newText: "babcd",
|
||
expected: &TextDiff{
|
||
start: 0,
|
||
end: 2,
|
||
previousText: "abc",
|
||
newText: "babcd",
|
||
operation: textOperationReplace,
|
||
},
|
||
},
|
||
{
|
||
oldText: "abc",
|
||
newText: "baebcd",
|
||
expected: &TextDiff{
|
||
start: 0,
|
||
end: 2,
|
||
previousText: "abc",
|
||
newText: "baebcd",
|
||
operation: textOperationReplace,
|
||
},
|
||
},
|
||
{
|
||
oldText: "abc",
|
||
newText: "aefc",
|
||
expected: &TextDiff{
|
||
start: 1,
|
||
end: 1,
|
||
previousText: "abc",
|
||
newText: "ef",
|
||
operation: textOperationReplace,
|
||
},
|
||
},
|
||
{
|
||
oldText: "abc",
|
||
newText: "adc",
|
||
expected: &TextDiff{
|
||
start: 1,
|
||
end: 1,
|
||
previousText: "abc",
|
||
newText: "d",
|
||
operation: textOperationReplace,
|
||
},
|
||
},
|
||
{
|
||
oldText: "abc",
|
||
newText: "abd",
|
||
expected: &TextDiff{
|
||
start: 2,
|
||
end: 2,
|
||
previousText: "abc",
|
||
newText: "d",
|
||
operation: textOperationReplace,
|
||
},
|
||
},
|
||
{
|
||
oldText: "abc",
|
||
newText: "cbc",
|
||
expected: &TextDiff{
|
||
start: 0,
|
||
end: 0,
|
||
previousText: "abc",
|
||
newText: "c",
|
||
operation: textOperationReplace,
|
||
},
|
||
},
|
||
{
|
||
oldText: "abc",
|
||
newText: "ffbc",
|
||
expected: &TextDiff{
|
||
start: 0,
|
||
end: 0,
|
||
previousText: "abc",
|
||
newText: "ff",
|
||
operation: textOperationReplace,
|
||
},
|
||
},
|
||
}
|
||
for i, tc := range testCases {
|
||
t.Run(fmt.Sprintf("%d", i+1), func(t *testing.T) {
|
||
diff := diffText(tc.oldText, tc.newText)
|
||
require.Equal(t, tc.expected, diff)
|
||
})
|
||
}
|
||
}
|
||
|
||
type MockMentionableUserGetter struct {
|
||
mentionableUserMap map[string]*MentionableUser
|
||
}
|
||
|
||
func (m *MockMentionableUserGetter) getMentionableUsers(chatID string) (map[string]*MentionableUser, error) {
|
||
return m.mentionableUserMap, nil
|
||
}
|
||
|
||
func (m *MockMentionableUserGetter) getMentionableUser(chatID string, pk string) (*MentionableUser, error) {
|
||
return m.mentionableUserMap[pk], nil
|
||
}
|
||
|
||
func TestMentionSuggestionCases(t *testing.T) {
|
||
mentionableUserMap, chatID, mentionManager := setupMentionSuggestionTest(t, nil)
|
||
|
||
testCases := []struct {
|
||
inputText string
|
||
expectedSize int
|
||
}{
|
||
{"@", len(mentionableUserMap)},
|
||
{"@u", len(mentionableUserMap)},
|
||
{"@u2", 1},
|
||
{"@u23", 0},
|
||
{"@u2", 1},
|
||
{"@u2 abc", 0},
|
||
{"@u2 abc @u3", 1},
|
||
{"@u2 abc@u3", 0},
|
||
{"@u2 abc@u3 ", 0},
|
||
{"@u2 abc @u3", 1},
|
||
}
|
||
|
||
for i, tc := range testCases {
|
||
t.Run(fmt.Sprintf("%d", i+1), func(t *testing.T) {
|
||
ctx, err := mentionManager.OnChangeText(chatID, tc.inputText, uint64(i+1))
|
||
require.NoError(t, err)
|
||
t.Logf("Input: %+v, MentionState:%+v, InputSegments:%+v\n", tc.inputText, ctx.MentionState, ctx.InputSegments)
|
||
require.Equal(t, tc.expectedSize, len(ctx.MentionSuggestions))
|
||
})
|
||
}
|
||
}
|
||
|
||
func TestMentionSuggestionAfterToInputField(t *testing.T) {
|
||
mentionableUserMap, chatID, mentionManager := setupMentionSuggestionTest(t, nil)
|
||
_, err := mentionManager.ToInputField(chatID, "abc")
|
||
require.NoError(t, err)
|
||
ctx, err := mentionManager.OnChangeText(chatID, "@", 1)
|
||
require.NoError(t, err)
|
||
require.Equal(t, len(mentionableUserMap), len(ctx.MentionSuggestions))
|
||
}
|
||
|
||
func TestMentionSuggestionSpecialInputModeForAndroid(t *testing.T) {
|
||
mentionableUserMap, chatID, mentionManager := setupMentionSuggestionTest(t, nil)
|
||
|
||
testCases := []struct {
|
||
inputText string
|
||
expectedSize int
|
||
}{
|
||
{"A", 0},
|
||
{"As", 0},
|
||
{"Asd", 0},
|
||
{"Asd@", len(mentionableUserMap)},
|
||
}
|
||
|
||
for i, tc := range testCases {
|
||
t.Run(fmt.Sprintf("%d", i+1), func(t *testing.T) {
|
||
ctx, err := mentionManager.OnChangeText(chatID, tc.inputText, uint64(i+1))
|
||
require.NoError(t, err)
|
||
require.Equal(t, tc.expectedSize, len(ctx.MentionSuggestions))
|
||
t.Logf("Input: %+v, MentionState:%+v, InputSegments:%+v\n", tc.inputText, ctx.MentionState, ctx.InputSegments)
|
||
})
|
||
}
|
||
}
|
||
|
||
func TestMentionSuggestionSpecialChars(t *testing.T) {
|
||
mentionableUserMap, chatID, mentionManager := setupMentionSuggestionTest(t, nil)
|
||
|
||
testCases := []struct {
|
||
inputText string
|
||
expectedSize int
|
||
}{
|
||
{"'", 0},
|
||
{"‘", 0},
|
||
{"‘@", len(mentionableUserMap)},
|
||
{"‘@自由人", 1},
|
||
}
|
||
|
||
for i, tc := range testCases {
|
||
t.Run(fmt.Sprintf("%d", i+1), func(t *testing.T) {
|
||
ctx, err := mentionManager.OnChangeText(chatID, tc.inputText, uint64(i+1))
|
||
require.NoError(t, err)
|
||
t.Logf("Input: %+v, MentionState:%+v, InputSegments:%+v\n", tc.inputText, ctx.MentionState, ctx.InputSegments)
|
||
require.Equal(t, tc.expectedSize, len(ctx.MentionSuggestions))
|
||
})
|
||
}
|
||
}
|
||
|
||
func TestMentionSuggestionAtSignSpaceCases(t *testing.T) {
|
||
mentionableUserMap, chatID, mentionManager := setupMentionSuggestionTest(t, map[string]*MentionableUser{
|
||
"0xpk1": {
|
||
Contact: &Contact{
|
||
ID: "0xpk1",
|
||
LocalNickname: "User Number One",
|
||
},
|
||
},
|
||
})
|
||
|
||
testCases := []struct {
|
||
inputText string
|
||
expectedSize int
|
||
}{
|
||
{"@", len(mentionableUserMap)},
|
||
{"@ ", 0},
|
||
{"@ @", len(mentionableUserMap)},
|
||
}
|
||
|
||
var ctx *ChatMentionContext
|
||
var err error
|
||
for i, tc := range testCases {
|
||
ctx, err = mentionManager.OnChangeText(chatID, tc.inputText, uint64(i+1))
|
||
require.NoError(t, err)
|
||
t.Logf("After OnChangeText, Input: %+v, MentionState:%+v, InputSegments:%+v\n", tc.inputText, ctx.MentionState, ctx.InputSegments)
|
||
require.Equal(t, tc.expectedSize, len(ctx.MentionSuggestions))
|
||
}
|
||
require.Len(t, ctx.InputSegments, 2)
|
||
require.Equal(t, Text, ctx.InputSegments[0].Type)
|
||
require.Equal(t, "@ ", ctx.InputSegments[0].Value)
|
||
require.Equal(t, Text, ctx.InputSegments[1].Type)
|
||
require.Equal(t, "@", ctx.InputSegments[1].Value)
|
||
}
|
||
|
||
func TestSelectMention(t *testing.T) {
|
||
mentionableUsers, chatID, mentionManager := setupMentionSuggestionTest(t, nil)
|
||
|
||
var callID uint64 = 1
|
||
text := "@u2 abc"
|
||
ctx, err := mentionManager.OnChangeText(chatID, text, callID)
|
||
require.NoError(t, err)
|
||
require.Equal(t, 0, len(ctx.MentionSuggestions))
|
||
|
||
callID++
|
||
ctx, err = mentionManager.OnChangeText(chatID, "@u abc", callID)
|
||
require.NoError(t, err)
|
||
require.Equal(t, len(mentionableUsers), len(ctx.MentionSuggestions))
|
||
|
||
ctx, err = mentionManager.SelectMention(chatID, "@u abc", "u2", "0xpk2")
|
||
require.NoError(t, err)
|
||
require.Equal(t, 0, len(ctx.MentionSuggestions))
|
||
require.Equal(t, text, ctx.NewText)
|
||
require.Equal(t, text, ctx.PreviousText)
|
||
|
||
callID++
|
||
ctx, err = mentionManager.OnChangeText(chatID, text, callID)
|
||
require.NoError(t, err)
|
||
require.Equal(t, 0, len(ctx.MentionSuggestions))
|
||
}
|
||
|
||
func TestInputSegments(t *testing.T) {
|
||
_, chatID, mentionManager := setupMentionSuggestionTest(t, nil)
|
||
var callID uint64 = 1
|
||
ctx, err := mentionManager.OnChangeText(chatID, "@u1", callID)
|
||
require.NoError(t, err)
|
||
require.Equal(t, 1, len(ctx.InputSegments))
|
||
require.Equal(t, Text, ctx.InputSegments[0].Type)
|
||
require.Equal(t, "@u1", ctx.InputSegments[0].Value)
|
||
|
||
callID++
|
||
ctx, err = mentionManager.OnChangeText(chatID, "@u1 @User Number One", callID)
|
||
require.NoError(t, err)
|
||
require.Equal(t, 2, len(ctx.InputSegments))
|
||
require.Equal(t, Text, ctx.InputSegments[0].Type)
|
||
require.Equal(t, "@u1 ", ctx.InputSegments[0].Value)
|
||
require.Equal(t, Mention, ctx.InputSegments[1].Type)
|
||
require.Equal(t, "@User Number One", ctx.InputSegments[1].Value)
|
||
|
||
callID++
|
||
ctx, err = mentionManager.OnChangeText(chatID, "@u1 @User Number O", callID)
|
||
require.NoError(t, err)
|
||
require.Equal(t, 2, len(ctx.InputSegments))
|
||
require.Equal(t, Text, ctx.InputSegments[1].Type)
|
||
require.Equal(t, "@User Number O", ctx.InputSegments[1].Value)
|
||
|
||
callID++
|
||
ctx, err = mentionManager.OnChangeText(chatID, "@u2 @User Number One", callID)
|
||
require.NoError(t, err)
|
||
require.Equal(t, 3, len(ctx.InputSegments))
|
||
require.Equal(t, Mention, ctx.InputSegments[0].Type)
|
||
require.Equal(t, "@u2", ctx.InputSegments[0].Value)
|
||
require.Equal(t, Text, ctx.InputSegments[1].Type)
|
||
require.Equal(t, " ", ctx.InputSegments[1].Value)
|
||
require.Equal(t, Mention, ctx.InputSegments[2].Type)
|
||
require.Equal(t, "@User Number One", ctx.InputSegments[2].Value)
|
||
|
||
callID++
|
||
ctx, err = mentionManager.OnChangeText(chatID, "@u2 @User Number One a ", callID)
|
||
require.NoError(t, err)
|
||
require.Equal(t, 4, len(ctx.InputSegments))
|
||
require.Equal(t, Mention, ctx.InputSegments[2].Type)
|
||
require.Equal(t, "@User Number One", ctx.InputSegments[2].Value)
|
||
require.Equal(t, Text, ctx.InputSegments[3].Type)
|
||
require.Equal(t, " a ", ctx.InputSegments[3].Value)
|
||
|
||
callID++
|
||
ctx, err = mentionManager.OnChangeText(chatID, "@u2 @User Numbed One a ", callID)
|
||
require.NoError(t, err)
|
||
require.Equal(t, 3, len(ctx.InputSegments))
|
||
require.Equal(t, Mention, ctx.InputSegments[0].Type)
|
||
require.Equal(t, "@u2", ctx.InputSegments[0].Value)
|
||
require.Equal(t, Text, ctx.InputSegments[2].Type)
|
||
require.Equal(t, "@User Numbed One a ", ctx.InputSegments[2].Value)
|
||
|
||
callID++
|
||
ctx, err = mentionManager.OnChangeText(chatID, "@ @ ", callID)
|
||
require.NoError(t, err)
|
||
require.Equal(t, 2, len(ctx.InputSegments))
|
||
require.Equal(t, Text, ctx.InputSegments[0].Type)
|
||
require.Equal(t, "@ ", ctx.InputSegments[0].Value)
|
||
require.Equal(t, Text, ctx.InputSegments[1].Type)
|
||
require.Equal(t, "@ ", ctx.InputSegments[1].Value)
|
||
|
||
callID++
|
||
ctx, err = mentionManager.OnChangeText(chatID, "@u3 @ ", callID)
|
||
require.NoError(t, err)
|
||
require.Equal(t, 3, len(ctx.InputSegments))
|
||
require.Equal(t, Mention, ctx.InputSegments[0].Type)
|
||
require.Equal(t, "@u3", ctx.InputSegments[0].Value)
|
||
require.Equal(t, Text, ctx.InputSegments[1].Type)
|
||
require.Equal(t, " ", ctx.InputSegments[1].Value)
|
||
require.Equal(t, Text, ctx.InputSegments[2].Type)
|
||
require.Equal(t, "@ ", ctx.InputSegments[2].Value)
|
||
|
||
callID++
|
||
_, err = mentionManager.OnChangeText(chatID, " @ @User Number Three ", callID)
|
||
require.NoError(t, err)
|
||
callID++
|
||
_, err = mentionManager.OnChangeText(chatID, "@U @ @User Number Three ", callID)
|
||
require.NoError(t, err)
|
||
ctx, err = mentionManager.SelectMention(chatID, "@U @ @User Number Three ", "User Number Three", "0xpk3")
|
||
require.NoError(t, err)
|
||
require.Equal(t, 2, mentionTypeNum(ctx.InputSegments))
|
||
|
||
callID++
|
||
ctx, _ = mentionManager.OnChangeText(chatID, "@User Number Threea", callID)
|
||
require.Equal(t, 0, mentionTypeNum(ctx.InputSegments))
|
||
|
||
callID++
|
||
ctx, _ = mentionManager.OnChangeText(chatID, "@User Number Threea\n@u2\nabc@u3 asa", callID)
|
||
require.Equal(t, 2, mentionTypeNum(ctx.InputSegments))
|
||
callID++
|
||
ctx, _ = mentionManager.OnChangeText(chatID, "@User Number Thre\n@u2\nabc@u3 asa", callID)
|
||
require.Equal(t, 2, mentionTypeNum(ctx.InputSegments))
|
||
require.Equal(t, "@u2", ctx.InputSegments[1].Value)
|
||
require.Equal(t, "@u3", ctx.InputSegments[3].Value)
|
||
}
|
||
|
||
func mentionTypeNum(inputSegments []InputSegment) int {
|
||
var num int
|
||
for _, s := range inputSegments {
|
||
if s.Type == Mention {
|
||
num++
|
||
}
|
||
}
|
||
return num
|
||
}
|
||
|
||
func setupMentionSuggestionTest(t *testing.T, mentionableUserMapInput map[string]*MentionableUser) (map[string]*MentionableUser, string, *MentionManager) {
|
||
mentionableUserMap := mentionableUserMapInput
|
||
if mentionableUserMap == nil {
|
||
mentionableUserMap = getDefaultMentionableUserMap()
|
||
}
|
||
|
||
for _, u := range mentionableUserMap {
|
||
addSearchablePhrases(u)
|
||
}
|
||
|
||
mockMentionableUserGetter := &MockMentionableUserGetter{
|
||
mentionableUserMap: mentionableUserMap,
|
||
}
|
||
|
||
chatID := "0xchatID"
|
||
allChats := new(chatMap)
|
||
allChats.Store(chatID, &Chat{})
|
||
|
||
key, err := crypto.GenerateKey()
|
||
require.NoError(t, err)
|
||
mentionManager := &MentionManager{
|
||
mentionableUserGetter: mockMentionableUserGetter,
|
||
mentionContexts: make(map[string]*ChatMentionContext),
|
||
Messenger: &Messenger{
|
||
allChats: allChats,
|
||
identity: key,
|
||
},
|
||
logger: logutils.ZapLogger().Named("MentionManager"),
|
||
}
|
||
|
||
return mentionableUserMap, chatID, mentionManager
|
||
}
|
||
|
||
func getDefaultMentionableUserMap() map[string]*MentionableUser {
|
||
return map[string]*MentionableUser{
|
||
"0xpk1": {
|
||
Contact: &Contact{
|
||
ID: "0xpk1",
|
||
LocalNickname: "User Number One",
|
||
},
|
||
},
|
||
"0xpk2": {
|
||
Contact: &Contact{
|
||
ID: "0xpk2",
|
||
LocalNickname: "u2",
|
||
ENSVerified: true,
|
||
EnsName: "User Number Two",
|
||
},
|
||
},
|
||
"0xpk3": {
|
||
Contact: &Contact{
|
||
ID: "0xpk3",
|
||
LocalNickname: "u3",
|
||
ENSVerified: true,
|
||
EnsName: "User Number Three",
|
||
},
|
||
},
|
||
"0xpk4": {
|
||
Contact: &Contact{
|
||
ID: "0xpk4",
|
||
LocalNickname: "自由人",
|
||
ENSVerified: true,
|
||
EnsName: "User Number Four",
|
||
},
|
||
},
|
||
}
|
||
}
|