Java 如何获得动态生成的类名？

我想检查我的页面样式是否为特定格式，是否与photocentric、缩略图等不同。例如，如果我已选择以缩略图格式显示我的页面。首先，我选择了模板样式，然后我需要断言页面是否为缩略图样式。它会动态生成id和类名。如何获取动态生成的类名。我的代码如下所示

//Click Thumbnails
            pubDriver.findElement(By.cssSelector("img[alt=Template-option-thumbnails]")).click();
            pubDriver.findElement(By.id("saveForm")).click();

//Check if the  Page has been in thumbnail style template
         WebElement thumbnailStyle = wait.until(ExpectedConditions.presenceOfElementLocated(By.className("thumbnails")));
         Assert.assertTrue(thumbnailStyle.isDisplayed());
         sleep(1000);

如果我为断言提供上述编码，它将抛出以下错误消息。因为我在前面应用了样式模板格式，然后从中获取了类名

等待位于以下位置的元素出现10秒后超时：by.className:thumbnails

在应用了模板样式格式之后，我得到了以下东西，所以我不知道如何从这里动态获取类名

    <div ng-class="layout" ng-switch-default="" class="ng-scope thumbnails">
<div ng-switch="" on="layout"> 

 <!-- thumbnails template -->

---

您是否有任何其他常量参数用于尝试获取的项目

我会使用另一个选择器来获取项目，并使用类似

WebElement item = By(//some selector);
String itemClass = item.getAttribute("class");
//assert that class contains thumbnails

---

我没有任何常量参数。问题是我不知道如何让定位器指向特定的类。然后在应用程序的代码中添加一些常量参数。这将是我所知道的最干净、最具可读性的解决方案