001    /*
002     * Copyright 2001,2004 The Apache Software Foundation.
003     * 
004     * Licensed under the Apache License, Version 2.0 (the "License");
005     * you may not use this file except in compliance with the License.
006     * You may obtain a copy of the License at
007     * 
008     *      http://www.apache.org/licenses/LICENSE-2.0
009     * 
010     * Unless required by applicable law or agreed to in writing, software
011     * distributed under the License is distributed on an "AS IS" BASIS,
012     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013     * See the License for the specific language governing permissions and
014     * limitations under the License.
015     */
016    
017    package org.apache.commons.scaffold.sql;
018    
019    
020    /**
021     * MockAdaptor for testing StorageBeans.
022     *
023     * @author Ted Husted
024     * @version $Revision: 155464 $ $Date: 2005-02-26 13:26:54 +0000 (Sat, 26 Feb 2005) $
025     */
026    
027    import java.sql.SQLException;
028    
029    import javax.sql.DataSource;
030    
031    import com.mockobjects.sql.MockDataSource;
032    
033    
034    public class MockAdaptor extends ConnectionAdaptor {
035    
036    
037            // Inherits JavaDoc
038        protected DataSource getDataSource(String key)
039                throws SQLException {
040    
041            if (null==datasource) {
042                datasource = new MockDataSource();
043            }
044    
045            return datasource;
046        }
047    
048    
049    
050        /**
051         * Set internal datasource field.
052         * Call once *before* getPool().
053         */
054        public void setupDataSource(DataSource datasource) {
055    
056            ConnectionAdaptor.datasource = datasource;
057    
058        }
059    
060    
061         // Required
062        public MockAdaptor() {
063    
064            if (null==pool) pool = this;
065    
066        }
067    
068    
069    } // end MockAdaptor
070