1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one or more
3    * contributor license agreements.  See the NOTICE file distributed with
4    * this work for additional information regarding copyright ownership.
5    * The ASF licenses this file to You under the Apache License, Version 2.0
6    * (the "License"); you may not use this file except in compliance with
7    * the License.  You may obtain a copy of the License at
8    *
9    *      http://www.apache.org/licenses/LICENSE-2.0
10   *
11   * Unless required by applicable law or agreed to in writing, software
12   * distributed under the License is distributed on an "AS IS" BASIS,
13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14   * See the License for the specific language governing permissions and
15   * limitations under the License.
16   */
17  package org.apache.commons.dbutils.handlers;
18  
19  import java.sql.SQLException;
20  import java.util.List;
21  
22  import org.apache.commons.dbutils.BaseTestCase;
23  import org.apache.commons.dbutils.ResultSetHandler;
24  
25  /**
26   * ColumnListHandlerTest
27   */
28  public class ColumnListHandlerTest extends BaseTestCase {
29  
30      public void testHandle() throws SQLException {
31          ResultSetHandler h = new ColumnListHandler();
32  
33          List results = (List) h.handle(this.rs);
34  
35          assertNotNull(results);
36          assertEquals(ROWS, results.size());
37  
38          assertEquals("1", results.get(0));
39          assertEquals("4", results.get(1));
40      }
41  
42      public void testColumnIndexHandle() throws SQLException {
43          ResultSetHandler h = new ColumnListHandler(2);
44          List results = (List) h.handle(this.rs);
45  
46          assertNotNull(results);
47          assertEquals(ROWS, results.size());
48  
49          assertEquals("2", results.get(0));
50          assertEquals("5", results.get(1));
51      }
52  
53      public void testColumnNameHandle() throws SQLException {
54          ResultSetHandler h = new ColumnListHandler("Three");
55          List results = (List) h.handle(this.rs);
56  
57          assertNotNull(results);
58          assertEquals(ROWS, results.size());
59  
60          assertEquals("3", results.get(0));
61          assertEquals("6", results.get(1));
62      }
63  
64      public void testEmptyResultSetHandle() throws SQLException {
65          ResultSetHandler h = new ColumnListHandler();
66          List results = (List) h.handle(this.emptyResultSet);
67  
68          assertNotNull(results);
69          assertTrue(results.isEmpty());
70      }
71  
72  }