public Statement buildStatement(final ParserRuleContext parseTree) {
    return build(Optional.of(getSources(parseTree)), parseTree);
  }
@Test
  public void shouldParsePause() {
    // Given:
    final SingleStatementContext allStmt = givenQuery("PAUSE ALL;");
    final SingleStatementContext oneStmt = givenQuery("PAUSE QUERY1;");

    // When:
    final PauseQuery allResult = (PauseQuery) builder.buildStatement(allStmt);
    final PauseQuery oneResult = (PauseQuery) builder.buildStatement(oneStmt);

    // Then:
    assertThat(allResult.getQueryId(), is(Optional.empty()));
    assertThat(oneResult.getQueryId(), is(Optional.of(new QueryId("QUERY1"))));
  }