### Question:
PinotStatement extends AbstractBaseStatement { @Override public ResultSet executeQuery(String sql) throws SQLException { validateState(); try { Request request = new Request(QUERY_FORMAT, sql); _resultSetGroup = _session.execute(request); if (_resultSetGroup.getResultSetCount() == 0) { _resultSet = PinotResultSet.empty(); return _resultSet; } _resultSet = new PinotResultSet(_resultSetGroup.getResultSet(0)); return _resultSet; } catch (PinotClientException e) { throw new SQLException(String.format("Failed to execute query : %s", sql), e); } } PinotStatement(PinotConnection connection); @Override void close(); @Override ResultSet executeQuery(String sql); @Override boolean execute(String sql); @Override boolean execute(String sql, int autoGeneratedKeys); @Override boolean execute(String sql, int[] columnIndexes); @Override boolean execute(String sql, String[] columnNames); @Override ResultSet getResultSet(); @Override Connection getConnection(); @Override boolean isClosed();  }### Answer:
@Test public void testExecuteQuery() throws Exception { PinotConnection connection = new PinotConnection(Collections.singletonList("dummy"), _dummyPinotClientTransport); Statement statement = new PinotStatement(connection); ResultSet resultSet = statement.executeQuery("dummy"); Assert.assertNotNull(resultSet); Assert.assertEquals(statement.getConnection(), connection); }