Add decoding tests for missing fields in ExecutableCode and CodeExecutionResult

Author: andrewheard

FirebaseAI/Tests/Unit/PartTests.swift

@@ -103,6 +103,52 @@ final class PartTests: XCTestCase {
     XCTAssertEqual(part.code, "print('hello')")
   }
 
+  func testDecodeExecutableCodePart_missingLanguage() throws {
+    let json = """
+    {
+      "executableCode": {
+        "code": "print('hello')"
+      }
+    }
+    """
+    let jsonData = try XCTUnwrap(json.data(using: .utf8))
+
+    let part = try decoder.decode(ExecutableCodePart.self, from: jsonData)
+
+    XCTAssertEqual(part.language.description, "LANGUAGE_UNSPECIFIED")
+    XCTAssertEqual(part.code, "print('hello')")
+  }
+
+  func testDecodeExecutableCodePart_missingCode() throws {
+    let json = """
+    {
+      "executableCode": {
+        "language": "PYTHON"
+      }
+    }
+    """
+    let jsonData = try XCTUnwrap(json.data(using: .utf8))
+
+    let part = try decoder.decode(ExecutableCodePart.self, from: jsonData)
+
+    XCTAssertEqual(part.language, .python)
+    XCTAssertEqual(part.code, "")
+  }
+
+  func testDecodeExecutableCodePart_missingLanguageAndCode() throws {
+    let json = """
+    {
+      "executableCode": {}
+    }
+    """
+    let jsonData = try XCTUnwrap(json.data(using: .utf8))
+
+    let part = try decoder.decode(ExecutableCodePart.self, from: jsonData)
+
+    XCTAssertEqual(part.language.description, "LANGUAGE_UNSPECIFIED")
+    XCTAssertEqual(part.code, "")
+  }
+
   func testDecodeCodeExecutionResultPart() throws {
     let json = """
     {
@@ -120,6 +166,52 @@ final class PartTests: XCTestCase {
     XCTAssertEqual(part.output, "hello")
   }
 
+  func testDecodeCodeExecutionResultPart_missingOutcome() throws {
+    let json = """
+    {
+      "codeExecutionResult": {
+        "output": "hello"
+      }
+    }
+    """
+    let jsonData = try XCTUnwrap(json.data(using: .utf8))
+
+    let part = try decoder.decode(CodeExecutionResultPart.self, from: jsonData)
+
+    XCTAssertEqual(part.outcome.description, "OUTCOME_UNSPECIFIED")
+    XCTAssertEqual(part.output, "hello")
+  }
+
+  func testDecodeCodeExecutionResultPart_missingOutput() throws {
+    let json = """
+    {
+      "codeExecutionResult": {
+        "outcome": "OUTCOME_OK"
+      }
+    }
+    """
+    let jsonData = try XCTUnwrap(json.data(using: .utf8))
+
+    let part = try decoder.decode(CodeExecutionResultPart.self, from: jsonData)
+
+    XCTAssertEqual(part.outcome, .ok)
+    XCTAssertNil(part.output)
+  }
+
+  func testDecodeCodeExecutionResultPart_missingOutcomeAndOutput() throws {
+    let json = """
+    {
+      "codeExecutionResult": {}
+    }
+    """
+    let jsonData = try XCTUnwrap(json.data(using: .utf8))
+
+    let part = try decoder.decode(CodeExecutionResultPart.self, from: jsonData)
+
+    XCTAssertEqual(part.outcome.description, "OUTCOME_UNSPECIFIED")
+    XCTAssertNil(part.output)
+  }
+
   // MARK: - Part Encoding
 
   func testEncodeTextPart() throws {

FirebaseAI/Tests/Unit/Types/InternalPartTests.swift

@@ -306,6 +306,67 @@ final class InternalPartTests: XCTestCase {
     XCTAssertEqual(executableCode.code, "print('hello')")
   }
 
+  func testDecodeExecutableCodePart_missingLanguage() throws {
+    let json = """
+    {
+      "executableCode": {
+        "code": "print('hello')"
+      }
+    }
+    """
+    let jsonData = try XCTUnwrap(json.data(using: .utf8))
+
+    let part = try decoder.decode(InternalPart.self, from: jsonData)
+
+    XCTAssertNil(part.isThought)
+    guard case let .executableCode(executableCode) = part.data else {
+      XCTFail("Decoded part is not an executableCode part.")
+      return
+    }
+    XCTAssertNil(executableCode.language)
+    XCTAssertEqual(executableCode.code, "print('hello')")
+  }
+
+  func testDecodeExecutableCodePart_missingCode() throws {
+    let json = """
+    {
+      "executableCode": {
+        "language": "PYTHON"
+      }
+    }
+    """
+    let jsonData = try XCTUnwrap(json.data(using: .utf8))
+
+    let part = try decoder.decode(InternalPart.self, from: jsonData)
+
+    XCTAssertNil(part.isThought)
+    guard case let .executableCode(executableCode) = part.data else {
+      XCTFail("Decoded part is not an executableCode part.")
+      return
+    }
+    XCTAssertEqual(executableCode.language, .init(kind: .python))
+    XCTAssertNil(executableCode.code)
+  }
+
+  func testDecodeExecutableCodePart_missingLanguageAndCode() throws {
+    let json = """
+    {
+      "executableCode": {}
+    }
+    """
+    let jsonData = try XCTUnwrap(json.data(using: .utf8))
+
+    let part = try decoder.decode(InternalPart.self, from: jsonData)
+
+    XCTAssertNil(part.isThought)
+    guard case let .executableCode(executableCode) = part.data else {
+      XCTFail("Decoded part is not an executableCode part.")
+      return
+    }
+    XCTAssertNil(executableCode.language)
+    XCTAssertNil(executableCode.code)
+  }
+
   func testDecodeCodeExecutionResultPart() throws {
     let json = """
     {
@@ -327,4 +388,65 @@ final class InternalPartTests: XCTestCase {
     XCTAssertEqual(codeExecutionResult.outcome, .init(kind: .ok))
     XCTAssertEqual(codeExecutionResult.output, "hello")
   }
+
+  func testDecodeCodeExecutionResultPart_missingOutcome() throws {
+    let json = """
+    {
+      "codeExecutionResult": {
+        "output": "hello"
+      }
+    }
+    """
+    let jsonData = try XCTUnwrap(json.data(using: .utf8))
+
+    let part = try decoder.decode(InternalPart.self, from: jsonData)
+
+    XCTAssertNil(part.isThought)
+    guard case let .codeExecutionResult(codeExecutionResult) = part.data else {
+      XCTFail("Decoded part is not a codeExecutionResult part.")
+      return
+    }
+    XCTAssertNil(codeExecutionResult.outcome)
+    XCTAssertEqual(codeExecutionResult.output, "hello")
+  }
+
+  func testDecodeCodeExecutionResultPart_missingOutput() throws {
+    let json = """
+    {
+      "codeExecutionResult": {
+        "outcome": "OUTCOME_OK"
+      }
+    }
+    """
+    let jsonData = try XCTUnwrap(json.data(using: .utf8))
+
+    let part = try decoder.decode(InternalPart.self, from: jsonData)
+
+    XCTAssertNil(part.isThought)
+    guard case let .codeExecutionResult(codeExecutionResult) = part.data else {
+      XCTFail("Decoded part is not a codeExecutionResult part.")
+      return
+    }
+    XCTAssertEqual(codeExecutionResult.outcome, .init(kind: .ok))
+    XCTAssertNil(codeExecutionResult.output)
+  }
+
+  func testDecodeCodeExecutionResultPart_missingOutcomeAndOutput() throws {
+    let json = """
+    {
+      "codeExecutionResult": {}
+    }
+    """
+    let jsonData = try XCTUnwrap(json.data(using: .utf8))
+
+    let part = try decoder.decode(InternalPart.self, from: jsonData)
+
+    XCTAssertNil(part.isThought)
+    guard case let .codeExecutionResult(codeExecutionResult) = part.data else {
+      XCTFail("Decoded part is not a codeExecutionResult part.")
+      return
+    }
+    XCTAssertNil(codeExecutionResult.outcome)
+    XCTAssertNil(codeExecutionResult.output)
+  }
 }