[ActionSheet] Allow clients to get the view associated with a given action.

PiperOrigin-RevId: 366489478

Author: codeman7

components/ActionSheet/src/MDCActionSheetController.h

@@ -93,6 +93,13 @@ __attribute__((objc_subclassing_restricted)) @interface MDCActionSheetController
  */
 - (void)addAction:(nonnull MDCActionSheetAction *)action;
 
+/**
+ Returns the view associated with a given @c action.
+
+ @param action The action used to create the view.
+ */
+- (nullable UIView *)viewForAction:(nonnull MDCActionSheetAction *)action;
+
 /**
  The object that acts as the delegate of the @c MDCActionSheetController
 */

components/ActionSheet/src/MDCActionSheetController.m

@@ -18,8 +18,11 @@
 #import "private/MDCActionSheetItemTableViewCell.h"
 #import "private/MaterialActionSheetStrings.h"
 #import "private/MaterialActionSheetStrings_table.h"
+#import "MDCActionSheetAction.h"
 #import "MDCActionSheetControllerDelegate.h"
 #import "MaterialAvailability.h"
+#import "MaterialBottomSheet.h"
+#import "MaterialElevation.h"
 #import "MaterialShadowElevations.h"
 #import "MaterialTypography.h"
 #import "MaterialMath.h"
@@ -184,6 +187,16 @@ - (void)addAction:(MDCActionSheetAction *)action {
   [self updateTable];
 }
 
+- (UIView *)viewForAction:(MDCActionSheetAction *)action {
+  if (![self.actions containsObject:action]) {
+    return nil;
+  }
+  [self.view layoutIfNeeded];
+  NSUInteger rowIndex = [self.actions indexOfObject:action];
+  NSIndexPath *indexPath = [NSIndexPath indexPathForRow:rowIndex inSection:0];
+  return [self.tableView cellForRowAtIndexPath:indexPath];
+}
+
 - (NSArray<MDCActionSheetAction *> *)actions {
   return [_actions copy];
 }

components/ActionSheet/tests/snapshot/MDCActionSheetControllerViewForActionSnapshotTests.m

@@ -0,0 +1,115 @@
+// Copyright 2021-present the Material Components for iOS authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#import "MaterialSnapshot.h"
+
+#import "MaterialActionSheet.h"
+
+/**
+ Snapshot test for the @c viewForAction: API.
+ */
+@interface MDCActionSheetControllerViewForActionSnapshotTests : MDCSnapshotTestCase
+/** The @c MDCActionSheetController being tested. */
+@property(nonatomic, strong, nullable) MDCActionSheetController *actionSheet;
+
+/** An Action Sheet action. */
+@property(nonatomic, strong) MDCActionSheetAction *action1;
+
+/** An Action Sheet action. */
+@property(nonatomic, strong) MDCActionSheetAction *action2;
+
+@end
+
+@implementation MDCActionSheetControllerViewForActionSnapshotTests
+
+- (void)setUp {
+  [super setUp];
+
+  // Uncomment below to recreate all the goldens (or add the following line to the specific
+  // test you wish to recreate the golden for).
+  //  self.recordMode = YES;
+
+  self.actionSheet = [[MDCActionSheetController alloc] init];
+  self.action1 = [MDCActionSheetAction
+      actionWithTitle:@"Foo"
+                image:[UIImage mdc_testImageOfSize:CGSizeMake(24, 24)
+                                         withStyle:MDCSnapshotTestImageStyleCheckerboard]
+              handler:nil];
+  self.action2 = [MDCActionSheetAction
+      actionWithTitle:@"Bar"
+                image:[UIImage mdc_testImageOfSize:CGSizeMake(24, 24)
+                                         withStyle:MDCSnapshotTestImageStyleEllipses]
+              handler:nil];
+}
+
+- (void)tearDown {
+  if (self.actionSheet.presentingViewController) {
+    XCTestExpectation *expectation =
+        [[XCTestExpectation alloc] initWithDescription:@"Action sheet is dismissed"];
+    [self.actionSheet dismissViewControllerAnimated:NO
+                                         completion:^{
+                                           [expectation fulfill];
+                                         }];
+    [self waitForExpectations:@[ expectation ] timeout:5];
+  }
+  self.action1 = nil;
+  self.action2 = nil;
+  self.actionSheet = nil;
+
+  [super tearDown];
+}
+
+- (void)testOneActionAddedToActionSheet {
+  // Given
+  [self.actionSheet addAction:self.action1];
+
+  // When
+  UIView *view = [self.actionSheet viewForAction:self.action1];
+
+  // Then
+  [self generateSnapshotAndVerifyForView:view];
+}
+
+- (void)testMultipleActionsAddedToActionSheetViewForFirstIndex {
+  // Given
+  [self.actionSheet addAction:self.action1];
+  [self.actionSheet addAction:self.action2];
+
+  // When
+  UIView *view = [self.actionSheet viewForAction:self.action1];
+
+  // Then
+  [self generateSnapshotAndVerifyForView:view];
+}
+
+- (void)testMultipleActionsAddedToActionSheetViewForLastIndex {
+  // Given
+  [self.actionSheet addAction:self.action1];
+  [self.actionSheet addAction:self.action2];
+
+  // When
+  UIView *view = [self.actionSheet viewForAction:self.action2];
+
+  // Then
+  [self generateSnapshotAndVerifyForView:view];
+}
+
+#pragma mark - Helpers
+
+- (void)generateSnapshotAndVerifyForView:(UIView *)view {
+  UIView *snapshotView = [view mdc_addToBackgroundView];
+  [self snapshotVerifyView:snapshotView];
+}
+
+@end

components/ActionSheet/tests/unit/MDCActionSheetControllerTests.m

@@ -569,4 +569,58 @@ - (void)testTableViewContentInsetsWithHeaderDividerViewAndNoTitleOrMessage {
   XCTAssertGreaterThan(self.actionSheet.tableView.contentInset.top, originalTableContentInset);
 }
 
+- (void)testViewForActionWhenNoActionsAdded {
+  // Given
+  self.actionSheet.view.bounds = CGRectMake(0, 0, 500, 500);
+  MDCActionSheetAction *action = [MDCActionSheetAction actionWithTitle:@"Foo"
+                                                                 image:nil
+                                                               handler:nil];
+
+  // When
+  UIView *view = [self.actionSheet viewForAction:action];
+
+  // Then
+  XCTAssertNil(view);
+}
+
+- (void)testViewForActionWhenMultipleActionsAreAdded {
+  // Given
+  self.actionSheet.view.bounds = CGRectMake(0, 0, 500, 500);
+  MDCActionSheetAction *actionOne = [MDCActionSheetAction actionWithTitle:@"Foo"
+                                                                    image:nil
+                                                                  handler:nil];
+  MDCActionSheetAction *actionTwo = [MDCActionSheetAction actionWithTitle:@"Bar"
+                                                                    image:nil
+                                                                  handler:nil];
+  [self.actionSheet addAction:actionOne];
+  [self.actionSheet addAction:actionTwo];
+
+  // When
+  UIView *view = [self.actionSheet viewForAction:actionTwo];
+
+  // Then
+  XCTAssertNotNil(view);
+}
+
+- (void)testViewForActionWhenActionHasNotBeenAddedThenAdded {
+  // Given
+  self.actionSheet.view.bounds = CGRectMake(0, 0, 500, 500);
+  MDCActionSheetAction *action = [MDCActionSheetAction actionWithTitle:@"Foo"
+                                                                 image:nil
+                                                               handler:nil];
+
+  // When
+  UIView *view = [self.actionSheet viewForAction:action];
+
+  // Then
+  XCTAssertNil(view);
+
+  // When
+  [self.actionSheet addAction:action];
+  view = [self.actionSheet viewForAction:action];
+
+  // Then
+  XCTAssertNotNil(view);
+}
+
 @end