org.sonar.server.db.ResultSetIteratorTest.java Source code

Introduction

Here is the source code for org.sonar.server.db.ResultSetIteratorTest.java

Source

/*
 * SonarQube, open source software quality management tool.
 * Copyright (C) 2008-2014 SonarSource
 * mailto:contact AT sonarsource DOT com
 *
 * SonarQube is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3 of the License, or (at your option) any later version.
 *
 * SonarQube is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */
package org.sonar.server.db;

import org.apache.commons.dbutils.DbUtils;
import org.junit.After;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.sonar.core.persistence.DbTester;
import org.sonar.test.DbTests;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.NoSuchElementException;

import static org.fest.assertions.Assertions.assertThat;
import static org.fest.assertions.Fail.fail;

@Category(DbTests.class)
public class ResultSetIteratorTest {

    @Rule
    public DbTester dbTester = new DbTester().schema(ResultSetIteratorTest.class, "schema.sql");

    Connection connection = null;

    @Before
    public void setUp() throws Exception {
        connection = dbTester.openConnection();
    }

    @After
    public void tearDown() throws Exception {
        DbUtils.closeQuietly(connection);
    }

    @Test
    public void create_iterator_from_statement() throws Exception {
        dbTester.prepareDbUnit(getClass(), "feed.xml");

        PreparedStatement stmt = connection.prepareStatement("select * from fake order by id");
        FirstIntColumnIterator iterator = new FirstIntColumnIterator(stmt);

        assertThat(iterator.hasNext()).isTrue();

        // calling multiple times hasNext() is ok
        assertThat(iterator.hasNext()).isTrue();

        assertThat(iterator.next()).isEqualTo(10);
        assertThat(iterator.hasNext()).isTrue();
        assertThat(iterator.next()).isEqualTo(20);

        // call next() without calling hasNext()
        assertThat(iterator.next()).isEqualTo(30);
        assertThat(iterator.hasNext()).isFalse();

        try {
            iterator.next();
            fail();
        } catch (NoSuchElementException e) {
            // ok
        }

        iterator.close();
        // statement is closed by ResultSetIterator
        assertThat(stmt.isClosed()).isTrue();
    }

    @Test
    public void iterate_empty_list() throws Exception {
        dbTester.prepareDbUnit(getClass(), "feed.xml");

        PreparedStatement stmt = connection.prepareStatement("select * from fake where id < 0");
        FirstIntColumnIterator iterator = new FirstIntColumnIterator(stmt);

        assertThat(iterator.hasNext()).isFalse();
    }

    @Test
    public void create_iterator_from_result_set() throws Exception {
        dbTester.prepareDbUnit(getClass(), "feed.xml");

        PreparedStatement stmt = connection.prepareStatement("select * from fake order by id");
        ResultSet rs = stmt.executeQuery();
        FirstIntColumnIterator iterator = new FirstIntColumnIterator(rs);

        assertThat(iterator.next()).isEqualTo(10);
        assertThat(iterator.next()).isEqualTo(20);
        assertThat(iterator.next()).isEqualTo(30);

        iterator.close();
        assertThat(rs.isClosed()).isTrue();
        stmt.close();
    }

    @Test
    public void remove_row_is_not_supported() throws Exception {
        PreparedStatement stmt = connection.prepareStatement("select * from fake order by id");
        FirstIntColumnIterator iterator = new FirstIntColumnIterator(stmt);

        try {
            iterator.remove();
            fail();
        } catch (UnsupportedOperationException ok) {
            // ok
        }

        iterator.close();
    }

    @Test
    public void fail_to_read_row() throws Exception {
        dbTester.prepareDbUnit(getClass(), "feed.xml");

        PreparedStatement stmt = connection.prepareStatement("select * from fake order by id");
        FailIterator iterator = new FailIterator(stmt);

        assertThat(iterator.hasNext()).isTrue();
        try {
            iterator.next();
            fail();
        } catch (IllegalStateException e) {
            assertThat(e.getCause()).isInstanceOf(SQLException.class);
        }
        iterator.close();
    }

    private static class FirstIntColumnIterator extends ResultSetIterator<Integer> {

        public FirstIntColumnIterator(PreparedStatement stmt) throws SQLException {
            super(stmt);
        }

        public FirstIntColumnIterator(ResultSet rs) {
            super(rs);
        }

        @Override
        protected Integer read(ResultSet rs) throws SQLException {
            return rs.getInt(1);
        }
    }

    private static class FailIterator extends ResultSetIterator<Integer> {

        public FailIterator(PreparedStatement stmt) throws SQLException {
            super(stmt);
        }

        @Override
        protected Integer read(ResultSet rs) throws SQLException {
            // column does not exist
            return rs.getInt(1234);
        }
    }
}