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 ColumnListHandlerTest(String name) {
31          super(name);
32      }
33  
34      public void testHandle() throws SQLException {
35          ResultSetHandler h = new ColumnListHandler();
36  
37          List results = (List) h.handle(this.rs);
38  
39          assertNotNull(results);
40          assertEquals(ROWS, results.size());
41  
42          assertEquals("1", results.get(0));
43          assertEquals("4", results.get(1));
44      }
45  
46      public void testColumnIndexHandle() throws SQLException {
47          ResultSetHandler h = new ColumnListHandler(2);
48          List results = (List) h.handle(this.rs);
49  
50          assertNotNull(results);
51          assertEquals(ROWS, results.size());
52  
53          assertEquals("2", results.get(0));
54          assertEquals("5", results.get(1));
55      }
56  
57      public void testColumnNameHandle() throws SQLException {
58          ResultSetHandler h = new ColumnListHandler("Three");
59          List results = (List) h.handle(this.rs);
60  
61          assertNotNull(results);
62          assertEquals(ROWS, results.size());
63  
64          assertEquals("3", results.get(0));
65          assertEquals("6", results.get(1));
66      }
67  
68      public void testEmptyResultSetHandle() throws SQLException {
69          ResultSetHandler h = new ColumnListHandler();
70          List results = (List) h.handle(this.emptyResultSet);
71  
72          assertNotNull(results);
73          assertTrue(results.isEmpty());
74      }
75  
76  }