| /** |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package org.apache.sqoop.manager.oracle; |
| |
| import java.io.UnsupportedEncodingException; |
| import java.sql.SQLException; |
| import java.util.Formatter; |
| |
| import org.apache.sqoop.manager.oracle.util.OracleUtils; |
| import org.apache.commons.logging.Log; |
| import org.apache.commons.logging.LogFactory; |
| |
| import org.apache.hadoop.conf.Configuration; |
| import org.apache.sqoop.SqoopOptions; |
| import org.apache.sqoop.testutil.ManagerCompatTestCase; |
| import org.junit.Test; |
| |
| import static org.junit.Assert.fail; |
| |
| /** |
| * Test the basic Oracle connection manager with the various column types. |
| */ |
| public class OracleCompatTest extends ManagerCompatTestCase { |
| |
| public static final Log LOG = LogFactory.getLog( |
| OracleCompatTest.class.getName()); |
| |
| @Override |
| protected Log getLogger() { |
| return LOG; |
| } |
| |
| @Override |
| protected String getDbFriendlyName() { |
| return "Oracle"; |
| } |
| |
| @Override |
| protected String getConnectString() { |
| return OracleUtils.CONNECT_STRING; |
| } |
| |
| @Override |
| protected SqoopOptions getSqoopOptions(Configuration conf) { |
| SqoopOptions opts = new SqoopOptions(conf); |
| OracleUtils.setOracleAuth(opts); |
| return opts; |
| |
| } |
| |
| @Override |
| protected void dropTableIfExists(String table) throws SQLException { |
| OracleUtils.dropTable(table, getManager()); |
| } |
| |
| @Override |
| public void tearDown() { |
| super.tearDown(); |
| |
| // If we actually ran the test, we'll need to 'cool off' afterwards. |
| if (!skipped) { |
| // Oracle XE will block connections if you create new ones too quickly. |
| // See http://forums.oracle.com/forums/thread.jspa?messageID=1145120 |
| LOG.info("Sleeping to wait for Oracle connection cache clear..."); |
| try { |
| Thread.sleep(250); |
| } catch (InterruptedException ie) { |
| // This delay may run a bit short.. no problem. |
| } |
| } |
| } |
| |
| @Override |
| protected String getDoubleType() { |
| return "DOUBLE PRECISION"; |
| } |
| |
| @Override |
| protected String getVarBinaryType() { |
| return "RAW(12)"; |
| } |
| |
| // Oracle does not provide a BOOLEAN type. |
| @Override |
| protected boolean supportsBoolean() { |
| return false; |
| } |
| |
| // Oracle does not provide a BIGINT type. |
| @Override |
| protected boolean supportsBigInt() { |
| return false; |
| } |
| |
| // Oracle does not provide a TINYINT type. |
| @Override |
| protected boolean supportsTinyInt() { |
| return false; |
| } |
| |
| // Oracle does not provide a LONGVARCHAR type. |
| @Override |
| protected boolean supportsLongVarChar() { |
| return false; |
| } |
| |
| // Oracle does not provide a TIME type. We test DATE and TIMESTAMP |
| @Override |
| protected boolean supportsTime() { |
| return false; |
| } |
| |
| @Override |
| protected String getDateInsertStr(String dateStr) { |
| return "TO_DATE(" + dateStr + ", 'YYYY-MM-DD')"; |
| } |
| |
| @Override |
| protected String getTimestampInsertStr(String tsStr) { |
| return "TO_TIMESTAMP(" + tsStr + ", 'YYYY-MM-DD HH24:MI:SS.FF')"; |
| } |
| |
| @Override |
| protected String getDateSeqOutput(String asInserted) { |
| // DATE is actually a TIMESTAMP in Oracle; add a time component. |
| return asInserted + " 00:00:00.0"; |
| } |
| |
| @Override |
| protected String getFixedCharSeqOut(int fieldWidth, String asInserted) { |
| return padString(fieldWidth, asInserted); |
| } |
| |
| @Override |
| protected String getRealSeqOutput(String realAsInserted) { |
| return realAsInserted; |
| } |
| |
| @Override |
| protected String getFloatSeqOutput(String floatAsInserted) { |
| return floatAsInserted; |
| } |
| |
| @Override |
| protected String getDoubleSeqOutput(String doubleAsInserted) { |
| return doubleAsInserted; |
| } |
| |
| @Override |
| protected String getVarBinarySeqOutput(String asInserted) { |
| return toLowerHexString(asInserted); |
| } |
| |
| @Override |
| protected String getBlobInsertStr(String blobData) { |
| // Oracle wants blob data encoded as hex (e.g. '01fca3b5'). |
| |
| StringBuilder sb = new StringBuilder(); |
| sb.append("'"); |
| |
| Formatter fmt = new Formatter(sb); |
| try { |
| for (byte b : blobData.getBytes("UTF-8")) { |
| fmt.format("%02X", b); |
| } |
| } catch (UnsupportedEncodingException uee) { |
| // Should not happen; Java always supports UTF-8. |
| fail("Could not get utf-8 bytes for blob string"); |
| return null; |
| } |
| sb.append("'"); |
| return sb.toString(); |
| } |
| |
| protected String getBinaryFloatInsertStr(float f) { |
| return "TO_BINARY_FLOAT('" + f + "')"; |
| } |
| |
| protected String getBinaryDoubleInsertStr(double d) { |
| return "TO_BINARY_DOUBLE('" + d + "')"; |
| } |
| |
| // Disable this test since Oracle isn't ANSI compliant. |
| @Override |
| @Test |
| public void testEmptyStringCol() { |
| this.skipped = true; |
| LOG.info( |
| "Oracle treats empty strings as null (non-ANSI compliant). Skipping."); |
| } |
| |
| @Override |
| @Test |
| public void testTimestamp1() { |
| verifyType(getTimestampType(), |
| getTimestampInsertStr("'2009-04-24 18:24:00'"), |
| "2009-04-24 18:24:00.0"); |
| } |
| |
| @Override |
| @Test |
| public void testTimestamp2() { |
| verifyType(getTimestampType(), |
| getTimestampInsertStr("'2009-04-24 18:24:00.0002'"), |
| "2009-04-24 18:24:00.0002"); |
| } |
| |
| @Override |
| @Test |
| public void testDate1() { |
| verifyType("DATE", getDateInsertStr("'2009-01-12'"), |
| getDateSeqOutput("2009-01-12")); |
| } |
| |
| @Override |
| @Test |
| public void testDate2() { |
| verifyType("DATE", getDateInsertStr("'2009-04-24'"), |
| getDateSeqOutput("2009-04-24")); |
| } |
| |
| @Test |
| public void testRawVal() { |
| verifyType("RAW(8)", "'12ABCD'", getVarBinarySeqOutput("12ABCD"), true); |
| } |
| |
| @Test |
| public void testBinaryFloat() { |
| verifyType("BINARY_FLOAT", getBinaryFloatInsertStr(25f), "25.0"); |
| verifyType("BINARY_FLOAT", getBinaryFloatInsertStr(+6.34f), "6.34"); |
| |
| // Max and min are from Oracle DB SQL reference for 10g release 2 |
| float max = 3.40282E+38F; |
| verifyType("BINARY_FLOAT", getBinaryFloatInsertStr(max), "3.40282E38"); |
| float min = 1.17549E-38F; |
| verifyType("BINARY_FLOAT", getBinaryFloatInsertStr(min), "1.17549E-38"); |
| } |
| |
| @Test |
| public void testBinaryDouble() { |
| verifyType("BINARY_DOUBLE", getBinaryDoubleInsertStr(0.5d), "0.5"); |
| verifyType("BINARY_DOUBLE", getBinaryDoubleInsertStr(-1d), "-1.0"); |
| |
| // Max and min are from Oracle DB SQL reference for 10g release 2 |
| double max = 1.79769313486231E+308; |
| verifyType("BINARY_DOUBLE", getBinaryDoubleInsertStr(max), |
| "1.79769313486231E308"); |
| double min = 2.22507485850720E-308; |
| verifyType("BINARY_DOUBLE", getBinaryDoubleInsertStr(min), |
| "2.2250748585072E-308"); |
| } |
| } |
| |