Swift中如何根据逻辑切换视图变量实现UI测试代码复用？

在UI测试场景里，我需要和页面元素交互并验证视图的显示状态，每个视图都有对应的操作流程和元素定义。但得根据逻辑判断，和不同视图做交互、验证——比如当request.response为true时验证ViewA，否则和ViewB交互。现有代码能正常运行，但我想通过定义本地变量指代视图，复用交互和验证代码。

原有冗余代码 #

if request.response {
    // 点击ViewA的按钮One
    // 点击ViewA的按钮Two
    // 验证ViewA的内容一
    // 验证ViewA的内容二
    // 验证ViewA的内容三
    // 点击ViewA的按钮Three
    // 验证ViewA已消失
} else {
    // 点击ViewB的按钮One
    // 点击ViewB的按钮Two
    // 验证ViewB的内容一
    // 验证ViewB的内容二
    // 验证ViewB的内容三
    // 点击ViewB的按钮Three
    // 验证ViewB已消失
}

优化后复用代码 #

var interactedView = ViewB
if request.response {
    interactedView = ViewA
}
// 点击interactedView的按钮One
// 点击interactedView的按钮Two
// 验证interactedView的内容一
// 验证interactedView的内容二
// 验证interactedView的内容三
// 点击interactedView的按钮Three
// 验证interactedView已消失

内容的提问来源于stack exchange，提问作者jacobcan118