使用 TestBench 从上下文中查找对话框

如何解决使用 TestBench 从上下文中查找对话框

我正在寻找一种从子上下文中查找 DialogElement 的方法。从“根”上下文中查找它工作正常，但我正在构建一个用于测试的页面组件，当我尝试使用 DialogElement 方法查找 $(DialogElement.class) 时，它会尝试在上下文中找到它的当前元素，但 Dialog 似乎被添加到主/根上下文，因此它总是失败。有什么建议怎么做吗？

更新

我找到了一个适用于测试中任何地方的解决方案：

DialogElement dialogElement = new ElementQuery<>(DialogElement.class).context(getDriver()).first()

更新 2

设法改进它：

DialogElement dialogElement = $(DialogElement.class).context(getDriver()).first()

如果有人知道更优雅的，请随时发布。