0.10.6

Tests/MessageCorrectionTests.swift

@@ -0,0 +1,69 @@
+import Foundation
+import XCTest
+@testable import Luma
+
+final class MessageCorrectionTests: XCTestCase {
+    func testOnlyDeliveredOrSentOutgoingTextCanBeEdited() {
+        let editable = ChatMessage(
+            conversationID: "bob@example.org",
+            senderJID: "alice@example.org",
+            body: "До исправления",
+            direction: .outgoing,
+            delivery: .sent,
+            security: .omemo
+        )
+        var incoming = editable
+        incoming.direction = .incoming
+        var attachment = editable
+        attachment.kind = .attachment
+        var failed = editable
+        failed.delivery = .failed
+        var retracted = editable
+        retracted.retractedAt = Date()
+
+        XCTAssertTrue(editable.canBeEdited)
+        XCTAssertFalse(incoming.canBeEdited)
+        XCTAssertFalse(attachment.canBeEdited)
+        XCTAssertFalse(failed.canBeEdited)
+        XCTAssertFalse(retracted.canBeEdited)
+    }
+
+    func testEditedTimestampRoundTrips() throws {
+        let editedAt = Date(timeIntervalSince1970: 1_700_000_000)
+        let message = ChatMessage(
+            conversationID: "bob@example.org",
+            senderJID: "alice@example.org",
+            body: "После исправления",
+            direction: .outgoing,
+            delivery: .delivered,
+            security: .plaintext,
+            editedAt: editedAt
+        )
+
+        let data = try JSONEncoder().encode(message)
+        let decoded = try JSONDecoder().decode(ChatMessage.self, from: data)
+
+        XCTAssertEqual(decoded.editedAt, editedAt)
+    }
+
+    func testLegacyMessageWithoutEditedTimestampStillDecodes() throws {
+        let message = ChatMessage(
+            conversationID: "bob@example.org",
+            senderJID: "alice@example.org",
+            body: "Старое сообщение",
+            direction: .incoming,
+            delivery: .delivered,
+            security: .omemo
+        )
+        let encoded = try JSONEncoder().encode(message)
+        var object = try XCTUnwrap(JSONSerialization.jsonObject(with: encoded) as? [String: Any])
+        object.removeValue(forKey: "editedAt")
+
+        let decoded = try JSONDecoder().decode(
+            ChatMessage.self,
+            from: JSONSerialization.data(withJSONObject: object)
+        )
+
+        XCTAssertNil(decoded.editedAt)
+    }
+}