Added Codable conformance and unit tests

orchetect · Jul 24, 2022 · 1b1f3ea · 1b1f3ea
1 parent 32f5eae
commit 1b1f3ea
Show file tree

Hide file tree

Showing 4 changed files with 85 additions and 0 deletions.
diff --git a/Sources/SwiftASCII/ASCIICharacter.swift b/Sources/SwiftASCII/ASCIICharacter.swift
@@ -186,6 +186,38 @@ extension ASCIICharacter: Equatable {
 
 }
 
+extension ASCIICharacter: Codable {
+
+    enum CodingKeys: String, CodingKey {
+
+        case asciiValue
+
+    }
+
+    public init(from decoder: Decoder) throws {
+
+        let container = try decoder.container(keyedBy: CodingKeys.self)
+        let asciiValue = try container.decode(UInt8.self, forKey: .asciiValue)
+        guard let newInstance = Self(asciiValue) else {
+            throw DecodingError.dataCorrupted(
+                .init(codingPath: container.codingPath,
+                      debugDescription: "Value was not valid ASCII character number.")
+            )
+        }
+        self = newInstance
+
+    }
+
+    public func encode(to encoder: Encoder) throws {
+
+        var container = encoder.container(keyedBy: CodingKeys.self)
+
+        try container.encode(asciiValue, forKey: .asciiValue)
+
+    }
+
+}
+
 extension ASCIICharacter {
 
     public static func + (lhs: ASCIICharacter, rhs: ASCIICharacter) -> ASCIIString {

diff --git a/Sources/SwiftASCII/ASCIIString.swift b/Sources/SwiftASCII/ASCIIString.swift
@@ -147,6 +147,31 @@ extension ASCIIString: Equatable {
 
 }
 
+extension ASCIIString: Codable {
+
+    public init(from decoder: Decoder) throws {
+
+        let container = try decoder.singleValueContainer()
+        let stringValue = try container.decode(String.self)
+        guard let newInstance = Self(exactly: stringValue) else {
+            throw DecodingError.dataCorrupted(
+                .init(codingPath: container.codingPath,
+                      debugDescription: "Encoded string is not a valid ASCII string.")
+            )
+        }
+        self = newInstance
+
+    }
+
+    public func encode(to encoder: Encoder) throws {
+
+        var container = encoder.singleValueContainer()
+        try container.encode(stringValue)
+
+    }
+
+}
+
 extension ASCIIString {
 
     public static func + (lhs: ASCIIString, rhs: ASCIIString) -> ASCIIString {

diff --git a/Tests/SwiftASCIITests/ASCIICharacter Tests.swift b/Tests/SwiftASCIITests/ASCIICharacter Tests.swift
@@ -129,6 +129,20 @@ class ASCIICharacterTests: XCTestCase {
 
     }
 
+    func testCodable() throws {
+
+        let encoder = JSONEncoder()
+        let decoder = JSONDecoder()
+
+        let str = ASCIICharacter("A")
+
+        let encoded = try encoder.encode(str)
+        let decoded = try decoder.decode(ASCIICharacter.self, from: encoded)
+
+        XCTAssertEqual(str, decoded)
+
+    }
+
     func testStaticInits() {
 
         let str: ASCIICharacter = .lossy("😃")

diff --git a/Tests/SwiftASCIITests/ASCIIString Tests.swift b/Tests/SwiftASCIITests/ASCIIString Tests.swift
@@ -96,6 +96,20 @@ class ASCIIStringTests: XCTestCase {
 
     }
 
+    func testCodable() throws {
+
+        let encoder = JSONEncoder()
+        let decoder = JSONDecoder()
+
+        let str = ASCIIString("A string")
+
+        let encoded = try encoder.encode(str)
+        let decoded = try decoder.decode(ASCIIString.self, from: encoded)
+
+        XCTAssertEqual(str, decoded)
+
+    }
+
     func testStaticInits() {
 
         let str: ASCIIString = .lossy("Emöji 😃")