status-go/protocol/messenger_mention_test.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)
			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, "@")
	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)
			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)
			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 _, tc := range testCases {
		ctx, err = mentionManager.OnChangeText(chatID, tc.inputText)
		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)

	text := "@u2 abc"
	ctx, err := mentionManager.OnChangeText(chatID, text)
	require.NoError(t, err)
	require.Equal(t, 0, len(ctx.MentionSuggestions))

	ctx, err = mentionManager.OnChangeText(chatID, "@u abc")
	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)

	ctx, err = mentionManager.OnChangeText(chatID, text)
	require.NoError(t, err)
	require.Equal(t, 0, len(ctx.MentionSuggestions))
}

func TestInputSegments(t *testing.T) {
	_, chatID, mentionManager := setupMentionSuggestionTest(t, nil)
	ctx, err := mentionManager.OnChangeText(chatID, "@u1")
	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)

	ctx, err = mentionManager.OnChangeText(chatID, "@u1 @User Number One")
	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)

	ctx, err = mentionManager.OnChangeText(chatID, "@u1 @User Number O")
	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)

	ctx, err = mentionManager.OnChangeText(chatID, "@u2 @User Number One")
	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)

	ctx, err = mentionManager.OnChangeText(chatID, "@u2 @User Number One a ")
	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)

	ctx, err = mentionManager.OnChangeText(chatID, "@u2 @User Numbed One a ")
	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)

	ctx, err = mentionManager.OnChangeText(chatID, "@ @ ")
	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)

	ctx, err = mentionManager.OnChangeText(chatID, "@u3 @ ")
	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)

	_, err = mentionManager.OnChangeText(chatID, " @ @User Number Three ")
	require.NoError(t, err)
	_, err = mentionManager.OnChangeText(chatID, "@U @ @User Number Three ")
	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))

	ctx, _ = mentionManager.OnChangeText(chatID, "@User Number Threea")
	require.Equal(t, 0, mentionTypeNum(ctx.InputSegments))

	ctx, _ = mentionManager.OnChangeText(chatID, "@User Number Threea\n@u2\nabc@u3 asa")
	require.Equal(t, 2, mentionTypeNum(ctx.InputSegments))
	ctx, _ = mentionManager.OnChangeText(chatID, "@User Number Thre\n@u2\nabc@u3 asa")
	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",
			},
		},
	}
}