bitchat/bitchatTests/Services/NotificationRedactionTests.swift

import BitFoundation
import Foundation
import Testing
import UserNotifications
@testable import bitchat

/// Lock-screen notifications used to carry the DM body and the sender's
/// nickname verbatim, and the geohash in the title, so a locked phone lying on
/// a table narrated conversations to anyone looking at it. These cover the
/// redaction that is now the default.
///
/// Serialized because the preference lives in `UserDefaults.standard`; parallel
/// cases would race each other's saves.
@Suite(.serialized)
struct NotificationRedactionTests {
    private final class RecordingDeliverer: NotificationRequestDelivering {
        var requests: [UNNotificationRequest] = []

        func add(_ request: UNNotificationRequest) {
            requests.append(request)
        }
    }

    private struct StubAuthorizer: NotificationAuthorizing {
        func requestAuthorization(
            options: UNAuthorizationOptions,
            completionHandler: @escaping (Bool, Error?) -> Void
        ) {
            completionHandler(true, nil)
        }
    }

    /// Runs `body` with the preference forced to `hidePreviews`, restoring
    /// whatever the environment had afterwards.
    private func withHidePreviews(
        _ hidePreviews: Bool,
        _ body: (NotificationService, RecordingDeliverer) throws -> Void
    ) rethrows {
        let original = NotificationPrivacySettings.hideMessagePreviews
        defer { NotificationPrivacySettings.hideMessagePreviews = original }
        NotificationPrivacySettings.hideMessagePreviews = hidePreviews

        let deliverer = RecordingDeliverer()
        let service = NotificationService(
            isRunningTestsProvider: { false },
            authorizer: StubAuthorizer(),
            requestDeliverer: deliverer
        )
        try body(service, deliverer)
    }

    @Test func previewsAreHiddenByDefault() {
        // A fresh install must start quiet rather than opt-in quiet.
        UserDefaults.standard.removeObject(forKey: "notifications.hideMessagePreviews")
        #expect(NotificationPrivacySettings.hideMessagePreviews)
    }

    @Test func redactedDirectMessageWithholdsSenderAndBody() {
        withHidePreviews(true) { service, deliverer in
            service.sendPrivateMessageNotification(
                from: "alice",
                message: "meet at the north gate",
                peerID: PeerID(str: "00112233445566ff")
            )

            let content = try! #require(deliverer.requests.first).content
            #expect(!content.title.contains("alice"))
            #expect(!content.body.contains("north gate"))
            #expect(!content.title.isEmpty)
            // Still routable: userInfo is never rendered on the lock screen.
            #expect(content.userInfo["peerID"] as? String == "00112233445566ff")
        }
    }

    @Test func redactedMentionWithholdsSenderAndBody() {
        withHidePreviews(true) { service, deliverer in
            service.sendMentionNotification(from: "bob", message: "regroup now")

            let content = try! #require(deliverer.requests.first).content
            #expect(!content.title.contains("bob"))
            #expect(!content.body.contains("regroup"))
        }
    }

    @Test func redactedGeohashActivityWithholdsTheGeohash() {
        withHidePreviews(true) { service, deliverer in
            service.sendGeohashActivityNotification(
                geohash: "u4pruyd",
                bodyPreview: "someone said something"
            )

            let content = try! #require(deliverer.requests.first).content
            #expect(!content.title.contains("u4pruyd"))
            #expect(!content.body.contains("someone said"))
            // The deep link still carries it: tapping must land in the channel.
            #expect(content.userInfo["deeplink"] as? String == "bitchat://geohash/u4pruyd")
        }
    }

    @Test func previewsShownWhenTheSettingIsOff() {
        withHidePreviews(false) { service, deliverer in
            service.sendPrivateMessageNotification(
                from: "alice",
                message: "meet at the north gate",
                peerID: PeerID(str: "00112233445566ff")
            )
            service.sendGeohashActivityNotification(
                geohash: "u4pruyd",
                bodyPreview: "someone said something"
            )

            #expect(deliverer.requests.count == 2)
            let dm = deliverer.requests[0].content
            #expect(dm.title.contains("alice"))
            #expect(dm.body == "meet at the north gate")
            let geo = deliverer.requests[1].content
            #expect(geo.title.contains("u4pruyd"))
            #expect(geo.body == "someone said something")
        }
    }
}