update class pass

Author: Skn0tt

playwright/src/test/java/com/microsoft/playwright/TestLocatorAssertions.java

@@ -18,6 +18,8 @@
 
 import com.microsoft.playwright.assertions.LocatorAssertions;
 import com.microsoft.playwright.assertions.PlaywrightAssertions;
+import com.microsoft.playwright.assertions.LocatorAssertions.ContainsClassOptions;
+
 import org.junit.jupiter.api.Test;
 import org.opentest4j.AssertionFailedError;
 
@@ -1109,4 +1111,49 @@ void defaultTimeoutZeroHasTextPass() {
     // Restore default.
     PlaywrightAssertions.setDefaultAssertionTimeout(5_000);
   }
+
+  @Test
+  void containsClassPass() {
+    page.setContent("<div class='foo bar baz'></div>");
+    Locator locator = page.locator("div");
+    assertThat(locator).containsClass("");
+    assertThat(locator).containsClass("bar");
+    assertThat(locator).containsClass("baz bar");
+    assertThat(locator).containsClass("  bar   foo ");
+    assertThat(locator).not().containsClass("  baz not-matching");
+  }
+
+  @Test
+  void containsClassPassWithSvgs() {
+    page.setContent("<svg class='c1 c2' role='img' xmlns='http://www.w3.org/2000/svg' viewBox='0 0 512 512'></svg>");
+    assertThat(page.locator("svg")).containsClass("c1");
+    assertThat(page.locator("svg")).containsClass("c2 c1");
+  }
+
+  @Test
+  void containsClassFail() {
+    page.setContent("<div class='bar baz'></div>");
+    AssertionFailedError e = assertThrows(AssertionFailedError.class, () -> {
+      assertThat(page.locator("div")).containsClass("does-not-exist", new ContainsClassOptions().setTimeout(1000));
+    });
+    assertTrue(e.getMessage().contains("Locator.expect with timeout 1000ms"), e.getMessage());
+  }
+
+  @Test
+  void containsClassPassWithArray() {
+    page.setContent("<div class='foo'></div><div class='hello bar'></div><div class='baz'></div>");
+    Locator locator = page.locator("div");
+    assertThat(locator).containsClass(asList("foo", "hello", "baz"));
+    assertThat(locator).not().hasClass(new String[]{"not-there", "hello", "baz"}); // Class not there
+    assertThat(locator).not().hasClass(new String[]{"foo", "hello"}); // length mismatch
+  }
+
+  @Test
+  void containsClassFailWithArray() {
+    page.setContent("<div class='foo'></div><div class='bar'></div><div class='bar'></div>");
+    AssertionFailedError e = assertThrows(AssertionFailedError.class, () -> {
+      assertThat(page.locator("div")).containsClass(asList("foo", "bar", "baz"), new ContainsClassOptions().setTimeout(1000));
+    });
+    assertTrue(e.getMessage().contains("Locator.expect with timeout 1000ms"), e.getMessage());
+  }
 }