| import XCTest |
| import Foundation |
| @testable import FlatBuffers |
| |
| typealias Test1 = MyGame.Example.Test |
| typealias Monster1 = MyGame.Example.Monster |
| typealias Vec3 = MyGame.Example.Vec3 |
| |
| class FlatBuffersMonsterWriterTests: XCTestCase { |
| |
| func testData() { |
| let data = Data([48, 0, 0, 0, 77, 79, 78, 83, 0, 0, 0, 0, 36, 0, 72, 0, 40, 0, 0, 0, 38, 0, 32, 0, 0, 0, 28, 0, 0, 0, 27, 0, 20, 0, 16, 0, 12, 0, 4, 0, 0, 0, 0, 0, 0, 0, 11, 0, 36, 0, 0, 0, 164, 0, 0, 0, 0, 0, 0, 1, 60, 0, 0, 0, 68, 0, 0, 0, 76, 0, 0, 0, 0, 0, 0, 1, 88, 0, 0, 0, 120, 0, 0, 0, 0, 0, 80, 0, 0, 0, 128, 63, 0, 0, 0, 64, 0, 0, 64, 64, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 8, 64, 2, 0, 5, 0, 6, 0, 0, 0, 2, 0, 0, 0, 64, 0, 0, 0, 48, 0, 0, 0, 2, 0, 0, 0, 30, 0, 40, 0, 10, 0, 20, 0, 152, 255, 255, 255, 4, 0, 0, 0, 4, 0, 0, 0, 70, 114, 101, 100, 0, 0, 0, 0, 5, 0, 0, 0, 0, 1, 2, 3, 4, 0, 0, 0, 5, 0, 0, 0, 116, 101, 115, 116, 50, 0, 0, 0, 5, 0, 0, 0, 116, 101, 115, 116, 49, 0, 0, 0, 9, 0, 0, 0, 77, 121, 77, 111, 110, 115, 116, 101, 114, 0, 0, 0, 3, 0, 0, 0, 20, 0, 0, 0, 36, 0, 0, 0, 4, 0, 0, 0, 240, 255, 255, 255, 32, 0, 0, 0, 248, 255, 255, 255, 36, 0, 0, 0, 12, 0, 8, 0, 0, 0, 0, 0, 0, 0, 4, 0, 12, 0, 0, 0, 28, 0, 0, 0, 5, 0, 0, 0, 87, 105, 108, 109, 97, 0, 0, 0, 6, 0, 0, 0, 66, 97, 114, 110, 101, 121, 0, 0, 5, 0, 0, 0, 70, 114, 111, 100, 111, 0, 0, 0]) |
| let _data = ByteBuffer(data: data) |
| readMonster(fb: _data) |
| } |
| |
| func testReadFromOtherLangagues() { |
| let path = FileManager.default.currentDirectoryPath |
| let url = URL(fileURLWithPath: path, isDirectory: true).appendingPathComponent("monsterdata_test").appendingPathExtension("mon") |
| guard let data = try? Data(contentsOf: url) else { return } |
| let _data = ByteBuffer(data: data) |
| readMonster(fb: _data) |
| } |
| |
| func testCreateMonster() { |
| let bytes = createMonster(withPrefix: false) |
| XCTAssertEqual(bytes.sizedByteArray, [48, 0, 0, 0, 77, 79, 78, 83, 0, 0, 0, 0, 36, 0, 72, 0, 40, 0, 0, 0, 38, 0, 32, 0, 0, 0, 28, 0, 0, 0, 27, 0, 20, 0, 16, 0, 12, 0, 4, 0, 0, 0, 0, 0, 0, 0, 11, 0, 36, 0, 0, 0, 164, 0, 0, 0, 0, 0, 0, 1, 60, 0, 0, 0, 68, 0, 0, 0, 76, 0, 0, 0, 0, 0, 0, 1, 88, 0, 0, 0, 120, 0, 0, 0, 0, 0, 80, 0, 0, 0, 128, 63, 0, 0, 0, 64, 0, 0, 64, 64, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 8, 64, 2, 0, 5, 0, 6, 0, 0, 0, 2, 0, 0, 0, 64, 0, 0, 0, 48, 0, 0, 0, 2, 0, 0, 0, 30, 0, 40, 0, 10, 0, 20, 0, 152, 255, 255, 255, 4, 0, 0, 0, 4, 0, 0, 0, 70, 114, 101, 100, 0, 0, 0, 0, 5, 0, 0, 0, 0, 1, 2, 3, 4, 0, 0, 0, 5, 0, 0, 0, 116, 101, 115, 116, 50, 0, 0, 0, 5, 0, 0, 0, 116, 101, 115, 116, 49, 0, 0, 0, 9, 0, 0, 0, 77, 121, 77, 111, 110, 115, 116, 101, 114, 0, 0, 0, 3, 0, 0, 0, 20, 0, 0, 0, 36, 0, 0, 0, 4, 0, 0, 0, 240, 255, 255, 255, 32, 0, 0, 0, 248, 255, 255, 255, 36, 0, 0, 0, 12, 0, 8, 0, 0, 0, 0, 0, 0, 0, 4, 0, 12, 0, 0, 0, 28, 0, 0, 0, 5, 0, 0, 0, 87, 105, 108, 109, 97, 0, 0, 0, 6, 0, 0, 0, 66, 97, 114, 110, 101, 121, 0, 0, 5, 0, 0, 0, 70, 114, 111, 100, 111, 0, 0, 0]) |
| readMonster(fb: bytes.buffer) |
| mutateMonster(fb: bytes.buffer) |
| readMonster(fb: bytes.buffer) |
| } |
| |
| func testCreateMonsterResizedBuffer() { |
| let bytes = createMonster(withPrefix: false) |
| XCTAssertEqual(bytes.sizedByteArray, [48, 0, 0, 0, 77, 79, 78, 83, 0, 0, 0, 0, 36, 0, 72, 0, 40, 0, 0, 0, 38, 0, 32, 0, 0, 0, 28, 0, 0, 0, 27, 0, 20, 0, 16, 0, 12, 0, 4, 0, 0, 0, 0, 0, 0, 0, 11, 0, 36, 0, 0, 0, 164, 0, 0, 0, 0, 0, 0, 1, 60, 0, 0, 0, 68, 0, 0, 0, 76, 0, 0, 0, 0, 0, 0, 1, 88, 0, 0, 0, 120, 0, 0, 0, 0, 0, 80, 0, 0, 0, 128, 63, 0, 0, 0, 64, 0, 0, 64, 64, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 8, 64, 2, 0, 5, 0, 6, 0, 0, 0, 2, 0, 0, 0, 64, 0, 0, 0, 48, 0, 0, 0, 2, 0, 0, 0, 30, 0, 40, 0, 10, 0, 20, 0, 152, 255, 255, 255, 4, 0, 0, 0, 4, 0, 0, 0, 70, 114, 101, 100, 0, 0, 0, 0, 5, 0, 0, 0, 0, 1, 2, 3, 4, 0, 0, 0, 5, 0, 0, 0, 116, 101, 115, 116, 50, 0, 0, 0, 5, 0, 0, 0, 116, 101, 115, 116, 49, 0, 0, 0, 9, 0, 0, 0, 77, 121, 77, 111, 110, 115, 116, 101, 114, 0, 0, 0, 3, 0, 0, 0, 20, 0, 0, 0, 36, 0, 0, 0, 4, 0, 0, 0, 240, 255, 255, 255, 32, 0, 0, 0, 248, 255, 255, 255, 36, 0, 0, 0, 12, 0, 8, 0, 0, 0, 0, 0, 0, 0, 4, 0, 12, 0, 0, 0, 28, 0, 0, 0, 5, 0, 0, 0, 87, 105, 108, 109, 97, 0, 0, 0, 6, 0, 0, 0, 66, 97, 114, 110, 101, 121, 0, 0, 5, 0, 0, 0, 70, 114, 111, 100, 111, 0, 0, 0]) |
| readMonster(fb: bytes.sizedBuffer) |
| } |
| |
| func testCreateMonsterPrefixed() { |
| let bytes = createMonster(withPrefix: true) |
| XCTAssertEqual(bytes.sizedByteArray, [44, 1, 0, 0, 44, 0, 0, 0, 77, 79, 78, 83, 36, 0, 72, 0, 40, 0, 0, 0, 38, 0, 32, 0, 0, 0, 28, 0, 0, 0, 27, 0, 20, 0, 16, 0, 12, 0, 4, 0, 0, 0, 0, 0, 0, 0, 11, 0, 36, 0, 0, 0, 164, 0, 0, 0, 0, 0, 0, 1, 60, 0, 0, 0, 68, 0, 0, 0, 76, 0, 0, 0, 0, 0, 0, 1, 88, 0, 0, 0, 120, 0, 0, 0, 0, 0, 80, 0, 0, 0, 128, 63, 0, 0, 0, 64, 0, 0, 64, 64, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 8, 64, 2, 0, 5, 0, 6, 0, 0, 0, 2, 0, 0, 0, 64, 0, 0, 0, 48, 0, 0, 0, 2, 0, 0, 0, 30, 0, 40, 0, 10, 0, 20, 0, 152, 255, 255, 255, 4, 0, 0, 0, 4, 0, 0, 0, 70, 114, 101, 100, 0, 0, 0, 0, 5, 0, 0, 0, 0, 1, 2, 3, 4, 0, 0, 0, 5, 0, 0, 0, 116, 101, 115, 116, 50, 0, 0, 0, 5, 0, 0, 0, 116, 101, 115, 116, 49, 0, 0, 0, 9, 0, 0, 0, 77, 121, 77, 111, 110, 115, 116, 101, 114, 0, 0, 0, 3, 0, 0, 0, 20, 0, 0, 0, 36, 0, 0, 0, 4, 0, 0, 0, 240, 255, 255, 255, 32, 0, 0, 0, 248, 255, 255, 255, 36, 0, 0, 0, 12, 0, 8, 0, 0, 0, 0, 0, 0, 0, 4, 0, 12, 0, 0, 0, 28, 0, 0, 0, 5, 0, 0, 0, 87, 105, 108, 109, 97, 0, 0, 0, 6, 0, 0, 0, 66, 97, 114, 110, 101, 121, 0, 0, 5, 0, 0, 0, 70, 114, 111, 100, 111, 0, 0, 0]) |
| |
| let newBuf = FlatBuffersUtils.removeSizePrefix(bb: bytes.buffer) |
| readMonster(fb: newBuf) |
| } |
| |
| func readMonster(fb: ByteBuffer) { |
| var monster = Monster1.getRootAsMonster(bb: fb) |
| readFlatbufferMonster(monster: &monster) |
| var unpacked: MyGame.Example.MonsterT? = monster.unpack() |
| readObjectApi(monster: unpacked!) |
| var builder = FlatBufferBuilder() |
| let root = Monster1.pack(&builder, obj: &unpacked) |
| builder.finish(offset: root) |
| var newMonster = Monster1.getRootAsMonster(bb: builder.sizedBuffer) |
| readFlatbufferMonster(monster: &newMonster) |
| } |
| |
| func createMonster(withPrefix prefix: Bool) -> FlatBufferBuilder { |
| var fbb = FlatBufferBuilder(initialSize: 1) |
| let names = [fbb.create(string: "Frodo"), fbb.create(string: "Barney"), fbb.create(string: "Wilma")] |
| var offsets: [Offset<UOffset>] = [] |
| let start1 = Monster1.startMonster(&fbb) |
| Monster1.add(name: names[0], &fbb) |
| offsets.append(Monster1.endMonster(&fbb, start: start1)) |
| let start2 = Monster1.startMonster(&fbb) |
| Monster1.add(name: names[1], &fbb) |
| offsets.append(Monster1.endMonster(&fbb, start: start2)) |
| let start3 = Monster1.startMonster(&fbb) |
| Monster1.add(name: names[2], &fbb) |
| offsets.append(Monster1.endMonster(&fbb, start: start3)) |
| |
| let sortedArray = Monster1.sortVectorOfMonster(offsets: offsets, &fbb) |
| |
| let str = fbb.create(string: "MyMonster") |
| let test1 = fbb.create(string: "test1") |
| let test2 = fbb.create(string: "test2") |
| let _inv: [Byte] = [0, 1, 2, 3, 4] |
| let inv = fbb.createVector(_inv) |
| |
| let fred = fbb.create(string: "Fred") |
| let mon1Start = Monster1.startMonster(&fbb) |
| Monster1.add(name: fred, &fbb) |
| let mon2 = Monster1.endMonster(&fbb, start: mon1Start) |
| let test4 = fbb.createVector(structs: [MyGame.Example.createTest(a: 30, b: 40), |
| MyGame.Example.createTest(a: 10, b: 20)], |
| type: Test1.self) |
| |
| let stringTestVector = fbb.createVector(ofOffsets: [test1, test2]) |
| |
| let mStart = Monster1.startMonster(&fbb) |
| let posOffset = fbb.create(struct: MyGame.Example.createVec3(x: 1, y: 2, z: 3, test1: 3, test2: .green, test3a: 5, test3b: 6), type: Vec3.self) |
| Monster1.add(pos: posOffset, &fbb) |
| Monster1.add(hp: 80, &fbb) |
| Monster1.add(name: str, &fbb) |
| Monster1.addVectorOf(inventory: inv, &fbb) |
| Monster1.add(testType: .monster, &fbb) |
| Monster1.add(test: mon2, &fbb) |
| Monster1.addVectorOf(test4: test4, &fbb) |
| Monster1.addVectorOf(testarrayofstring: stringTestVector, &fbb) |
| Monster1.add(testbool: true, &fbb) |
| Monster1.addVectorOf(testarrayoftables: sortedArray, &fbb) |
| let end = Monster1.endMonster(&fbb, start: mStart) |
| Monster1.finish(&fbb, end: end, prefix: prefix) |
| return fbb |
| } |
| |
| func mutateMonster(fb: ByteBuffer) { |
| let monster = Monster1.getRootAsMonster(bb: fb) |
| XCTAssertFalse(monster.mutate(mana: 10)) |
| XCTAssertEqual(monster.testarrayoftables(at: 0)?.name, "Barney") |
| XCTAssertEqual(monster.testarrayoftables(at: 1)?.name, "Frodo") |
| XCTAssertEqual(monster.testarrayoftables(at: 2)?.name, "Wilma") |
| |
| // Example of searching for a table by the key |
| XCTAssertNotNil(monster.testarrayoftablesBy(key: "Frodo")) |
| XCTAssertNotNil(monster.testarrayoftablesBy(key: "Barney")) |
| XCTAssertNotNil(monster.testarrayoftablesBy(key: "Wilma")) |
| |
| XCTAssertEqual(monster.testType, .monster) |
| |
| XCTAssertEqual(monster.mutate(inventory: 1, at: 0), true) |
| XCTAssertEqual(monster.mutate(inventory: 2, at: 1), true) |
| XCTAssertEqual(monster.mutate(inventory: 3, at: 2), true) |
| XCTAssertEqual(monster.mutate(inventory: 4, at: 3), true) |
| XCTAssertEqual(monster.mutate(inventory: 5, at: 4), true) |
| |
| for i in 0..<monster.inventoryCount { |
| XCTAssertEqual(monster.inventory(at: i), Byte(i + 1)) |
| } |
| |
| XCTAssertEqual(monster.mutate(inventory: 0, at: 0), true) |
| XCTAssertEqual(monster.mutate(inventory: 1, at: 1), true) |
| XCTAssertEqual(monster.mutate(inventory: 2, at: 2), true) |
| XCTAssertEqual(monster.mutate(inventory: 3, at: 3), true) |
| XCTAssertEqual(monster.mutate(inventory: 4, at: 4), true) |
| |
| let vec = monster.pos |
| XCTAssertEqual(vec?.x, 1) |
| XCTAssertTrue(vec?.mutate(x: 55.0) ?? false) |
| XCTAssertTrue(vec?.mutate(test1: 55) ?? false) |
| XCTAssertEqual(vec?.x, 55.0) |
| XCTAssertEqual(vec?.test1, 55.0) |
| XCTAssertTrue(vec?.mutate(x: 1) ?? false) |
| XCTAssertEqual(vec?.x, 1) |
| XCTAssertTrue(vec?.mutate(test1: 3) ?? false) |
| } |
| |
| func readFlatbufferMonster(monster: inout MyGame.Example.Monster) { |
| XCTAssertEqual(monster.hp, 80) |
| XCTAssertEqual(monster.mana, 150) |
| XCTAssertEqual(monster.name, "MyMonster") |
| let pos = monster.pos |
| XCTAssertEqual(pos?.x, 1) |
| XCTAssertEqual(pos?.y, 2) |
| XCTAssertEqual(pos?.z, 3) |
| XCTAssertEqual(pos?.test1, 3) |
| XCTAssertEqual(pos?.test2, .green) |
| let test = pos?.test3 |
| XCTAssertEqual(test?.a, 5) |
| XCTAssertEqual(test?.b, 6) |
| XCTAssertEqual(monster.testType, .monster) |
| let monster2 = monster.test(type: Monster1.self) |
| XCTAssertEqual(monster2?.name, "Fred") |
| |
| XCTAssertEqual(monster.mutate(mana: 10), false) |
| |
| XCTAssertEqual(monster.mana, 150) |
| XCTAssertEqual(monster.inventoryCount, 5) |
| var sum: Byte = 0 |
| for i in 0...monster.inventoryCount { |
| sum += monster.inventory(at: i) |
| } |
| XCTAssertEqual(sum, 10) |
| XCTAssertEqual(monster.test4Count, 2) |
| let test0 = monster.test4(at: 0) |
| let test1 = monster.test4(at: 1) |
| var sum0 = 0 |
| var sum1 = 0 |
| if let a = test0?.a, let b = test0?.b { |
| sum0 = Int(a) + Int(b) |
| } |
| if let a = test1?.a, let b = test1?.b { |
| sum1 = Int(a) + Int(b) |
| } |
| XCTAssertEqual(sum0 + sum1, 100) |
| XCTAssertEqual(monster.testarrayofstringCount, 2) |
| XCTAssertEqual(monster.testarrayofstring(at: 0), "test1") |
| XCTAssertEqual(monster.testarrayofstring(at: 1), "test2") |
| XCTAssertEqual(monster.testbool, true) |
| |
| let array = monster.nameSegmentArray |
| XCTAssertEqual(String(bytes: array ?? [], encoding: .utf8), "MyMonster") |
| |
| if 0 == monster.testarrayofboolsCount { |
| XCTAssertEqual(monster.testarrayofbools.isEmpty, true) |
| } else { |
| XCTAssertEqual(monster.testarrayofbools.isEmpty, false) |
| } |
| } |
| |
| func readObjectApi(monster: MyGame.Example.MonsterT) { |
| XCTAssertEqual(monster.hp, 80) |
| XCTAssertEqual(monster.mana, 150) |
| XCTAssertEqual(monster.name, "MyMonster") |
| let pos = monster.pos |
| XCTAssertEqual(pos?.x, 1) |
| XCTAssertEqual(pos?.y, 2) |
| XCTAssertEqual(pos?.z, 3) |
| XCTAssertEqual(pos?.test1, 3) |
| XCTAssertEqual(pos?.test2, .green) |
| let test = pos?.test3 |
| XCTAssertEqual(test?.a, 5) |
| XCTAssertEqual(test?.b, 6) |
| let monster2 = monster.test?.value as? MyGame.Example.MonsterT |
| XCTAssertEqual(monster2?.name, "Fred") |
| XCTAssertEqual(monster.mana, 150) |
| monster.mana = 10 |
| XCTAssertEqual(monster.mana, 10) |
| monster.mana = 150 |
| XCTAssertEqual(monster.mana, 150) |
| |
| XCTAssertEqual(monster.inventory.count, 5) |
| var sum: Byte = 0 |
| for i in monster.inventory { |
| sum += i |
| } |
| XCTAssertEqual(sum, 10) |
| XCTAssertEqual(monster.test4.count, 2) |
| let test0 = monster.test4[0] |
| let test1 = monster.test4[1] |
| var sum0 = 0 |
| var sum1 = 0 |
| if let a = test0?.a, let b = test0?.b { |
| sum0 = Int(a) + Int(b) |
| } |
| if let a = test1?.a, let b = test1?.b { |
| sum1 = Int(a) + Int(b) |
| } |
| XCTAssertEqual(sum0 + sum1, 100) |
| XCTAssertEqual(monster.testbool, true) |
| } |
| } |
