ShadowsocksX-NG/ShadowsocksX-NG2/Tests/Architecture/PlatformEffectsTests.swift

import XCTest

@testable import ShadowsocksX_NG2

@MainActor
final class QrImageSaverTests: XCTestCase {
  func testSuggestedFileNameReplacesPathReservedCharacters() {
    XCTAssertEqual(
      QrImageSaveDraft.suggestedFileName(from: "机房/一号:备用"),
      "机房 一号 备用.png")
  }

  func testSuggestedFileNameFallsBackWhenCleaningLeavesNothing() {
    XCTAssertEqual(QrImageSaveDraft.suggestedFileName(from: " /: "), "ss-qrcode.png")
    XCTAssertEqual(QrImageSaveDraft.suggestedFileName(from: ""), "ss-qrcode.png")
  }

  func testSuggestedFileNameKeepsCleanNameAndAppendsPNGExtension() {
    XCTAssertEqual(QrImageSaveDraft.suggestedFileName(from: "  家里  "), "家里.png")
  }
}

@MainActor
final class DiagnosticReportExportActionTests: DiagnosticsWorkflowTestCase {
  func testCancelledExportDoesNotRecordCompletion() {
    let workflow = makeWorkflow(catalog: { DiagnosticCatalogFacts() })
    let exporter = InMemoryDiagnosticReportExporter(result: .cancelled)

    let outcome = DiagnosticReportExportAction(
      diagnostics: workflow, exporter: exporter
    ).perform()

    XCTAssertEqual(outcome, .cancelled)
    XCTAssertEqual(exporter.drafts.count, 1)
    XCTAssertTrue(events.snapshot.isEmpty)
  }

  func testFailedExportDoesNotRecordCompletion() {
    let workflow = makeWorkflow(catalog: { DiagnosticCatalogFacts() })
    let exporter = InMemoryDiagnosticReportExporter(result: .failed(.writeFailed))

    let outcome = DiagnosticReportExportAction(
      diagnostics: workflow, exporter: exporter
    ).perform()

    XCTAssertEqual(outcome, .exportFailed(.writeFailed))
    XCTAssertEqual(exporter.drafts.count, 1)
    XCTAssertTrue(events.snapshot.isEmpty)
  }

  func testSuccessfulExportReturnsSelectedURLBeforeCompletionIsRecorded() throws {
    let workflow = makeWorkflow(catalog: { DiagnosticCatalogFacts() })
    let url = workDir.appendingPathComponent("diagnostics.txt")
    let exporter = InMemoryDiagnosticReportExporter(result: .saved(url))

    let outcome = DiagnosticReportExportAction(
      diagnostics: workflow, exporter: exporter
    ).perform()

    XCTAssertEqual(outcome, .saved(url))
    XCTAssertTrue(events.snapshot.isEmpty)

    workflow.noteExportCompleted()

    let exportedEvents = events.snapshot.filter { entry in
      if case .diagnosticsExported = entry.event { return true }
      return false
    }
    XCTAssertEqual(exportedEvents.count, 1)
  }

  func testReportPreparationFailureDoesNotOpenExporterOrRecordCompletion() {
    let workflow = makeWorkflow(
      catalog: { DiagnosticCatalogFacts() },
      render: { _ in nil })
    let exporter = InMemoryDiagnosticReportExporter(result: .cancelled)

    let outcome = DiagnosticReportExportAction(
      diagnostics: workflow, exporter: exporter
    ).perform()

    XCTAssertEqual(outcome, .preparationFailed(.encodingFailed))
    XCTAssertTrue(exporter.drafts.isEmpty)
    XCTAssertTrue(events.snapshot.isEmpty)
  }
}

final class PlatformEffectsArchitectureTests: XCTestCase {
  func testClipboardAndFileExportAppKitSymbolsStayInFocusedAdapters() throws {
    let appDirectory = Self.testTargetRoot.appendingPathComponent("App")
    let sourceFiles = try Self.swiftSourceFiles(in: appDirectory)

    let pasteboardUsers = try Self.files(containing: "NSPasteboard", in: sourceFiles)
    XCTAssertEqual(
      pasteboardUsers.map(\.lastPathComponent).sorted(),
      ["AppKitImageClipboard.swift", "AppKitTextClipboard.swift"],
      "pasteboard 访问必须集中在 clipboard adapter（文本与图片各一）")

    let savePanelUsers = try Self.files(containing: "NSSavePanel", in: sourceFiles)
    XCTAssertEqual(
      Set(savePanelUsers.map(\.lastPathComponent)),
      Set([
        "AppKitDiagnosticReportExporter.swift", "ConfigurationGroupFileExporter.swift",
        "QrImageSaver.swift",
      ]),
      "保存面板访问必须集中在对应的 AppKit exporter/saver adapter")

    let reportWriterUsers = try Self.files(containing: ".data.write(to:", in: sourceFiles)
    XCTAssertEqual(
      Set(reportWriterUsers.map(\.lastPathComponent)),
      Set([
        "AppKitDiagnosticReportExporter.swift", "ConfigurationGroupFileExporter.swift",
        "QrImageSaver.swift",
      ]),
      "文件写入必须集中在对应的 AppKit exporter/saver adapter")
  }

  private static let testTargetRoot = TestSourceTree.ng2Root()

  private static func swiftSourceFiles(in directory: URL) throws -> [URL] {
    guard
      let enumerator = FileManager.default.enumerator(
        at: directory, includingPropertiesForKeys: nil)
    else {
      XCTFail("无法枚举 App 源码目录：\(directory.path)")
      return []
    }
    return enumerator.compactMap { item in
      guard let url = item as? URL, url.pathExtension == "swift" else { return nil }
      return url
    }
  }

  private static func files(containing token: String, in files: [URL]) throws -> [URL] {
    var matches: [URL] = []
    for file in files where (try? String(contentsOf: file, encoding: .utf8).contains(token)) == true
    {
      matches.append(file)
    }
    return matches
  }
}