From 62d5b82e9ba3e947d97893b15743fe1642ecae41 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Fri, 25 Oct 2024 16:51:21 -0400
Subject: [PATCH] Re-check FIRAAppCheckDebugToken

---
 .../TestApp/Tests/Integration/IntegrationTests.swift | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/FirebaseVertexAI/Tests/TestApp/Tests/Integration/IntegrationTests.swift b/FirebaseVertexAI/Tests/TestApp/Tests/Integration/IntegrationTests.swift
index b90a13515a6..bb971f1fbec 100644
--- a/FirebaseVertexAI/Tests/TestApp/Tests/Integration/IntegrationTests.swift
+++ b/FirebaseVertexAI/Tests/TestApp/Tests/Integration/IntegrationTests.swift
@@ -64,6 +64,18 @@ final class IntegrationTests: XCTestCase {
     storage = Storage.storage()
   }
 
+  func testDebugTokenProvided() {
+    if ProcessInfo.processInfo.environment["FIRAAppCheckDebugToken"] == nil {
+      XCTFail("No App Check debug token provided in FIRAAppCheckDebugToken.")
+    }
+  }
+
+  func testDebugTokenNotProvided() {
+    if ProcessInfo.processInfo.environment["FIRAAppCheckDebugToken"] != nil {
+      XCTFail("Token not provided; intentionally failing for testing.")
+    }
+  }
+
   // MARK: - Generate Content
 
   func testGenerateContent() async throws {