From dfd0800640ef5a3ae74def27f3cf7c1930590446 Mon Sep 17 00:00:00 2001
From: isabellekim <isabellekim@google.com>
Date: Thu, 1 Aug 2019 11:51:58 -0400
Subject: [PATCH] Add a test action for resetting the shape appearance model of
 a Shapeable View.

PiperOrigin-RevId: 261130040
---
 .../material/testutils/TestUtilsActions.java  | 39 +++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/tests/javatests/com/google/android/material/testutils/TestUtilsActions.java b/tests/javatests/com/google/android/material/testutils/TestUtilsActions.java
index ec1b25288..f12f27faf 100644
--- a/tests/javatests/com/google/android/material/testutils/TestUtilsActions.java
+++ b/tests/javatests/com/google/android/material/testutils/TestUtilsActions.java
@@ -28,6 +28,8 @@ import androidx.annotation.LayoutRes;
 import androidx.annotation.MenuRes;
 import androidx.annotation.Nullable;
 import com.google.android.material.appbar.CollapsingToolbarLayout;
+import com.google.android.material.shape.ShapeAppearanceModel;
+import com.google.android.material.shape.Shapeable;
 import androidx.core.view.ViewCompat;
 import androidx.core.widget.TextViewCompat;
 import android.util.SparseArray;
@@ -40,7 +42,9 @@ import androidx.test.espresso.ViewAction;
 import com.google.android.material.expandable.ExpandableWidget;
 import com.google.android.material.navigation.NavigationView;
 import com.google.android.material.tabs.TabLayout;
+import org.hamcrest.Description;
 import org.hamcrest.Matcher;
+import org.hamcrest.TypeSafeMatcher;
 
 public class TestUtilsActions {
   /**
@@ -463,4 +467,39 @@ public class TestUtilsActions {
       }
     };
   }
+
+  /**
+   * Returns a {@link ViewAction} that resets a {@link Shapeable} component to have the default (no
+   * corner, straight edge) shape.
+   */
+  public static ViewAction resetShape() {
+    return new ViewAction() {
+
+      @Override
+      public Matcher<View> getConstraints() {
+        return new TypeSafeMatcher<View>() {
+          @Override
+          public void describeTo(Description description) {
+            description.appendText("is shapeable");
+          }
+
+          @Override
+          public boolean matchesSafely(View view) {
+            return view instanceof Shapeable;
+          }
+        };
+      }
+
+      @Override
+      public String getDescription() {
+        return "reset shape";
+      }
+
+      @Override
+      public void perform(UiController uiController, View view) {
+        Shapeable shapeable = (Shapeable) view;
+        shapeable.setShapeAppearanceModel(new ShapeAppearanceModel());
+      }
+    };
+  }
 }