Fixed an issue where decoded CodableContextKey values became inaccesi…

…ble after the first access (#7)
Apodini · Feb 7, 2022 · f655813 · f655813
1 parent fe22deb
commit f655813
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/Sources/ApodiniContext/Context.swift b/Sources/ApodiniContext/Context.swift
@@ -121,7 +121,9 @@ public struct Context: ContextKeyRetrievable {
         }
 
         do {
-            return try Key.decode(from: dataValue)
+            let value = try Key.decode(from: dataValue)
+            boxedEntries.entries[ObjectIdentifier(key)] = StoredContextValue(key: key, value: value)
+            return value
         } catch {
             fatalError("Error occurred when trying to decode `CodableContextKey` `\(Key.self)` with stored value '\(dataValue)': \(error)")
         }

diff --git a/Tests/ApodiniContextTests/ContextKeyTests.swift b/Tests/ApodiniContextTests/ContextKeyTests.swift
@@ -141,6 +141,8 @@ class ContextKeyTests: XCTestCase {
         )
 
         let decodedContext = try decoder.decode(Context.self, from: encodedContext)
+        XCTAssertEqual(decodedContext.get(valueFor: CodableStringContextKey.self), "Hello Mars")
+        XCTAssertEqual(decodedContext.get(valueFor: CodableStringContextKey.self), "Hello Mars")
 
         decodedContext.unsafeAdd(CodableStringContextKey.self, value: "Hello Saturn", allowOverwrite: true)
         XCTAssertEqual(decodedContext.get(valueFor: CodableStringContextKey.self), "Hello Saturn")