implement toHaveClass

Author: lsoares

lib/src/main/kotlin/seleniumtestinglib/WebElement.kt

@@ -59,4 +59,8 @@ val WebElement.accessibleDescription: String
             ?: getAttribute("aria-description")
             ?: getAttribute("title")
 
+val WebElement.classList: Set<String>
+    get() = getAttribute("class").takeUnless(String::isNullOrBlank)
+        ?.split(Regex("\\s+"))?.toSet() ?: emptySet()
+
 internal val WebElement.wrappedDriver get() = (this as RemoteWebElement).wrappedDriver as RemoteWebDriver

lib/src/main/kotlin/seleniumtestinglib/jestdom/Api.kt

@@ -86,8 +86,17 @@ data class JestDomMatcher(
         }
     }
 
-    fun toHaveClass(className: String) {
-        validate(element?.getAttribute("class")?.contains(className) == null)
+    fun toHaveClass(vararg classNames: String, exact: Boolean = false) {
+        val expectedClasses = classNames.map { it.split(Regex("\\s+")) }.flatten().toSet()
+        val elementClasses = element?.classList ?: emptySet()
+        if (expectedClasses.isEmpty()) {
+            validate(elementClasses.isNotEmpty())
+            return
+        }
+        when (exact) {
+            false -> validate(elementClasses.containsAll(expectedClasses))
+            true  -> validate(expectedClasses, elementClasses)
+        }
     }
 
     fun toHaveFocus() {

lib/src/test/kotlin/seleniumtestinglib/jestdom/HaveAttributeTest.kt renamed to lib/src/test/kotlin/seleniumtestinglib/jestdom/AttributeTest.kt

@@ -9,7 +9,7 @@ import seleniumtestinglib.render
 import kotlin.test.Test
 
 @ExtendWith(DriverLifeCycle::class)
-class HaveAttributeTest(private val driver: RemoteWebDriver) {
+class AttributeTest(private val driver: RemoteWebDriver) {
 
     @Test
     fun `has attribute`() {

lib/src/test/kotlin/seleniumtestinglib/jestdom/ClassTest.kt

@@ -0,0 +1,43 @@
+package seleniumtestinglib.jestdom
+
+import org.junit.jupiter.api.extension.ExtendWith
+import org.openqa.selenium.remote.RemoteWebDriver
+import seleniumtestinglib.DriverLifeCycle
+import seleniumtestinglib.classList
+import seleniumtestinglib.queries.ByType.TestId
+import seleniumtestinglib.queries.getBy
+import seleniumtestinglib.render
+import kotlin.test.Test
+import kotlin.test.assertEquals
+
+@ExtendWith(DriverLifeCycle::class)
+class ClassTest(private val driver: RemoteWebDriver) {
+
+    @Test
+    fun `with classes`() {
+        driver.render(
+            """<button data-testid="delete-button" class="btn extra btn-danger">
+                       Delete item
+                     </button>"""
+        )
+
+        val deleteButton = driver.getBy(TestId, "delete-button")
+        assertEquals(setOf("btn", "btn-danger", "extra"), deleteButton.classList)
+        expect(deleteButton).toHaveClass()
+        expect(deleteButton).toHaveClass("extra")
+        expect(deleteButton).toHaveClass("btn-danger btn")
+        expect(deleteButton).toHaveClass("btn-danger", "btn")
+        expect(deleteButton).not.toHaveClass("btn-link")
+        expect(deleteButton).toHaveClass("btn-danger extra btn", exact = true)
+        expect(deleteButton).not.toHaveClass("btn-danger extra", exact = true)
+    }
+
+    @Test
+    fun `no classes`() {
+        driver.render("""<button data-testid="no-classes">No Classes</button>""")
+
+        val noClasses = driver.getBy(TestId, "no-classes")
+        assertEquals(emptySet(), noClasses.classList)
+        expect(noClasses).not.toHaveClass()
+    }
+}