Matched to_contain_class tests

Author: agg23

playwright/_impl/_assertions.py

@@ -313,14 +313,20 @@ async def to_contain_class(
             expected, str
         ):
             expected_text = to_expected_text_values(expected)
+            await self._expect_impl(
+                "to.contain.class.array",
+                FrameExpectOptions(expectedText=expected_text, timeout=timeout),
+                expected,
+                "Locator expected to contain class names",
+            )
         else:
             expected_text = to_expected_text_values([expected])
-        await self._expect_impl(
-            "to.contain.class",
-            FrameExpectOptions(expectedText=expected_text, timeout=timeout),
-            expected,
-            "Locator expected to contain class",
-        )
+            await self._expect_impl(
+                "to.contain.class",
+                FrameExpectOptions(expectedText=expected_text, timeout=timeout),
+                expected,
+                "Locator expected to contain class",
+            )
 
     async def not_to_contain_class(
         self,

tests/async/test_assertions.py

@@ -147,14 +147,28 @@ async def test_assertions_locator_to_have_class(page: Page, server: Server) -> N
 
 async def test_assertions_locator_to_contain_class(page: Page, server: Server) -> None:
     await page.goto(server.EMPTY_PAGE)
-    await page.set_content("<div class='foo bar another foobar'>kek</div>")
-    await expect(page.locator("div.foobar")).to_contain_class("foobar")
-    await expect(page.locator("div.foobar")).to_contain_class(["another foobar"])
-    await expect(page.locator("div.foobar")).not_to_contain_class(
-        "kekstar", timeout=100
+    await page.set_content("<div class='foo bar baz'></div>")
+    locator = page.locator("div")
+    await expect(locator).to_contain_class("")
+    await expect(locator).to_contain_class("bar")
+    await expect(locator).to_contain_class("baz bar")
+    await expect(locator).to_contain_class("  bar   foo ")
+    await expect(locator).not_to_contain_class(
+        "  baz   not-matching "
+    )  # Strip whitespace and match individual classes
+    with pytest.raises(AssertionError) as excinfo:
+        await expect(locator).to_contain_class("does-not-exist", timeout=100)
+
+    assert excinfo.match("Locator expected to contain class 'does-not-exist'")
+    assert excinfo.match("Actual value: foo bar baz")
+    assert excinfo.match("LocatorAssertions.to_contain_class with timeout 100ms")
+
+    await page.set_content(
+        '<div class="foo"></div><div class="hello bar"></div><div class="baz"></div>'
     )
-    with pytest.raises(AssertionError):
-        await expect(page.locator("div.foobar")).to_contain_class("oh-no", timeout=100)
+    await expect(locator).to_contain_class(["foo", "hello", "baz"])
+    await expect(locator).not_to_contain_class(["not-there", "hello", "baz"])
+    await expect(locator).not_to_contain_class(["foo", "hello"])
 
 
 async def test_assertions_locator_to_have_count(page: Page, server: Server) -> None: