add tests

Author: fumito-ito

Tests/FunctionCalling-FirebaseVertexAITests/Extensions.swift

@@ -0,0 +1,24 @@
+//
+//  Extensions.swift
+//  FunctionCalling-FirebaseVertexAI
+//
+//  Created by 伊藤史 on 2024/09/20.
+//
+
+@testable import FirebaseVertexAI
+
+extension FirebaseVertexAI.Tool {
+    var functions: [FirebaseVertexAI.FunctionDeclaration]? {
+        functionDeclarations
+    }
+}
+
+extension FirebaseVertexAI.FunctionDeclaration {
+    func getName() -> String {
+        name
+    }
+
+    func getDescription() -> String {
+        description
+    }
+}

Tests/FunctionCalling-FirebaseVertexAITests/FunctionCalling_FirebaseVertexAITests.swift

@@ -1,12 +1,38 @@
 import XCTest
 @testable import FunctionCalling_FirebaseVertexAI
+import FunctionCalling
 
 final class FunctionCalling_FirebaseVertexAITests: XCTestCase {
-    func testExample() throws {
-        // XCTest Documentation
-        // https://developer.apple.com/documentation/xctest
+    @FunctionCalling(service: .claude)
+        struct FunctionContainer {
+            /// Return current weather of location that passed by the argument
+            /// - Parameter location: location that I want to know how the weather
+            /// - Returns: string of weather
+            @CallableFunction
+            func getWeather(location: String) -> String {
+                return "Sunny"
+            }
 
-        // Defining Test Cases and Test Methods
-        // https://developer.apple.com/documentation/xctest/defining_test_cases_and_test_methods
-    }
+            @CallableFunction
+            func getStock(args: String) -> Int {
+                return 0
+            }
+        }
+
+        func testConvertedResults() throws {
+            guard let functions = FunctionContainer().firebaseVertexAITools.first?.functions else {
+                XCTFail("Conainer should contain some functions")
+                return
+            }
+
+            XCTAssertEqual(functions.count, 2)
+
+            let getWeather = try XCTUnwrap(functions.first)
+            XCTAssertEqual(getWeather.getName(), "getWeather")
+            XCTAssertEqual(getWeather.getDescription(), "Return current weather of location that passed by the argument- Parameter location: location that I want to know how the weather- Returns: string of weather")
+
+            let getStock = try XCTUnwrap(functions.last)
+            XCTAssertEqual(getStock.getName(), "getStock")
+            XCTAssertEqual(getStock.getDescription(), "")
+        }
 }