Add tests for mirror and applying

Author: Pouya Yarandi

Sources/SwiftProtobuf/ProtobufMirror.swift

@@ -15,7 +15,7 @@
 import Foundation
 
 /// ProtobufMirror is for mirroring a protobuf Message type.
-/// 
+///
 /// By using this object, a Message is being mirrored and it enables us to
 /// get all its children by their fieldNumber and value. Notice that it is not
 /// type safe and it should be used only in cases you need to get values
@@ -58,18 +58,54 @@ struct ReflectionVisitor: Visitor {
 
   internal init() {}
 
+  mutating func visitSingularFloatField(value: Float, fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
   mutating func visitSingularDoubleField(value: Double, fieldNumber: Int) throws {
     mirror.set(value, for: fieldNumber)
   }
 
+  mutating func visitSingularInt32Field(value: Int32, fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
   mutating func visitSingularInt64Field(value: Int64, fieldNumber: Int) throws {
     mirror.set(value, for: fieldNumber)
   }
 
+  mutating func visitSingularUInt32Field(value: UInt32, fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
   mutating func visitSingularUInt64Field(value: UInt64, fieldNumber: Int) throws {
     mirror.set(value, for: fieldNumber)
   }
 
+  mutating func visitSingularSInt32Field(value: Int32, fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitSingularSInt64Field(value: Int64, fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitSingularFixed32Field(value: UInt32, fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitSingularFixed64Field(value: UInt64, fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitSingularSFixed32Field(value: Int32, fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitSingularSFixed64Field(value: Int64, fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
   mutating func visitSingularBoolField(value: Bool, fieldNumber: Int) throws {
     mirror.set(value, for: fieldNumber)
   }
@@ -82,11 +118,146 @@ struct ReflectionVisitor: Visitor {
     mirror.set(value, for: fieldNumber)
   }
 
-  mutating func visitSingularEnumField<E>(value: E, fieldNumber: Int) throws {
+  mutating func visitSingularEnumField<E: Enum>(value: E, fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitSingularMessageField<M: Message>(value: M, fieldNumber: Int) throws {
     mirror.set(value, for: fieldNumber)
   }
 
-  mutating func visitSingularMessageField<M>(value: M, fieldNumber: Int) throws {
+  mutating func visitSingularGroupField<G: Message>(value: G, fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitRepeatedFloatField(value: [Float], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitRepeatedDoubleField(value: [Double], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitRepeatedInt32Field(value: [Int32], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitRepeatedInt64Field(value: [Int64], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitRepeatedUInt32Field(value: [UInt32], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitRepeatedUInt64Field(value: [UInt64], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitRepeatedSInt32Field(value: [Int32], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitRepeatedSInt64Field(value: [Int64], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitRepeatedFixed32Field(value: [UInt32], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitRepeatedFixed64Field(value: [UInt64], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitRepeatedSFixed32Field(value: [Int32], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitRepeatedSFixed64Field(value: [Int64], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitRepeatedBoolField(value: [Bool], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitRepeatedStringField(value: [String], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitRepeatedBytesField(value: [Data], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitRepeatedEnumField<E: Enum>(value: [E], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitRepeatedGroupField<G: Message>(value: [G], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitPackedFloatField(value: [Float], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitPackedDoubleField(value: [Double], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitPackedInt32Field(value: [Int32], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitPackedInt64Field(value: [Int64], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitPackedUInt32Field(value: [UInt32], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitPackedUInt64Field(value: [UInt64], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitPackedSInt32Field(value: [Int32], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitPackedSInt64Field(value: [Int64], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitPackedFixed32Field(value: [UInt32], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitPackedFixed64Field(value: [UInt64], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitPackedSFixed32Field(value: [Int32], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitPackedSFixed64Field(value: [Int64], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitPackedBoolField(value: [Bool], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitPackedEnumField<E: Enum>(value: [E], fieldNumber: Int) throws {
+    mirror.set(value, for: fieldNumber)
+  }
+
+  mutating func visitRepeatedMessageField<M: Message>(
+    value: [M],
+    fieldNumber: Int
+  ) throws {
     mirror.set(value, for: fieldNumber)
   }
 

Tests/SwiftProtobufTests/Test_Applying.swift

@@ -0,0 +1,50 @@
+// Tests/SwiftProtobufTests/Test_Applying.swift - Applying method
+//
+// Copyright (c) 2014 - 2016 Apple Inc. and the project authors
+// Licensed under Apache License v2.0 with Runtime Library Exception
+//
+// See LICENSE.txt for license information:
+// https://github.com/apple/swift-protobuf/blob/main/LICENSE.txt
+//
+// -----------------------------------------------------------------------------
+
+import Foundation
+import XCTest
+import SwiftProtobuf
+
+final class Test_Applying: XCTestCase {
+
+  func testApplying() throws {
+    let message = SwiftProtoTesting_TestAny()
+    let newMessage = try message.applying(Int32(2), for: 1)
+    XCTAssertEqual(newMessage.int32Value, 2)
+  }
+
+  func testApplyingSequential() throws {
+    let message = SwiftProtoTesting_TestAny()
+    let newMessage = try message
+      .applying(Int32(2), for: 1)
+      .applying("test", for: 4)
+    XCTAssertEqual(newMessage.int32Value, 2)
+    XCTAssertEqual(newMessage.text, "test")
+  }
+
+  func testApplyingMismatchType() throws {
+    let message = SwiftProtoTesting_TestAny()
+    XCTAssertThrowsError(try message.applying("", for: 1))
+  }
+
+  func testApplyingInvalidField() throws {
+    let message = SwiftProtoTesting_TestAny()
+    XCTAssertThrowsError(try message.applying(Int32(2), for: 1024))
+  }
+
+  func testApplyingOneof() throws {
+    let message = SwiftProtoTesting_TestOneof.with { oneof in
+      oneof.foo = .fooInt(1)
+    }
+    let newMessage = try message.applying("oneof", for: 2)
+    XCTAssertEqual(newMessage.foo, .fooString("oneof"))
+  }
+
+}

Tests/SwiftProtobufTests/Test_ProtoMirror.swift

@@ -0,0 +1,41 @@
+// Tests/SwiftProtobufTests/Test_ProtoMirror.swift - Mirroring a message
+//
+// Copyright (c) 2014 - 2016 Apple Inc. and the project authors
+// Licensed under Apache License v2.0 with Runtime Library Exception
+//
+// See LICENSE.txt for license information:
+// https://github.com/apple/swift-protobuf/blob/main/LICENSE.txt
+//
+// -----------------------------------------------------------------------------
+
+import Foundation
+import XCTest
+import SwiftProtobuf
+
+final class Test_ProtoMirror: XCTestCase {
+
+  func testMirroring() throws {
+    let anyMessage = try Google_Protobuf_Any(message: SwiftProtoTesting_TestAny())
+    let message = SwiftProtoTesting_TestAny.with { any in
+      any.int32Value = 1
+      any.anyValue = anyMessage
+      any.repeatedAnyValue = [anyMessage]
+      any.text = "test"
+    }
+    let mirror = try ProtobufMirror(reflecting: message)
+    XCTAssertEqual(mirror.children.count, 4)
+    XCTAssertEqual(mirror.value(for: 1) as? Int32, 1)
+    XCTAssertEqual(mirror.value(for: 2) as? Google_Protobuf_Any, anyMessage)
+    XCTAssertEqual(mirror.value(for: 3) as? [Google_Protobuf_Any], [anyMessage])
+    XCTAssertEqual(mirror.value(for: 4) as? String, "test")
+  }
+
+}
+
+private extension ProtobufMirror {
+  func value(for fieldNumber: Int) -> Any? {
+    children.first { child in
+      child.fieldNumber == fieldNumber
+    }?.value
+  }
+}