[iOS] Add +[ChromeEarlGreyAppInterface setIntegerValue:forUserPref:]

This change is needed for crrev.com/c/3059688.

Bug: 1232071
Change-Id: I88152341ce5e02932ee877be892dabd60eeecf4a
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/3063972
Commit-Queue: Jérôme <jlebel@chromium.org>
Reviewed-by: Olivier Robin <olivierrobin@chromium.org>
Cr-Commit-Position: refs/heads/master@{#907981}

ios/chrome/test/app/chrome_test_util.h

@@ -66,6 +66,11 @@ void SetBooleanUserPref(ChromeBrowserState* browser_state,
                         const char* pref_name,
                         bool value);
 
+// Sets the value of an integer user pref in the given browser state.
+void SetIntegerUserPref(ChromeBrowserState* browser_state,
+                        const char* pref_name,
+                        int value);
+
 // Sets the state of using cellular network.
 void SetWWANStateTo(bool value);
 

ios/chrome/test/app/chrome_test_util.mm

@@ -191,6 +191,16 @@ void SetBooleanUserPref(ChromeBrowserState* browser_state,
   pref.SetValue(value);
 }
 
+void SetIntegerUserPref(ChromeBrowserState* browser_state,
+                        const char* pref_name,
+                        int value) {
+  DCHECK(browser_state);
+  DCHECK(browser_state->GetPrefs());
+  IntegerPrefMember pref;
+  pref.Init(pref_name, browser_state->GetPrefs());
+  pref.SetValue(value);
+}
+
 void SetWWANStateTo(bool value) {
   MainController* mainController = chrome_test_util::GetMainController();
   net::NetworkChangeNotifier::ConnectionType connectionType =

ios/chrome/test/earl_grey/chrome_earl_grey.h

@@ -682,8 +682,9 @@ UIWindow* GetAnyKeyWindow();
 - (int)userIntegerPref:(const std::string&)prefName;
 - (std::string)userStringPref:(const std::string&)prefName;
 
-// Sets the value of a boolean user pref in the original browser state.
+// Sets the value of a user pref in the original browser state.
 - (void)setBoolValue:(BOOL)value forUserPref:(const std::string&)UTF8PrefName;
+- (void)setIntegerValue:(int)value forUserPref:(const std::string&)UTF8PrefName;
 
 // Resets the BrowsingDataPrefs, which defines if its selected or not when
 // clearing Browsing data.

ios/chrome/test/earl_grey/chrome_earl_grey.mm

@@ -1306,6 +1306,13 @@ - (void)setBoolValue:(BOOL)value forUserPref:(const std::string&)UTF8PrefName {
   return [ChromeEarlGreyAppInterface setBoolValue:value forUserPref:prefName];
 }
 
+- (void)setIntegerValue:(int)value
+            forUserPref:(const std::string&)UTF8PrefName {
+  NSString* prefName = base::SysUTF8ToNSString(UTF8PrefName);
+  return [ChromeEarlGreyAppInterface setIntegerValue:value
+                                         forUserPref:prefName];
+}
+
 - (void)resetBrowsingDataPrefs {
   return [ChromeEarlGreyAppInterface resetBrowsingDataPrefs];
 }

ios/chrome/test/earl_grey/chrome_earl_grey_app_interface.h

@@ -542,6 +542,9 @@
 // Sets the value of a boolean user pref in the original browser state.
 + (void)setBoolValue:(BOOL)value forUserPref:(NSString*)prefName;
 
+// Sets the value of a integer user pref in the original browser state.
++ (void)setIntegerValue:(int)value forUserPref:(NSString*)prefName;
+
 // Resets the BrowsingDataPrefs, which defines if its selected or not when
 // clearing Browsing data.
 + (void)resetBrowsingDataPrefs;

ios/chrome/test/earl_grey/chrome_earl_grey_app_interface.mm

@@ -1075,6 +1075,12 @@ + (void)setBoolValue:(BOOL)value forUserPref:(NSString*)prefName {
       base::SysNSStringToUTF8(prefName).c_str(), value);
 }
 
++ (void)setIntegerValue:(int)value forUserPref:(NSString*)prefName {
+  chrome_test_util::SetIntegerUserPref(
+      chrome_test_util::GetOriginalBrowserState(),
+      base::SysNSStringToUTF8(prefName).c_str(), value);
+}
+
 + (void)resetBrowsingDataPrefs {
   PrefService* prefs = chrome_test_util::GetOriginalBrowserState()->GetPrefs();
   prefs->ClearPref(browsing_data::prefs::kDeleteBrowsingHistory);