refactor: retire legacy tinode database routing

2026-09-23 17:04:30 +08:00
parent 204043382c
commit aa9f562e1e
19 changed files with 294 additions and 744 deletions
@@ -7549,6 +7549,37 @@ func TestServeIMFriendRequestCreate(t *testing.T) {
 	}
 }

+func TestServeIMFriendRequestCreateAllowsProfileCardSource(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+	mockStore := expectIMBearer(t, ctrl, "token-1", 7, 1)
+	mockStore.EXPECT().GetUidString().Return("card-1")
+	oldRelations := store.Relations
+	oldProfiles := store.Profiles
+	relations := &testRelationsStore{context: &types.RelationshipContext{Relationship: types.RelationshipNone}}
+	store.Relations = relations
+	store.Profiles = &testProfilesStore{}
+	defer func() {
+		store.Relations = oldRelations
+		store.Profiles = oldProfiles
+	}()
+
+	req := httptest.NewRequest(http.MethodPost, "/v1/im/friend-requests",
+		bytes.NewBufferString(`{"target_user_id":"`+types.Uid(2).UserId()+`","message":"你好","source":"profile_card"}`))
+	req.Header.Set("Authorization", "Bearer token-1")
+	req.Header.Set("Idempotency-Key", "fr-card-1")
+	rec := httptest.NewRecorder()
+
+	serveIMFriendRequests(rec, req)
+
+	if rec.Code != http.StatusOK {
+		t.Fatalf("status: got %d, want %d; body=%s", rec.Code, http.StatusOK, rec.Body.String())
+	}
+	if relations.created == nil || relations.created.Source != "profile_card" || relations.created.SourceTopic != "" {
+		t.Fatalf("unexpected friend request created from profile card: %#v", relations.created)
+	}
+}
+
 func TestServeIMFriendRequestCreateReturnsNotFoundForMissingTarget(t *testing.T) {
 	ctrl := gomock.NewController(t)
 	defer ctrl.Finish()
@@ -8105,6 +8136,70 @@ func TestServeIMUserProfile(t *testing.T) {
 	}
 }

+func TestServeIMUserProfilePrivateNonFriendReturnsLimitedProfile(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+	expectIMBearer(t, ctrl, "token-1", 7, 1)
+	oldUsers := store.Users
+	oldRelations := store.Relations
+	oldProfiles := store.Profiles
+	users := mock_store.NewMockUsersPersistenceInterface(ctrl)
+	peer := types.Uid(2)
+	users.EXPECT().Get(types.TenantID(7), peer).Return(&types.User{
+		ObjHeader: types.ObjHeader{Id: peer.String()},
+		TenantID:  7,
+		Username:  "alice7",
+		Public: map[string]any{
+			"display_name":   "Alice",
+			"nickname":       "Alice",
+			"avatar_file_id": "file-1",
+			"signature":      "hello",
+			"gender":         "female",
+			"region":         "Shanghai",
+			"public_account": "alice001",
+		},
+	}, nil)
+	store.Users = users
+	store.Relations = &testRelationsStore{context: &types.RelationshipContext{
+		TenantID:     7,
+		UserID:       1,
+		PeerID:       peer,
+		Topic:        types.Uid(1).P2PName(peer),
+		Relationship: types.RelationshipNone,
+		CanSend:      false,
+	}}
+	store.Profiles = &testProfilesStore{}
+	defer func() {
+		store.Users = oldUsers
+		store.Relations = oldRelations
+		store.Profiles = oldProfiles
+	}()
+
+	req := httptest.NewRequest(http.MethodGet, "/v1/im/users/"+peer.UserId()+"/profile", nil)
+	req.Header.Set("Authorization", "Bearer token-1")
+	rec := httptest.NewRecorder()
+
+	serveIMUserResource(rec, req)
+
+	if rec.Code != http.StatusOK {
+		t.Fatalf("status: got %d, want %d; body=%s", rec.Code, http.StatusOK, rec.Body.String())
+	}
+	data := decodeIMHTTPResponse(t, rec).Data.(map[string]any)
+	if data["relationship"] != types.RelationshipNone || data["display_name"] != "Alice" ||
+		data["avatar_file_id"] != "file-1" || data["public_account"] != "alice001" || data["username"] != "alice7" {
+		t.Fatalf("unexpected limited profile response: %#v", data)
+	}
+	capabilities := data["capabilities"].(map[string]any)
+	if capabilities["can_send"] != false || capabilities["can_call"] != false || capabilities["can_add_friend"] != true {
+		t.Fatalf("unexpected limited profile capabilities: %#v", data)
+	}
+	for _, field := range []string{"signature", "gender", "region"} {
+		if _, ok := data[field]; ok {
+			t.Fatalf("limited profile should omit private field %q: %#v", field, data)
+		}
+	}
+}
+
 func TestServeIMUserProfileBlockedByMeReturnsMinimalProfile(t *testing.T) {
 	ctrl := gomock.NewController(t)
 	defer ctrl.Finish()