permissionlesstech/bitchat
1
0
Fork 0
mirror of https://github.com/permissionlesstech/bitchat.git synced 2026-07-25 15:25:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Convert new tests to Swift Testing

Browse Source
This commit is contained in:
islam
2025-10-15 17:37:19 +01:00
parent 830f7ee4b8
commit 2b3264b581
2 changed files with 18 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bitchatTests/NotificationStreamAssemblerTests.swift
+17 -27
View File
@@ -107,9 +107,7 @@ struct NotificationStreamAssemblerTests {
mimeType: "application/octet-stream",
content: largeContent
)
guard let tlvPayload = filePacket.encode() else {
return XCTFail("Failed to encode file packet")
}
let tlvPayload = try #require(filePacket.encode(), "Failed to encode file packet")
let senderID = Data(repeating: 0xAA, count: BinaryProtocol.senderIDSize)
let packet = BitchatPacket(
@@ -123,39 +121,31 @@ struct NotificationStreamAssemblerTests {
version: 2
)
guard let frame = packet.toBinaryData(padding: false) else {
return XCTFail("Failed to encode packet frame")
}
let frame = try #require(packet.toBinaryData(padding: false), "Failed to encode packet frame")
XCTAssertLessThan(BinaryProtocol.Offsets.flags, frame.count)
#expect(BinaryProtocol.Offsets.flags < frame.count)
let flags = frame[frame.startIndex + BinaryProtocol.Offsets.flags]
XCTAssertNotEqual(flags & BinaryProtocol.Flags.isCompressed, 0, "Frame should be compressed for large payloads")
#expect((flags & BinaryProtocol.Flags.isCompressed) != 0, "Frame should be compressed for large payloads")
let splitIndex = min(4096, frame.count / 2)
var result = assembler.append(frame.prefix(splitIndex))
XCTAssertTrue(result.frames.isEmpty)
#expect(result.frames.isEmpty)
result = assembler.append(frame.suffix(from: splitIndex))
XCTAssertEqual(result.frames.count, 1)
XCTAssertTrue(result.droppedPrefixes.isEmpty)
XCTAssertFalse(result.reset)
#expect(result.frames.count == 1)
#expect(result.droppedPrefixes.isEmpty)
#expect(result.reset == false)
guard let assembled = result.frames.first else {
return XCTFail("Missing assembled frame")
}
XCTAssertEqual(assembled.count, frame.count)
let assembled = try #require(result.frames.first, "Missing assembled frame")
#expect(assembled.count == frame.count)
guard let decodedPacket = BinaryProtocol.decode(assembled) else {
return XCTFail("Failed to decode compressed frame")
}
XCTAssertEqual(decodedPacket.payload.count, tlvPayload.count)
let decodedPacket = try #require(BinaryProtocol.decode(assembled), "Failed to decode compressed frame")
#expect(decodedPacket.payload.count == tlvPayload.count)
guard let decodedFile = BitchatFilePacket.decode(decodedPacket.payload) else {
return XCTFail("Failed to decode TLV payload")
}
XCTAssertEqual(decodedFile.fileName, filePacket.fileName)
XCTAssertEqual(decodedFile.mimeType, filePacket.mimeType)
XCTAssertEqual(decodedFile.content.count, largeContent.count)
XCTAssertEqual(decodedFile.content.prefix(32), largeContent.prefix(32))
let decodedFile = try #require(BitchatFilePacket.decode(decodedPacket.payload), "Failed to decode TLV payload")
#expect(decodedFile.fileName == filePacket.fileName)
#expect(decodedFile.mimeType == filePacket.mimeType)
#expect(decodedFile.content.count == largeContent.count)
#expect(decodedFile.content.prefix(32) == largeContent.prefix(32))
}
}
bitchatTests/Protocol/BinaryProtocolTests.swift
+1 -4
View File
@@ -68,10 +68,7 @@ struct BinaryProtocolTests {
let encodedData = try #require(BinaryProtocol.encode(packet), "Failed to encode packet with large payload")
// The encoded size should be smaller than uncompressed due to compression
guard let headerSize = BinaryProtocol.headerSize(for: packet.version) else {
XCTFail("Invalid version")
return
}
let headerSize = try #require(BinaryProtocol.headerSize(for: packet.version), "Invalid packet version")
let uncompressedSize = headerSize + BinaryProtocol.senderIDSize + largePayload.count
#expect(encodedData.count < uncompressedSize, "Compressed packet should be smaller than uncompressed form")