feat: decide_chat_assignment: Log correct post_msg_exists value

rfc724_mid_exists() returns the message id not only when the message is fully downloaded, so we need
to check this additionally. Also improve test_receive_pre_message to check that receiving the
pre-message again doesn't affect the existing message.
This commit is contained in:
iequidoo
2026-04-15 15:08:43 -03:00
committed by iequidoo
parent 89f948028d
commit 8e91eecb3d
2 changed files with 13 additions and 4 deletions

View File

@@ -1313,8 +1313,9 @@ async fn decide_chat_assignment(
..
} = &mime_parser.pre_message
{
let msg_id = rfc724_mid_exists(context, post_msg_rfc724_mid).await?;
if let Some(msg_id) = msg_id {
let post_msg_exists = if let Some((msg_id, not_downloaded)) =
message::rfc724_mid_exists_ex(context, post_msg_rfc724_mid, "download_state<>0").await?
{
context
.sql
.execute(
@@ -1322,8 +1323,10 @@ async fn decide_chat_assignment(
(rfc724_mid, msg_id),
)
.await?;
}
let post_msg_exists = msg_id.is_some();
!not_downloaded
} else {
false
};
info!(
context,
"Message {rfc724_mid} is a pre-message for {post_msg_rfc724_mid} (post_msg_exists:{post_msg_exists})."

View File

@@ -118,7 +118,13 @@ async fn test_receive_pre_message() -> Result<()> {
let msg = bob.recv_msg(&pre_message).await;
assert_eq!(msg.download_state, DownloadState::Available);
assert_summary_texts(&msg, bob, "👤 test").await;
let chat_id = msg.chat_id;
assert!(bob.recv_msg_opt(&pre_message).await.is_none());
let msg = Message::load_from_db(bob, msg.id).await?;
assert_eq!(msg.download_state, DownloadState::Available);
assert_summary_texts(&msg, bob, "👤 test").await;
assert_eq!(msg.chat_id, chat_id);
Ok(())
}