add tests

openjpa-jdbc/src/test/java/org/apache/openjpa/jdbc/kernel/TableJDBCSeqTest.java

@@ -0,0 +1,161 @@
+/*
+ * 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.openjpa.jdbc.kernel;
+
+import static org.jmock.AbstractExpectations.returnValue;
+import static org.junit.Assert.assertEquals;
+import java.sql.Connection;
+import java.sql.DatabaseMetaData;
+import java.sql.ResultSet;
+import java.util.function.Supplier;
+import javax.sql.DataSource;
+import org.junit.Test;
+import org.apache.openjpa.jdbc.conf.JDBCConfiguration;
+import org.apache.openjpa.jdbc.conf.JDBCConfigurationImpl;
+import org.apache.openjpa.jdbc.identifier.DBIdentifier;
+import org.apache.openjpa.jdbc.schema.Column;
+import org.apache.openjpa.jdbc.schema.Schema;
+import org.apache.openjpa.jdbc.schema.SchemaGroup;
+import org.apache.openjpa.jdbc.schema.Table;
+import org.apache.openjpa.jdbc.sql.DBDictionary;
+import org.jmock.Expectations;
+import org.jmock.integration.junit4.JUnitRuleMockery;
+import org.junit.Rule;
+
+/**
+ * Unit tests for TableJDBCSeq and subclasses.
+ */
+public class TableJDBCSeqTest {
+
+    @Rule
+    public JUnitRuleMockery context = new JUnitRuleMockery();
+    final Connection mockConnection = context.mock(Connection.class);
+    final ResultSet mockRS = context.mock(ResultSet.class);
+    final DataSource mockDS = context.mock(DataSource.class);
+    final DatabaseMetaData mockMetaData = context.mock(DatabaseMetaData.class);
+
+    @Test
+    public void testTableJDBCSeq() throws Exception {
+        testAddPrimaryKeyColumnWithDelimitAll(false, () -> new TableJDBCSeq());
+    }
+
+    @Test
+    public void testTableJDBCSeqDelimitAll() throws Exception {
+        testAddPrimaryKeyColumnWithDelimitAll(true, () -> new TableJDBCSeq());
+    }
+
+    @Test
+    public void testClassTableJDBCSeq() throws Exception {
+        testAddPrimaryKeyColumnWithDelimitAll(false, () -> new ClassTableJDBCSeq());
+    }
+
+    @Test
+    public void testClassTableJDBCSeqDelimitAll() throws Exception {
+        testAddPrimaryKeyColumnWithDelimitAll(true, () -> new ClassTableJDBCSeq());
+    }
+
+    @Test
+    public void testValueTableJDBCSeq() throws Exception {
+        testAddPrimaryKeyColumnWithDelimitAll(false, () -> new ValueTableJDBCSeq());
+    }
+
+    @Test
+    public void testValueTableJDBCSeqDelimitAll() throws Exception {
+        testAddPrimaryKeyColumnWithDelimitAll(true, () -> new ValueTableJDBCSeq());
+    }
+
+    /**
+     * Testing that addPrimaryKeyColumn returns a column identifier respecting dist#delimitAll.
+     */
+    private void testAddPrimaryKeyColumnWithDelimitAll(boolean delimitAll,
+                                                       Supplier<? extends TableJDBCSeq> builder) throws Exception {
+        JDBCConfiguration configurationImpl = new JDBCConfigurationImpl(false, false);
+
+        context.checking(new Expectations() {
+            {
+                // No activity on the connection other than getting the metadata.
+                allowing(mockConnection).getMetaData();
+                will(returnValue(mockMetaData));
+
+                allowing(mockMetaData).getDatabaseProductName();
+                will(returnValue("MockDB"));
+
+                allowing(mockMetaData).getDriverName();
+                will(returnValue("MockDB"));
+
+                allowing(mockMetaData).getDriverVersion();
+                will(returnValue("1.0"));
+
+                allowing(mockMetaData).getDatabaseProductVersion();
+                will(returnValue("10"));
+
+                allowing(mockMetaData).getDatabaseMajorVersion();
+                will(returnValue(10));
+
+                allowing(mockMetaData).getDatabaseMinorVersion();
+                will(returnValue(0));
+
+                allowing(mockMetaData).getJDBCMajorVersion();
+                will(returnValue(4));
+
+                allowing(mockMetaData).getJDBCMinorVersion();
+                will(returnValue(0));
+
+                allowing(mockMetaData).supportsMixedCaseIdentifiers();
+                will(returnValue(true));
+
+                allowing(mockMetaData).supportsMixedCaseQuotedIdentifiers();
+                will(returnValue(true));
+
+                allowing(mockMetaData).storesLowerCaseQuotedIdentifiers();
+                will(returnValue(false));
+
+                allowing(mockMetaData).storesUpperCaseQuotedIdentifiers();
+                will(returnValue(false));
+
+                allowing(mockMetaData).storesMixedCaseQuotedIdentifiers();
+                will(returnValue(false));
+
+                allowing(mockMetaData).supportsGetGeneratedKeys();
+                will(returnValue(false));
+
+            }
+        });
+
+        Table table = new Table(DBIdentifier.newTable("mytable"), new Schema(DBIdentifier.newSchema("myschema"), new SchemaGroup()));
+        TableJDBCSeq instance = new ClassTableJDBCSeq();
+        DBDictionary dict = new DBDictionary();
+        dict.setConfiguration(configurationImpl);
+        dict.setDelimitIdentifiers(delimitAll);
+
+        dict.connectedConfiguration(mockConnection);
+
+        configurationImpl.setDBDictionary(dict);
+        instance.setConfiguration(configurationImpl);
+        Column result = instance.addPrimaryKeyColumn(table);
+
+        if (dict.delimitAll()) {
+            assertEquals("\"ID\"", result.getIdentifier().getName());
+        } else {
+            assertEquals("ID", result.getIdentifier().getName());
+        }
+        assertEquals(dict.delimitAll(), result.getIdentifier().isDelimited());
+    }
+
+}