Austin Schuh | 272c613 | 2020-11-14 16:37:52 -0800 | [diff] [blame^] | 1 | import XCTest |
| 2 | @testable import FlatBuffers |
| 3 | |
| 4 | final class FlatBuffersDoubleTests: XCTestCase { |
| 5 | |
| 6 | let country = "Norway" |
| 7 | |
| 8 | func testCreateFinish() { |
| 9 | var b = FlatBufferBuilder(initialSize: 16) |
| 10 | let countryOff = CountryDouble.createCountry(builder: &b, name: country, log: 200, lan: 100) |
| 11 | b.finish(offset: countryOff) |
| 12 | let v: [UInt8] = [16, 0, 0, 0, 0, 0, 10, 0, 28, 0, 4, 0, 8, 0, 16, 0, 10, 0, 0, 0, 24, 0, 0, 0, 0, 0, 0, 0, 0, 0, 89, 64, 0, 0, 0, 0, 0, 0, 105, 64, 0, 0, 0, 0, 6, 0, 0, 0, 78, 111, 114, 119, 97, 121, 0, 0] |
| 13 | XCTAssertEqual(b.sizedByteArray, v) |
| 14 | } |
| 15 | |
| 16 | func testCreateFinishWithPrefix() { |
| 17 | var b = FlatBufferBuilder(initialSize: 16) |
| 18 | let countryOff = CountryDouble.createCountry(builder: &b, name: country, log: 200, lan: 100) |
| 19 | b.finish(offset: countryOff, addPrefix: true) |
| 20 | let v: [UInt8] = [60, 0, 0, 0, 20, 0, 0, 0, 0, 0, 0, 0, 0, 0, 10, 0, 28, 0, 4, 0, 8, 0, 16, 0, 10, 0, 0, 0, 24, 0, 0, 0, 0, 0, 0, 0, 0, 0, 89, 64, 0, 0, 0, 0, 0, 0, 105, 64, 0, 0, 0, 0, 6, 0, 0, 0, 78, 111, 114, 119, 97, 121, 0, 0] |
| 21 | XCTAssertEqual(b.sizedByteArray, v) |
| 22 | } |
| 23 | } |
| 24 | |
| 25 | class CountryDouble { |
| 26 | |
| 27 | static let offsets: (name: VOffset, lan: VOffset, lng: VOffset) = (4, 6, 8) |
| 28 | |
| 29 | private var table: Table |
| 30 | |
| 31 | private init(table t: Table) { table = t } |
| 32 | |
| 33 | static func getRootAsCountry(_ bb: ByteBuffer) -> CountryDouble { |
| 34 | let pos = bb.read(def: Int32.self, position: Int(bb.size)) |
| 35 | return CountryDouble(table: Table(bb: bb, position: Int32(pos))) |
| 36 | } |
| 37 | |
| 38 | static func createCountry(builder: inout FlatBufferBuilder, name: String, log: Double, lan: Double) -> Offset<Country> { |
| 39 | return createCountry(builder: &builder, offset: builder.create(string: name), log: log, lan: lan) |
| 40 | } |
| 41 | |
| 42 | static func createCountry(builder: inout FlatBufferBuilder, offset: Offset<String>, log: Double, lan: Double) -> Offset<Country> { |
| 43 | let _start = builder.startTable(with: 3) |
| 44 | CountryDouble.add(builder: &builder, lng: log) |
| 45 | CountryDouble.add(builder: &builder, lan: lan) |
| 46 | CountryDouble.add(builder: &builder, name: offset) |
| 47 | return CountryDouble.end(builder: &builder, startOffset: _start) |
| 48 | } |
| 49 | |
| 50 | static func end(builder: inout FlatBufferBuilder, startOffset: UOffset) -> Offset<Country> { |
| 51 | return Offset(offset: builder.endTable(at: startOffset)) |
| 52 | } |
| 53 | |
| 54 | static func add(builder: inout FlatBufferBuilder, name: String) { |
| 55 | add(builder: &builder, name: builder.create(string: name)) |
| 56 | } |
| 57 | |
| 58 | static func add(builder: inout FlatBufferBuilder, name: Offset<String>) { |
| 59 | builder.add(offset: name, at: Country.offsets.name) |
| 60 | } |
| 61 | |
| 62 | static func add(builder: inout FlatBufferBuilder, lan: Double) { |
| 63 | builder.add(element: lan, def: 0, at: Country.offsets.lan) |
| 64 | } |
| 65 | |
| 66 | static func add(builder: inout FlatBufferBuilder, lng: Double) { |
| 67 | builder.add(element: lng, def: 0, at: Country.offsets.lng) |
| 68 | } |
| 69 | } |