Add TopLevelEncoder implementation (#175)

CoreOffice · Apr 13, 2020 · 9eb98ba · 9eb98ba
1 parent d57b8f2
commit 9eb98ba
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 4 deletions.
diff --git a/Sources/XMLCoder/Decoder/XMLDecoder.swift b/Sources/XMLCoder/Decoder/XMLDecoder.swift
@@ -374,10 +374,18 @@ open class XMLDecoder {
 
 #if canImport(Combine)
 import protocol Combine.TopLevelDecoder
+import protocol Combine.TopLevelEncoder
 #elseif canImport(OpenCombine)
 import protocol OpenCombine.TopLevelDecoder
+import protocol OpenCombine.TopLevelEncoder
 #endif
 
 #if canImport(Combine) || canImport(OpenCombine)
 extension XMLDecoder: TopLevelDecoder {}
+
+extension XMLEncoder: TopLevelEncoder {
+    public func encode<T>(_ value: T) throws -> Data where T: Encodable {
+        try encode(value, withRootKey: nil, rootAttributes: nil, header: nil)
+    }
+}
 #endif
diff --git a/Tests/XMLCoderTests/CombineTests.swift b/Tests/XMLCoderTests/CombineTests.swift
@@ -23,20 +23,33 @@ private let xml = """
 </foo>
 """.data(using: .utf8)!
 
-private struct Foo: Decodable {
+private struct Foo: Codable {
     var name: String
 }
 
 @available(iOS 13.0, macOS 10.15.0, tvOS 13.0, watchOS 6.0, *)
 final class CombineTests: XCTestCase {
-    func testDecodeFromXMLDecoder() {
-        let data = Just(xml)
+    func testDecode() {
         var foo: Foo?
-        _ = data.decode(type: Foo.self, decoder: XMLDecoder()).sink(
+        _ = Just(xml).decode(type: Foo.self, decoder: XMLDecoder()).sink(
             receiveCompletion: { _ in },
             receiveValue: { foo = $0 }
         )
         XCTAssertEqual(foo?.name, "Foo")
     }
+
+    func testEncode() {
+        var foo: Foo?
+        _ = Just(Foo(name: "Foo"))
+            .encode(encoder: XMLEncoder())
+            .decode(type: Foo.self, decoder: XMLDecoder())
+            .sink(
+                receiveCompletion: { _ in },
+                receiveValue: {
+                    foo = $0
+                }
+            )
+        XCTAssertEqual(foo?.name, "Foo")
+    }
 }
 #endif
diff --git a/Tests/XMLCoderTests/XCTestManifests.swift b/Tests/XMLCoderTests/XCTestManifests.swift
@@ -621,6 +621,15 @@ extension RootLevelAttributeTest {
     ]
 }
 
+extension RootLevetExtraAttributesTests {
+    // DO NOT MODIFY: This is autogenerated, use:
+    //   `swift test --generate-linuxmain`
+    // to regenerate.
+    static let __allTests__RootLevetExtraAttributesTests = [
+        ("testExtraAttributes", testExtraAttributes),
+    ]
+}
+
 extension SharedBoxTests {
     // DO NOT MODIFY: This is autogenerated, use:
     //   `swift test --generate-linuxmain`
@@ -893,6 +902,7 @@ public func __allTests() -> [XCTestCaseEntry] {
         testCase(RJITest.__allTests__RJITest),
         testCase(RelationshipsTest.__allTests__RelationshipsTest),
         testCase(RootLevelAttributeTest.__allTests__RootLevelAttributeTest),
+        testCase(RootLevetExtraAttributesTests.__allTests__RootLevetExtraAttributesTests),
         testCase(SharedBoxTests.__allTests__SharedBoxTests),
         testCase(SimpleChoiceTests.__allTests__SimpleChoiceTests),
         testCase(SingleChildTest.__allTests__SingleChildTest),