Fix CI tests

Author: cherylEnkidu

Firestore/Example/Firestore.xcodeproj/xcshareddata/xcschemes/Firestore_Example_iOS.xcscheme

@@ -89,6 +89,18 @@
             ReferencedContainer = "container:Firestore.xcodeproj">
          </BuildableReference>
       </BuildableProductRunnable>
+      <EnvironmentVariables>
+         <EnvironmentVariable
+            key = "PROJECT_ID"
+            value = "firestore-sdk-nightly"
+            isEnabled = "YES">
+         </EnvironmentVariable>
+         <EnvironmentVariable
+            key = "TARGET_BACKEND"
+            value = "nightly"
+            isEnabled = "YES">
+         </EnvironmentVariable>
+      </EnvironmentVariables>
    </LaunchAction>
    <ProfileAction
       buildConfiguration = "Release"

Firestore/Example/Tests/Util/FSTIntegrationTestCase.h

@@ -51,6 +51,8 @@ extern "C" {
 /** Returns the default Firestore database ID for testing. */
 + (NSString *)databaseID;
 
++ (void)switchToEnterpriseMode;
+
 + (bool)isRunningAgainstEmulator;
 
 /** Returns a FirestoreSettings configured to use either hexa or the emulator. */

Firestore/Example/Tests/Util/FSTIntegrationTestCase.mm

@@ -86,6 +86,7 @@
 
 static NSString *defaultProjectId;
 static NSString *defaultDatabaseId = @"(default)";
+static NSString *enterpriseDatabaseId = @"enterprise";
 static FIRFirestoreSettings *defaultSettings;
 
 static bool runningAgainstEmulator = false;
@@ -273,6 +274,10 @@ + (NSString *)databaseID {
   return defaultDatabaseId;
 }
 
++ (void)switchToEnterpriseMode {
+  defaultDatabaseId = enterpriseDatabaseId;
+}
+
 + (bool)isRunningAgainstEmulator {
   // The only way to determine whether or not we're running against the emulator is to figure out
   // which testing environment we're using.  Essentially `setUpDefaults` determines

Firestore/Swift/Source/SwiftAPI/Pipeline/PipelineSource.swift

@@ -37,12 +37,13 @@ public struct PipelineSource: @unchecked Sendable {
   }
 
   public func documents(_ docs: [DocumentReference]) -> Pipeline {
-    let paths = docs.map { $0.path }
+    let paths = docs.map { $0.path.hasPrefix("/") ? $0.path : "/" + $0.path }
     return Pipeline(stages: [DocumentsSource(paths: paths)], db: db)
   }
 
   public func documents(_ paths: [String]) -> Pipeline {
-    return Pipeline(stages: [DocumentsSource(paths: paths)], db: db)
+    let normalizedPaths = paths.map { $0.hasPrefix("/") ? $0 : "/" + $0 }
+    return Pipeline(stages: [DocumentsSource(paths: normalizedPaths)], db: db)
   }
 
   public func create(from query: Query) -> Pipeline {

Firestore/Swift/Tests/Integration/PipelineApiTests.swift

@@ -18,6 +18,11 @@ import XCTest
 import FirebaseFirestore
 
 final class PipelineTests: FSTIntegrationTestCase {
+  override func setUp() {
+    FSTIntegrationTestCase.switchToEnterpriseMode()
+    super.setUp()
+  }
+
   func testCreatePipeline() async throws {
     let pipelineSource: PipelineSource = db.pipeline()
 

Firestore/Swift/Tests/Integration/PipelineTests.swift

@@ -113,6 +113,11 @@ private let bookDocs: [String: [String: Any]] = [
 
 @available(iOS 13, tvOS 13, macOS 10.15, macCatalyst 13, watchOS 7, *)
 class PipelineIntegrationTests: FSTIntegrationTestCase {
+  override func setUp() {
+    FSTIntegrationTestCase.switchToEnterpriseMode()
+    super.setUp()
+  }
+
   func testCount() async throws {
     try await firestore().collection("foo").document("bar").setData(["foo": "bar", "x": 42])
     let snapshot = try await firestore()

Firestore/core/src/api/stages.cc

@@ -89,8 +89,8 @@ google_firestore_v1_Pipeline_Stage DocumentsSource::to_proto() const {
 
   for (size_t i = 0; i < documents_.size(); ++i) {
     result.args[i].which_value_type =
-        google_firestore_v1_Value_string_value_tag;
-    result.args[i].string_value = nanopb::MakeBytesArray(documents_[i]);
+        google_firestore_v1_Value_reference_value_tag;
+    result.args[i].reference_value = nanopb::MakeBytesArray(documents_[i]);
   }
 
   result.options_count = 0;