code: public Wait action(TestAction action) {
          if (action instanceof TestActionBuilder) {
              getCondition().setAction(((TestActionBuilder) action).build());
              this.action.setAction(((TestActionBuilder) action).build());
              getBuilder().actions(((TestActionBuilder) action).build());
          } else {
              getCondition().setAction(action);
              this.action.setAction(action);
              getBuilder().actions(action);
          }
  
          return getBuilder().build();
      }
docs: Sets the test action to execute and wait for.
  @param action
  @return