blob: 6f61ec6fd0accf3ef28f7aeb28198eadcaa87289 [file] [log] [blame]
Austin Schuh272c6132020-11-14 16:37:52 -08001import XCTest
2@testable import FlatBuffers
3
4final 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
25class 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}