From 20eeed98e4fa1248f372fc9c2c60d3c23cff8c6e Mon Sep 17 00:00:00 2001
From: Sung-Hyun Kang <sunghyunkang@microsoft.com>
Date: Mon, 3 Feb 2025 11:23:49 -0600
Subject: [PATCH] fix unit tests

---
 .../__snapshots__/queryDocuments.test.ts.snap |  6 +++++
 .../arm/generatedClients/cosmos/types.ts      | 24 +++++++++++++++++++
 2 files changed, 30 insertions(+)

diff --git a/src/Common/dataAccess/__snapshots__/queryDocuments.test.ts.snap b/src/Common/dataAccess/__snapshots__/queryDocuments.test.ts.snap
index bcf0034a4..79da51170 100644
--- a/src/Common/dataAccess/__snapshots__/queryDocuments.test.ts.snap
+++ b/src/Common/dataAccess/__snapshots__/queryDocuments.test.ts.snap
@@ -5,6 +5,9 @@ exports[`getCommonQueryOptions builds the correct default options objects 1`] =
   "disableNonStreamingOrderByQuery": true,
   "enableScanInQuery": true,
   "forceQueryPlan": true,
+  "initialHeaders": {
+    "x-ms-cosmos-throughput-bucket": "1",
+  },
   "maxDegreeOfParallelism": 0,
   "maxItemCount": 100,
   "populateQueryMetrics": true,
@@ -16,6 +19,9 @@ exports[`getCommonQueryOptions reads from localStorage 1`] = `
   "disableNonStreamingOrderByQuery": true,
   "enableScanInQuery": true,
   "forceQueryPlan": true,
+  "initialHeaders": {
+    "x-ms-cosmos-throughput-bucket": "1",
+  },
   "maxDegreeOfParallelism": 17,
   "maxItemCount": 37,
   "populateQueryMetrics": true,
diff --git a/src/Utils/arm/generatedClients/cosmos/types.ts b/src/Utils/arm/generatedClients/cosmos/types.ts
index b0c61a493..6f64ebd7a 100644
--- a/src/Utils/arm/generatedClients/cosmos/types.ts
+++ b/src/Utils/arm/generatedClients/cosmos/types.ts
@@ -1290,6 +1290,8 @@ export interface SqlContainerResource {
 
   /* The vector embedding policy for the container. */
   vectorEmbeddingPolicy?: VectorEmbeddingPolicy;
+
+  fullTextPolicy?: FullTextPolicy;
 }
 
 /* Cosmos DB indexing policy */
@@ -1368,6 +1370,28 @@ export interface VectorEmbedding {
   dimensions: number;
 }
 
+export interface FullTextPolicy {
+  /**
+   * The default language for the full text .
+   */
+  defaultLanguage: string;
+  /**
+   * The paths to be indexed for full text search.
+   */
+  fullTextPaths: FullTextPath[];
+}
+
+export interface FullTextPath {
+  /**
+   * The path to be indexed for full text search.
+   */
+  path: string;
+  /**
+   * The language for the full text path.
+   */
+  language: string;
+}
+
 /* List of composite path */
 export type CompositePathList = CompositePath[];