[iOS] Fix background color for IdentityButtonControl

Related to:
crrev.com/c/3067351

Fixed: 1238293
Change-Id: I2b5ba57703e672e7180b2cc279c7276a651698e8
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/3220793
Commit-Queue: Jérôme <jlebel@chromium.org>
Commit-Queue: Nohemi Fernandez <fernandex@chromium.org>
Auto-Submit: Jérôme <jlebel@chromium.org>
Reviewed-by: Nohemi Fernandez <fernandex@chromium.org>
Cr-Commit-Position: refs/heads/main@{#931063}

ios/chrome/browser/ui/authentication/signin/consistency_promo_signin/consistency_default_account/consistency_default_account_view_controller.mm

@@ -164,8 +164,6 @@ - (void)viewDidLoad {
   // Add IdentityButtonControl for the default identity.
   self.identityButtonControl =
       [[IdentityButtonControl alloc] initWithFrame:CGRectZero];
-  self.identityButtonControl.backgroundColor =
-      [UIColor colorNamed:kGroupedSecondaryBackgroundColor];
   self.identityButtonControl.arrowDirection = IdentityButtonControlArrowRight;
   self.identityButtonControl.identityViewStyle = IdentityViewStyleConsistency;
   [self.identityButtonControl addTarget:self

ios/chrome/browser/ui/authentication/views/identity_button_control.mm

@@ -44,7 +44,7 @@ - (instancetype)initWithFrame:(CGRect)frame {
   if (self) {
     self.accessibilityIdentifier = kIdentityButtonControlIdentifier;
     self.layer.cornerRadius = kIdentityButtonControlRadius;
-    self.backgroundColor = [UIColor colorNamed:kSecondaryBackgroundColor];
+    [self setUnhighlightedBackgroundColor];
 
     // Adding view elements inside.
     // Down or right arrow.
@@ -116,6 +116,8 @@ - (void)setArrowDirection:(IdentityButtonControlArrowDirection)arrowDirection {
 
 - (void)setIdentityViewStyle:(IdentityViewStyle)style {
   self.identityView.style = style;
+  if (!self.highlighted)
+    [self setUnhighlightedBackgroundColor];
 }
 
 - (IdentityViewStyle)identityViewStyle {
@@ -124,6 +126,21 @@ - (IdentityViewStyle)identityViewStyle {
 
 #pragma mark - Private
 
+// Sets the background color when not highlighted.
+- (void)setUnhighlightedBackgroundColor {
+  DCHECK(!self.highlighted);
+  switch (self.identityView.style) {
+    case IdentityViewStyleDefault:
+    case IdentityViewStyleIdentityChooser:
+      self.backgroundColor = [UIColor colorNamed:kSecondaryBackgroundColor];
+      break;
+    case IdentityViewStyleConsistency:
+      self.backgroundColor =
+          [UIColor colorNamed:kGroupedSecondaryBackgroundColor];
+      break;
+  }
+}
+
 - (CGPoint)locationFromTouches:(NSSet*)touches {
   UITouch* touch = [touches anyObject];
   return [touch locationInView:self];
@@ -153,9 +170,8 @@ - (void)setHighlighted:(BOOL)highlighted {
 
   if (highlighted) {
     self.backgroundColor = [UIColor colorNamed:kGrey300Color];
-  } else if (self.identityViewStyle != IdentityViewStyleConsistency) {
-    // Background color for the consistency web sign-in is reset manually.
-    self.backgroundColor = [UIColor colorNamed:kSecondaryBackgroundColor];
+  } else {
+    [self setUnhighlightedBackgroundColor];
   }
 }