KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > com > mockobjects > sql > CommonMockStatement


1 package com.mockobjects.sql;
2
3 import com.mockobjects.ExpectationCounter;
4 import com.mockobjects.MockObject;
5 import com.mockobjects.ReturnObjectBag;
6
7 import java.sql.*;
8
9 /**
10  * @version $Revision: 1.4 $
11  */

12 abstract class CommonMockStatement extends MockObject implements Statement {
13     protected final ExpectationCounter myCloseCalls = new ExpectationCounter("CommonMockStatement.closeCalls");
14     private final ReturnObjectBag executeQueryResults = new ReturnObjectBag("executeQuery");
15     private final ReturnObjectBag executeUpdateResults = new ReturnObjectBag("executeUpdate");
16     private final ReturnObjectBag executeResults = new ReturnObjectBag("execute");
17
18     private int myUpdateCount = 0;
19     private SQLException myExecuteException = null;
20
21     private Connection myConnection = null;
22
23     public void addExpectedExecuteQuery(String JavaDoc queryString, ResultSet resultSet) {
24         executeQueryResults.putObjectToReturn(queryString, resultSet);
25     }
26
27     public void addExpectedExecuteUpdate(String JavaDoc queryString, int updateCount) {
28         executeUpdateResults.putObjectToReturn(queryString, updateCount);
29     }
30
31     public void addExpectedExecute(String JavaDoc queryString, boolean success) {
32         executeResults.putObjectToReturn(queryString, success);
33     }
34
35     public void setExpectedCloseCalls(int callCount) {
36         myCloseCalls.setExpected(callCount);
37     }
38
39     public void setupConnection(Connection conn) {
40         myConnection = conn;
41     }
42
43     public void setupThrowExceptionOnExecute(SQLException exception) {
44         myExecuteException = exception;
45     }
46
47     public void setupUpdateCount(int updateCount) {
48         myUpdateCount = updateCount;
49     }
50
51     protected void innerExecute() throws SQLException {
52         if (null != myExecuteException) {
53             throw myExecuteException;
54         }
55     }
56
57     public void close() throws SQLException {
58         myCloseCalls.inc();
59     }
60
61     public boolean execute(String JavaDoc sql) throws SQLException {
62         innerExecute();
63         return executeResults.getNextReturnBoolean(sql);
64     }
65
66     public ResultSet executeQuery(String JavaDoc sql) throws SQLException {
67         innerExecute();
68         return (ResultSet) executeQueryResults.getNextReturnObject(sql);
69     }
70
71     public int executeUpdate(String JavaDoc sql) throws SQLException {
72         innerExecute();
73         return executeUpdateResults.getNextReturnInt(sql);
74     }
75
76     public int getMaxFieldSize() throws SQLException {
77         notImplemented();
78         return 0;
79     }
80
81     public void setMaxFieldSize(int max) throws SQLException {
82         notImplemented();
83     }
84
85     public int getMaxRows() throws SQLException {
86         notImplemented();
87         return 0;
88     }
89
90     public void setMaxRows(int max) throws SQLException {
91         notImplemented();
92     }
93
94     public void setEscapeProcessing(boolean enable) throws SQLException {
95         notImplemented();
96     }
97
98     public int getQueryTimeout() throws SQLException {
99         notImplemented();
100         return 0;
101     }
102
103     public void setQueryTimeout(int seconds) throws SQLException {
104         notImplemented();
105     }
106
107     public void cancel() throws SQLException {
108         notImplemented();
109     }
110
111     public SQLWarning getWarnings() throws SQLException {
112         notImplemented();
113         return null;
114     }
115
116     public void clearWarnings() throws SQLException {
117         notImplemented();
118     }
119
120     public void setCursorName(String JavaDoc name) throws SQLException {
121         notImplemented();
122     }
123
124     public ResultSet getResultSet() throws SQLException {
125         notImplemented();
126         return null;
127     }
128
129     public int getUpdateCount() throws SQLException {
130         return myUpdateCount;
131     }
132
133     public boolean getMoreResults() throws SQLException {
134         notImplemented();
135         return false;
136     }
137
138     public void setFetchDirection(int direction) throws SQLException {
139         notImplemented();
140     }
141
142     public int getFetchDirection() throws SQLException {
143         notImplemented();
144         return 0;
145     }
146
147     public void setFetchSize(int rows) throws SQLException {
148         notImplemented();
149     }
150
151     public int getFetchSize() throws SQLException {
152         notImplemented();
153         return 0;
154     }
155
156     public int getResultSetConcurrency() throws SQLException {
157         notImplemented();
158         return 0;
159     }
160
161     public int getResultSetType() throws SQLException {
162         notImplemented();
163         return 0;
164     }
165
166     public void addBatch(String JavaDoc sql) throws SQLException {
167         notImplemented();
168     }
169
170     public void clearBatch() throws SQLException {
171         notImplemented();
172     }
173
174     public int[] executeBatch() throws SQLException {
175         notImplemented();
176         return null;
177     }
178
179     public Connection getConnection() throws SQLException {
180         return myConnection;
181     }
182 }
183
Popular Tags