1 files changed, 9 insertions, 5 deletions
diff --git a/tests/dns_responder/dns_responder_client_ndk.h b/tests/dns_responder/dns_responder_client_ndk.h
index 1b4ba35d..0713a7ca 100644
--- a/tests/dns_responder/dns_responder_client_ndk.h
+++ b/tests/dns_responder/dns_responder_client_ndk.h
@@ -91,6 +91,15 @@ class ResolverParams {
             mParcel.retryCount = params[IDnsResolver::RESOLVER_PARAMS_RETRY_COUNT];
             return *this;
         }
+        constexpr Builder& setMetered(const bool metered) {
+            mParcel.meteredNetwork = metered;
+            return *this;
+        }
+        constexpr Builder& setDohParams(
+                const aidl::android::net::resolv::aidl::DohParamsParcel& dohParams) {
+            mParcel.dohParams = dohParams;
+            return *this;
+        }
         aidl::android::net::ResolverParamsParcel build() { return mParcel; }
 
       private:
@@ -114,11 +123,6 @@ class DnsResponderClient {
     static void SetupMappings(unsigned num_hosts, const std::vector<std::string>& domains,
                               std::vector<Mapping>* mappings);
 
-    // For dns_benchmark built from tm-mainline-prod.
-    // TODO: Remove it when possible.
-    bool SetResolversForNetwork(const std::vector<std::string>& servers,
-                                const std::vector<std::string>& domains, std::vector<int> params);
-
     // Sets up DnsResolver with given DNS servers. This is used to set up for private DNS off mode.
     bool SetResolversForNetwork(const std::vector<std::string>& servers = {kDefaultServer},
                                 const std::vector<std::string>& domains = {kDefaultSearchDomain});