Array Contains Queries (not added to public headers yet). (#1138)

2 files changed, 3 insertions, 1 deletions

diff --git a/Firestore/Example/Tests/Util/FSTHelpers.h b/Firestore/Example/Tests/Util/FSTHelpers.h
index 9b5f96a..131da2d 100644
--- a/Firestore/Example/Tests/Util/FSTHelpers.h
+++ b/Firestore/Example/Tests/Util/FSTHelpers.h
@@ -212,7 +212,7 @@ FSTQuery *FSTTestQuery(const absl::string_view path);
 
 /**
  * A convenience method to create a FSTFilter using a string representation for both field
- * and operator (<, <=, ==, >=, >).
+ * and operator (<, <=, ==, >=, >, array_contains).
  */
 id<FSTFilter> FSTTestFilter(const absl::string_view field, NSString *op, id value);
 
diff --git a/Firestore/Example/Tests/Util/FSTHelpers.mm b/Firestore/Example/Tests/Util/FSTHelpers.mm
index e1ed18c..888a45f 100644
--- a/Firestore/Example/Tests/Util/FSTHelpers.mm
+++ b/Firestore/Example/Tests/Util/FSTHelpers.mm
@@ -210,6 +210,8 @@ id<FSTFilter> FSTTestFilter(const absl::string_view field, NSString *opString, i
     op = FSTRelationFilterOperatorGreaterThanOrEqual;
   } else if ([opString isEqualToString:@">"]) {
     op = FSTRelationFilterOperatorGreaterThan;
+  } else if ([opString isEqualToString:@"array_contains"]) {
+    op = FSTRelationFilterOperatorArrayContains;
   } else {
     FSTCFail(@"Unsupported operator type: %@", opString);
   }