FlatBuffersMonsterWriterTests.swift

import XCTest
import Foundation
@testable import FlatBuffers

public typealias Test = MyGame.Example.Test
public typealias Monster = MyGame.Example.Monster
public typealias Vec3 = MyGame.Example.Vec3
public typealias Stat = MyGame.Example.Stat

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 = Monster.getRootAsMonster(bb: fb)
        readFlatbufferMonster(monster: &monster)
        var unpacked: MyGame.Example.MonsterT? = monster.unpack()
        readObjectApi(monster: unpacked!)
        var builder = FlatBufferBuilder()
        let root = Monster.pack(&builder, obj: &unpacked)
        builder.finish(offset: root)
        var newMonster = Monster.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 = Monster.startMonster(&fbb)
        Monster.add(name: names[0], &fbb)
        offsets.append(Monster.endMonster(&fbb, start: start1))
        let start2 = Monster.startMonster(&fbb)
        Monster.add(name: names[1], &fbb)
        offsets.append(Monster.endMonster(&fbb, start: start2))
        let start3 = Monster.startMonster(&fbb)
        Monster.add(name: names[2], &fbb)
        offsets.append(Monster.endMonster(&fbb, start: start3))
        
        let sortedArray = Monster.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 = Monster.startMonster(&fbb)
        Monster.add(name: fred, &fbb)
        let mon2 = Monster.endMonster(&fbb, start: mon1Start)
        let test4 = fbb.createVector(structs: [MyGame.Example.createTest(a: 30, b: 40),
                                               MyGame.Example.createTest(a: 10, b: 20)],
                                     type: Test.self)
        
        let stringTestVector = fbb.createVector(ofOffsets: [test1, test2])
        
        let mStart = Monster.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)
        Monster.add(pos: posOffset, &fbb)
        Monster.add(hp: 80, &fbb)
        Monster.add(name: str, &fbb)
        Monster.addVectorOf(inventory: inv, &fbb)
        Monster.add(testType: .monster, &fbb)
        Monster.add(test: mon2, &fbb)
        Monster.addVectorOf(test4: test4, &fbb)
        Monster.addVectorOf(testarrayofstring: stringTestVector, &fbb)
        Monster.add(testbool: true, &fbb)
        Monster.addVectorOf(testarrayoftables: sortedArray, &fbb)
        let end = Monster.endMonster(&fbb, start: mStart)
        Monster.finish(&fbb, end: end, prefix: prefix)
        return fbb
    }
    
    func mutateMonster(fb: ByteBuffer) {
        let monster = Monster.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: Monster.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)
    }
}