Merge pull request #94 from vacxe/recycler-view-child-count-impliment…

…ation Add child count assertation for recycler view
agoda-com · Apr 17, 2018 · 914d9bc · 914d9bc
2 parents 9ac9986 + 39bae9d
commit 914d9bc
Show file tree

Hide file tree

Showing 5 changed files with 72 additions and 12 deletions.
diff --git a/kakao/src/main/kotlin/com/agoda/kakao/Assertions.kt b/kakao/src/main/kotlin/com/agoda/kakao/Assertions.kt
@@ -20,10 +20,6 @@ import android.support.test.espresso.web.webdriver.DriverAtoms
 import org.hamcrest.CoreMatchers
 import org.hamcrest.Matcher
 import org.hamcrest.Matchers
-import org.hamcrest.TypeSafeMatcher
-import org.hamcrest.Description
-import android.view.View
-import android.widget.RatingBar
 import java.lang.AssertionError
 
 /**
@@ -489,6 +485,43 @@ interface ViewPagerAssertions : BaseAssertions {
     }
 }
 
+/**
+ * Provides assertions for view with adapters
+ *
+ * @see RecyclerAdapterAssertions
+ */
+interface AdapterAssertions {
+    val view: ViewInteraction
+}
+
+/**
+ * Provides assertions for recyclerView adapter
+ */
+interface RecyclerAdapterAssertions : AdapterAssertions {
+    /**
+     * Check size of recycler view
+     *
+     * @param size expected child count size in recycler view
+     */
+    fun hasSize(size: Int) {
+        view.check(ViewAssertions.matches(RecyclerViewAdapterSizeMatcher(size)))
+    }
+}
+
+/**
+ * Provides assertions for listView adapter
+ */
+interface ListViewAdapterAssertions : AdapterAssertions {
+    /**
+     * Check size of recycler view
+     *
+     * @param size expected child count size in recycler view
+     */
+    fun hasSize(size: Int) {
+        view.check(ViewAssertions.matches(ListViewViewAdapterSizeMatcher(size)))
+    }
+}
+
 /**
  * Provides assertions for NavigationView
  */

diff --git a/kakao/src/main/kotlin/com/agoda/kakao/Matchers.kt b/kakao/src/main/kotlin/com/agoda/kakao/Matchers.kt
@@ -19,10 +19,7 @@ import android.support.v4.view.ViewPager
 import android.support.v4.widget.SwipeRefreshLayout
 import android.support.v7.widget.RecyclerView
 import android.view.View
-import android.widget.ImageView
-import android.widget.ProgressBar
-import android.widget.RatingBar
-import android.widget.TextView
+import android.widget.*
 import org.hamcrest.Description
 import org.hamcrest.Matcher
 import org.hamcrest.TypeSafeMatcher
@@ -192,6 +189,34 @@ class RatingBarMatcher(private val value: Float) : BoundedMatcher<View, RatingBa
     }
 }
 
+/**
+ * Matches RecyclerView with count of childs
+ *
+ * @param size of childs count in RecyclerView
+ */
+class RecyclerViewAdapterSizeMatcher(private val size: Int) : BoundedMatcher<View, RecyclerView>(RecyclerView::class.java) {
+    override fun matchesSafely(view: RecyclerView?) = view?.adapter?.let {
+        it.itemCount == size } ?: false
+
+    override fun describeTo(description: Description) {
+        description.appendText("recycle view size is: $size")
+    }
+}
+
+/**
+ * Matches ListView with count of childs
+ *
+ * @param size of childs count in ListView
+ */
+class ListViewViewAdapterSizeMatcher(private val size: Int) : BoundedMatcher<View, ListView>(ListView::class.java) {
+    override fun matchesSafely(view: ListView?) = view?.adapter?.let {
+        it.count == size } ?: false
+
+    override fun describeTo(description: Description) {
+        description.appendText("recycle view size is: $size")
+    }
+}
+
 /**
  * Matches given drawable with current one
  *

diff --git a/kakao/src/main/kotlin/com/agoda/kakao/Views.kt b/kakao/src/main/kotlin/com/agoda/kakao/Views.kt
@@ -340,7 +340,7 @@ class KTextInputLayout : KBaseView<KTextInputLayout>, TextInputLayoutAssertions
  * @see KAdapterItem
  * @see KAdapterItemTypeBuilder
  */
-class KListView : ScrollViewActions, BaseAssertions {
+class KListView : ScrollViewActions, BaseAssertions, ListViewAdapterAssertions {
     val matcher: Matcher<View>
     val itemTypes: Map<KClass<out KAdapterItem<*>>, KAdapterItemType<KAdapterItem<*>>>
 
@@ -509,7 +509,7 @@ class KListView : ScrollViewActions, BaseAssertions {
  * @see KRecyclerItem
  * @see KRecyclerItemTypeBuilder
  */
-class KRecyclerView : RecyclerActions, BaseAssertions {
+class KRecyclerView : RecyclerActions, BaseAssertions, RecyclerAdapterAssertions {
     val matcher: Matcher<View>
     val itemTypes: Map<KClass<out KRecyclerItem<*>>, KRecyclerItemType<KRecyclerItem<*>>>
 

diff --git a/sample/src/androidTest/kotlin/com/agoda/sample/ListTest.kt b/sample/src/androidTest/kotlin/com/agoda/sample/ListTest.kt
@@ -20,6 +20,7 @@ class ListTest {
         screen {
             list {
                 isVisible()
+                hasSize(10)
 
                 firstChild<TestListScreen.Item> {
                     isVisible()
@@ -65,4 +66,4 @@ class ListTest {
             }
         }
     }
-}
+}
diff --git a/sample/src/androidTest/kotlin/com/agoda/sample/RecyclerTest.kt b/sample/src/androidTest/kotlin/com/agoda/sample/RecyclerTest.kt
@@ -20,6 +20,7 @@ class RecyclerTest {
         screen {
             recycler {
                 isVisible()
+                hasSize(10)
 
                 firstChild<TestRecyclerScreen.Item> {
                     isVisible()
@@ -51,4 +52,4 @@ class RecyclerTest {
             }
         }
     }
-}
+}