我已经使用selenium web driver JAVA将产品添加到购物车中，并将其从购物车中删除。我如何断言产品已移除或未移除？

// 获取购物车中的产品数量（示例假设购物车产品数量显示在页面上的一个元素中）
WebElement cartProductCount = driver.findElement(By.id("cartProductCount"));
int initialProductCount = Integer.parseInt(cartProductCount.getText());

// 执行产品移除操作（示例假设点击“删除”按钮来移除产品）
WebElement deleteButton = driver.findElement(By.xpath("//button[@class='deleteButton']"));
deleteButton.click();

// 等待购物车页面更新
WebDriverWait wait = new WebDriverWait(driver, 10);
wait.until(ExpectedConditions.stalenessOf(cartProductCount));

// 获取移除后的产品数量
int currentProductCount = Integer.parseInt(cartProductCount.getText());

// 断言产品是否已移除
if (currentProductCount == initialProductCount - 1) {
    System.out.println("产品已成功移除");
} else {
    System.out.println("产品未移除");
}