forked from CoreOffice/XMLCoder
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add nested choice tests (CoreOffice#18)
- Loading branch information
Showing
2 changed files
with
254 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,250 @@ | ||
| // | ||
| // NestedChoiceTests.swift | ||
| // XMLCoderTests | ||
| // | ||
| // Created by James Bean on 7/15/19. | ||
| // | ||
|
|
||
| import XCTest | ||
| import XMLCoder | ||
|
|
||
| private struct Container: Equatable { | ||
| let paragraphs: [Paragraph] | ||
| } | ||
|
|
||
| private struct Paragraph: Equatable { | ||
| let entries: [Entry] | ||
| } | ||
|
|
||
| private enum Entry: Equatable { | ||
| case run(Run) | ||
| case properties(Properties) | ||
| case br(Break) | ||
| } | ||
|
|
||
| private struct Run: Codable, Equatable { | ||
| let id: Int | ||
| let text: String | ||
| } | ||
|
|
||
| private struct Properties: Codable, Equatable { | ||
| let id: Int | ||
| let title: String | ||
| } | ||
|
|
||
| private struct Break: Codable, Equatable { } | ||
|
|
||
| extension Container: Codable { | ||
| enum CodingKeys: String, CodingKey { | ||
| case paragraphs = "p" | ||
| } | ||
| } | ||
|
|
||
| extension Paragraph: Codable { | ||
| init(from decoder: Decoder) throws { | ||
| let container = try decoder.singleValueContainer() | ||
| self.entries = try container.decode([Entry].self) | ||
| } | ||
|
|
||
| func encode(to encoder: Encoder) throws { | ||
| var container = encoder.singleValueContainer() | ||
| try container.encode(entries) | ||
| } | ||
| } | ||
|
|
||
| extension Entry: XMLChoiceCodable { | ||
| private enum CodingKeys: String, CodingKey { | ||
| case run, properties, br | ||
| } | ||
| public init(from decoder: Decoder) throws { | ||
| let container = try decoder.container(keyedBy: CodingKeys.self) | ||
| do { | ||
| self = .run(try container.decode(Run.self, forKey: .run)) | ||
| } catch DecodingError.keyNotFound { | ||
| do { | ||
| self = .properties(try container.decode(Properties.self, forKey: .properties)) | ||
| } catch DecodingError.keyNotFound { | ||
| self = .br(try container.decode(Break.self, forKey: .br)) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| func encode(to encoder: Encoder) throws { | ||
| var container = encoder.container(keyedBy: CodingKeys.self) | ||
| switch self { | ||
| case let .run(value): | ||
| try container.encode(value, forKey: .run) | ||
| case let .properties(value): | ||
| try container.encode(value, forKey: .properties) | ||
| case let .br(value): | ||
| try container.encode(value, forKey: .br) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| class NestedChoiceTests: XCTestCase { | ||
|
|
||
| func testBreakDecoding() throws { | ||
| let xml = "<br></br>" | ||
| let result = try XMLDecoder().decode(Break.self, from: xml.data(using: .utf8)!) | ||
| let expected = Break() | ||
| XCTAssertEqual(result, expected) | ||
| } | ||
|
|
||
| func testPropertiesDecoding() throws { | ||
| let xml = """ | ||
| <properties> | ||
| <id>431</id> | ||
| <title>A Word About Wake Times</title> | ||
| </properties> | ||
| """ | ||
| let result = try XMLDecoder().decode(Properties.self, from: xml.data(using: .utf8)!) | ||
| let expected = Properties(id: 431, title: "A Word About Wake Times") | ||
| XCTAssertEqual(result, expected) | ||
| } | ||
|
|
||
| func testPropertiesAsEntryDecoding() throws { | ||
| let xml = """ | ||
| <entry> | ||
| <properties> | ||
| <id>431</id> | ||
| <title>A Word About Wake Times</title> | ||
| </properties> | ||
| </entry> | ||
| """ | ||
| let result = try XMLDecoder().decode(Entry.self, from: xml.data(using: .utf8)!) | ||
| let expected: Entry = .properties(Properties(id: 431, title: "A Word About Wake Times")) | ||
| XCTAssertEqual(result, expected) | ||
| } | ||
|
|
||
| func testRunDecoding() throws { | ||
| let xml = """ | ||
| <run> | ||
| <id>1518</id> | ||
| <text>I am answering it again.</text> | ||
| </run> | ||
| """ | ||
| let result = try XMLDecoder().decode(Run.self, from: xml.data(using: .utf8)!) | ||
| let expected = Run(id: 1518, text: "I am answering it again.") | ||
| XCTAssertEqual(result, expected) | ||
| } | ||
|
|
||
| func testRunAsEntryDecoding() throws { | ||
| let xml = """ | ||
| <entry> | ||
| <run> | ||
| <id>1518</id> | ||
| <text>I am answering it again.</text> | ||
| </run> | ||
| </entry> | ||
| """ | ||
| let result = try XMLDecoder().decode(Entry.self, from: xml.data(using: .utf8)!) | ||
| let expected = Entry.run(Run(id: 1518, text: "I am answering it again.")) | ||
| XCTAssertEqual(result, expected) | ||
| } | ||
|
|
||
| func testEntriesDecoding() throws { | ||
| let xml = """ | ||
| <entries> | ||
| <run> | ||
| <id>1518</id> | ||
| <text>I am answering it again.</text> | ||
| </run> | ||
| <properties> | ||
| <id>431</id> | ||
| <title>A Word About Wake Times</title> | ||
| </properties> | ||
| </entries> | ||
| """ | ||
| let result = try XMLDecoder().decode([Entry].self, from: xml.data(using: .utf8)!) | ||
| let expected: [Entry] = [ | ||
| .run(Run(id: 1518, text: "I am answering it again.")), | ||
| .properties(Properties(id: 431, title: "A Word About Wake Times")) | ||
| ] | ||
| XCTAssertEqual(result, expected) | ||
| } | ||
|
|
||
| func testParagraphDecoding() throws { | ||
| let xml = """ | ||
| <p> | ||
| <run> | ||
| <id>1518</id> | ||
| <text>I am answering it again.</text> | ||
| </run> | ||
| <properties> | ||
| <id>431</id> | ||
| <title>A Word About Wake Times</title> | ||
| </properties> | ||
| </p> | ||
| """ | ||
| let result = try XMLDecoder().decode(Paragraph.self, from: xml.data(using: .utf8)!) | ||
| let expected = Paragraph( | ||
| entries: [ | ||
| .run(Run(id: 1518, text: "I am answering it again.")), | ||
| .properties(Properties(id: 431, title: "A Word About Wake Times")) | ||
| ] | ||
| ) | ||
| XCTAssertEqual(result, expected) | ||
| } | ||
|
|
||
| func testNestedEnums() throws { | ||
| let xml = """ | ||
| <container> | ||
| <p> | ||
| <run> | ||
| <id>1518</id> | ||
| <text>I am answering it again.</text> | ||
| </run> | ||
| <properties> | ||
| <id>431</id> | ||
| <title>A Word About Wake Times</title> | ||
| </properties> | ||
| </p> | ||
| <p> | ||
| <run> | ||
| <id>1519</id> | ||
| <text>I am answering it again.</text> | ||
| </run> | ||
| </p> | ||
| </container> | ||
| """ | ||
| let result = try XMLDecoder().decode(Container.self, from: xml.data(using: .utf8)!) | ||
| let expected = Container( | ||
| paragraphs: [ | ||
| Paragraph( | ||
| entries: [ | ||
| .run(Run(id: 1518, text: "I am answering it again.")), | ||
| .properties(Properties(id: 431, title: "A Word About Wake Times")), | ||
| ] | ||
| ), | ||
| Paragraph( | ||
| entries: [ | ||
| .run(Run(id: 1519, text: "I am answering it again.")), | ||
| ] | ||
| ) | ||
| ] | ||
| ) | ||
| XCTAssertEqual(result, expected) | ||
| } | ||
|
|
||
| func testNestedEnumsRoundTrip() throws { | ||
| let original = Container( | ||
| paragraphs: [ | ||
| Paragraph( | ||
| entries: [ | ||
| .run(Run(id: 1518, text: "I am answering it again.")), | ||
| .properties(Properties(id: 431, title: "A Word About Wake Times")), | ||
| ] | ||
| ), | ||
| Paragraph( | ||
| entries: [ | ||
| .run(Run(id: 1519, text: "I am answering it again.")), | ||
| ] | ||
| ) | ||
| ] | ||
| ) | ||
| let encoded = try XMLEncoder().encode(original, withRootKey: "container") | ||
| let decoded = try XMLDecoder().decode(Container.self, from: encoded) | ||
| XCTAssertEqual(decoded, original) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters