Add message ID to DB entries

This should prevent duplicate indexes.

src/main/db/client.py

@@ -48,7 +48,9 @@ def _create_indexes() -> None:
         keys=[("chat_id", ASCENDING), ("feed_name", ASCENDING), ("feed_type", ASCENDING)],
         unique=True,
     )
-    pinned_index = _pinned_collection.create_index(keys=[("chat_id", ASCENDING)], unique=True)
+    pinned_index = _pinned_collection.create_index(
+        keys=[("chat_id", ASCENDING), ("message_id", ASCENDING)], unique=True
+    )
     logger.info(f"Created indexes [{feed_index}] [{pinned_index}]")
 
 

src/main/db/wrapper.py

@@ -129,7 +129,10 @@ def store_pinned_message(chat_id: int, message: Message) -> None:
     logger.info(f"[{chat_id}] Pinning message [{message.id}]")
     message_pickled = dumps(message)
     message_b64 = b64encode(message_pickled).decode("ascii")
-    insert_one({"chat_id": chat_id, "message": message_b64}, collection=DB_PINNED_NAME)
+    insert_one(
+        {"chat_id": chat_id, "message_id": message.id, "message": message_b64},
+        collection=DB_PINNED_NAME,
+    )
 
 
 def pop_pinned_messages(chat_id: int, bot: Bot) -> list[Message]:

src/test/db/test_client.py

@@ -66,7 +66,7 @@ def test_initialize_db(mongo_client_mock: MagicMock) -> None:
     create_pinned_index = pinned_collection_mock.create_index
     create_pinned_index.assert_called()
     create_pinned_index_kwargs = create_pinned_index.call_args.kwargs
-    expected_pinned_keys = [("chat_id", ASCENDING)]
+    expected_pinned_keys = [("chat_id", ASCENDING), ("message_id", ASCENDING)]
     assert expected_pinned_keys == create_pinned_index_kwargs.get("keys")
     assert create_pinned_index_kwargs.get("unique")