Apply magicAssert to ImageViewAssertions

AdevintaSpain · Dec 18, 2017 · a26c1fe · a26c1fe
1 parent 5ffefe0
commit a26c1fe
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/library/src/main/java/com/schibsted/spain/barista/assertion/BaristaImageViewAssertions.kt b/library/src/main/java/com/schibsted/spain/barista/assertion/BaristaImageViewAssertions.kt
@@ -5,6 +5,7 @@ import android.support.annotation.IdRes
 import android.support.test.espresso.Espresso.onView
 import android.support.test.espresso.assertion.ViewAssertions.matches
 import android.support.test.espresso.matcher.ViewMatchers.withId
+import com.schibsted.spain.barista.internal.magicAssert
 import com.schibsted.spain.barista.internal.matcher.DrawableMatcher.Companion.withAnyDrawable
 import com.schibsted.spain.barista.internal.matcher.DrawableMatcher.Companion.withDrawable
 import com.schibsted.spain.barista.internal.matcher.DrawableMatcher.Companion.withoutDrawable
@@ -13,16 +14,16 @@ object BaristaImageViewAssertions {
 
     @JvmStatic
     fun assertHasDrawable(@IdRes imageViewId: Int, @DrawableRes drawable: Int) {
-        onView(withId(imageViewId)).check(matches(withDrawable(drawable)))
+        withId(imageViewId) magicAssert withDrawable(drawable)
     }
 
     @JvmStatic
     fun assertHasAnyDrawable(@IdRes imageViewId: Int) {
-        onView(withId(imageViewId)).check(matches(withAnyDrawable()))
+        withId(imageViewId) magicAssert withAnyDrawable()
     }
 
     @JvmStatic
     fun assertHasNoDrawable(@IdRes imageViewId: Int) {
-        onView(withId(imageViewId)).check(matches(withoutDrawable()))
+        withId(imageViewId) magicAssert withoutDrawable()
     }
 }