public Statement buildStatement(final ParserRuleContext parseTree) {
    return build(Optional.of(getSources(parseTree)), parseTree);
  }
@Test
  public void shouldBuildAssertSchemaWithSubject() {
    // Given:
    final SingleStatementContext stmt
        = givenQuery("ASSERT SCHEMA SUBJECT X;");

    // When:
    final AssertSchema assertSchema = (AssertSchema) builder.buildStatement(stmt);

    // Then:
    assertThat(assertSchema.getSubject(), is(Optional.of("X")));
    assertThat(assertSchema.getId(), is(Optional.empty()));
    assertThat(assertSchema.getTimeout(), is(Optional.empty()));
    assertThat(assertSchema.checkExists(), is(true));
  }