Add test and callable reference internal for options

Author: AustinBenoit

functions/integration_test/src/integration_test.cc

@@ -380,4 +380,54 @@ TEST_F(FirebaseFunctionsTest, TestFunctionFromURL) {
   EXPECT_EQ(result.map()["operationResult"], 6);
 }
 
+TEST_F(FirebaseFunctionsTest, TestFunctionWithLimitedUseAppCheckToken) {
+  SignIn();
+
+  // addNumbers(5, 7) = 12
+  firebase::Variant data(firebase::Variant::EmptyMap());
+  data.map()["firstNumber"] = 5;
+  data.map()["secondNumber"] = 7;
+
+  firebase::functions::HttpsCallableOptions options;
+  options.limited_use_app_check_token = true;
+
+  LogDebug("Calling addNumbers with Limited Use App Check Token");
+  firebase::functions::HttpsCallableReference ref =
+      functions_->GetHttpsCallable("addNumbers", options);
+
+  firebase::Variant result =
+      TestFunctionHelper("addNumbers", ref, &data, firebase::Variant::Null())
+          .result()
+          ->data();
+  EXPECT_TRUE(result.is_map());
+  EXPECT_EQ(result.map()["operationResult"], 12);
+}
+
+TEST_F(FirebaseFunctionsTest, TestFunctionFromURLWithLimitedUseAppCheckToken) {
+  SignIn();
+
+  // addNumbers(4, 2) = 6
+  firebase::Variant data(firebase::Variant::EmptyMap());
+  data.map()["firstNumber"] = 4;
+  data.map()["secondNumber"] = 2;
+
+  std::string proj = app_->options().project_id();
+  std::string url =
+      "https://us-central1-" + proj + ".cloudfunctions.net/addNumbers";
+
+  firebase::functions::HttpsCallableOptions options;
+  options.limited_use_app_check_token = true;
+
+  LogDebug("Calling by URL %s with Limited Use App Check Token", url.c_str());
+  firebase::functions::HttpsCallableReference ref =
+      functions_->GetHttpsCallableFromURL(url.c_str(), options);
+
+  firebase::Variant result =
+      TestFunctionHelper(url.c_str(), ref, &data, firebase::Variant::Null())
+          .result()
+          ->data();
+  EXPECT_TRUE(result.is_map());
+  EXPECT_EQ(result.map()["operationResult"], 6);
+}
+
 }  // namespace firebase_testapp_automated

functions/src/desktop/functions_desktop.cc

@@ -34,14 +34,24 @@ const char* FunctionsInternal::region() const { return region_.c_str(); }
 
 HttpsCallableReferenceInternal* FunctionsInternal::GetHttpsCallable(
     const char* name) const {
+  return GetHttpsCallable(name, HttpsCallableOptions());
+}
+
+HttpsCallableReferenceInternal* FunctionsInternal::GetHttpsCallable(
+    const char* name, const HttpsCallableOptions& options) const {
   return new HttpsCallableReferenceInternal(
-      const_cast<FunctionsInternal*>(this), GetUrl(name).c_str());
+      const_cast<FunctionsInternal*>(this), GetUrl(name).c_str(), options);
 }
 
 HttpsCallableReferenceInternal* FunctionsInternal::GetHttpsCallableFromURL(
     const char* url) const {
+  return GetHttpsCallableFromURL(url, HttpsCallableOptions());
+}
+
+HttpsCallableReferenceInternal* FunctionsInternal::GetHttpsCallableFromURL(
+    const char* url, const HttpsCallableOptions& options) const {
   return new HttpsCallableReferenceInternal(
-      const_cast<FunctionsInternal*>(this), url);
+      const_cast<FunctionsInternal*>(this), url, options);
 }
 
 std::string FunctionsInternal::GetUrl(const std::string& name) const {