/*******************************************************************************
* Copyright (c) 1998, 2015 Oracle and/or its affiliates. All rights reserved.
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v1.0 and Eclipse Distribution License v. 1.0
* which accompanies this distribution.
* The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
* and the Eclipse Distribution License is available at
* http://www.eclipse.org/org/documents/edl-v10.php.
*
* Contributors:
* Oracle - initial API and implementation from Oracle TopLink
******************************************************************************/
package org.eclipse.persistence.testing.tests.sessionbroker;
import java.math.BigDecimal;
import org.eclipse.persistence.expressions.Expression;
import org.eclipse.persistence.expressions.ExpressionBuilder;
import org.eclipse.persistence.internal.sessions.DatabaseSessionImpl;
import org.eclipse.persistence.mappings.ManyToManyMapping;
import org.eclipse.persistence.mappings.OneToOneMapping;
import org.eclipse.persistence.queries.ReadAllQuery;
import org.eclipse.persistence.sessions.broker.SessionBroker;
import org.eclipse.persistence.sessions.DatabaseLogin;
import org.eclipse.persistence.sessions.DatabaseSession;
import org.eclipse.persistence.sessions.Session;
import org.eclipse.persistence.testing.framework.OracleDBPlatformHelper;
import org.eclipse.persistence.testing.framework.ReadAllTest;
import org.eclipse.persistence.testing.framework.ReadObjectTest;
import org.eclipse.persistence.testing.framework.TestModel;
import org.eclipse.persistence.testing.framework.TestSuite;
import org.eclipse.persistence.testing.models.employee.domain.Employee;
import org.eclipse.persistence.testing.models.employee.domain.LargeProject;
import org.eclipse.persistence.testing.models.employee.domain.PhoneNumber;
import org.eclipse.persistence.testing.models.employee.domain.SmallProject;
import org.eclipse.persistence.sessions.server.ClientSession;
import org.eclipse.persistence.sessions.server.ServerSession;
import org.eclipse.persistence.tools.schemaframework.PopulationManager;
import org.eclipse.persistence.tools.schemaframework.SchemaManager;
public class ServerBrokerTestModel extends TestModel {
public Session oldSession;
public static final String QUERY_NAME = "localNumbers";
public ServerBrokerTestModel() {
setDescription("This model tests reading/writing/deleting using the session broker with the employee demo.");
}
/**
* Add a named query to a session Broker
*/
public void addQuery1(SessionBroker broker) {
// Add a predefined query with argument for employee with first name Bob.
ExpressionBuilder builder = new ExpressionBuilder();
ReadAllQuery query = new ReadAllQuery(Employee.class, builder);
Expression exp = builder.get("id").equal(builder.getParameter("ID"));
query.setSelectionCriteria(exp.and(builder.get("firstName").equal("Bob")));
query.addArgument("ID", BigDecimal.class);
broker.addQuery(QUERY_NAME, query);
}
/**
* Add a named query to a session Broker
*/
public void addQuery2(SessionBroker broker) {
// Add a predefined query without argument for retrieving numbers with 613 area code.
ExpressionBuilder builder = new ExpressionBuilder();
ReadAllQuery query = new ReadAllQuery(PhoneNumber.class, builder);
query.setSelectionCriteria(builder.get("type").equal("work"));
broker.addQuery(QUERY_NAME, query);
}
public void addTests() {
addTest(getReadObjectTestSuite());
addTest(getReadAllTestSuite());
addTest(getSessionBrokerClientQueryTestSuite());
//cr 2923
addTest(new SessionBrokerPlatformTest());
addTest(new ReadOnlyClassesInSessionBrokerTest());
addTest(new VerifyClientBrokerCreationTest());
addTest(new SessionBrokerCustomFunctionTest(SessionBrokerCustomFunctionTest.READALLQUERY_TEST));
addTest(new SessionBrokerCustomFunctionTest(SessionBrokerCustomFunctionTest.UPDATEALLQUERY_TEST));
addTest(new SessionBrokerCustomFunctionTest(SessionBrokerCustomFunctionTest.DELETEALLQUERY_TEST));
}
/**
* Build the session broker, this assume two data-sources Broker1 and Broker2 (JConnect).
*/
public SessionBroker buildBroker() {
createTables();
SessionBroker broker = new SessionBroker();
addQuery1(broker);
addQuery2(broker);
ServerSession ssession1 = new ServerSession(getLogin1());
ServerSession ssession2 = new ServerSession(getLogin2());
ssession1.addDescriptors(new EmployeeProject1());
ssession2.addDescriptors(new EmployeeProject2());
broker.registerSession("broker1", ssession1);
broker.registerSession("broker2", ssession2);
broker.setLog(oldSession.getLog());
broker.setLogLevel(oldSession.getLogLevel());
broker.login();
// Set session for join table.
((ManyToManyMapping)ssession1.getDescriptor(Employee.class).getObjectBuilder().getMappingForAttributeName("projects")).setSessionName("broker2");
// Disable delete verify.
((OneToOneMapping)ssession1.getDescriptor(Employee.class).getObjectBuilder().getMappingForAttributeName("address")).setShouldVerifyDelete(false);
SessionBroker clientBroker = broker.acquireClientSessionBroker();
new org.eclipse.persistence.testing.models.employee.relational.EmployeeSystem().populate(clientBroker);
return clientBroker;
}
/**
* Create the tables on both databases.
*/
public void createTables() {
SessionBroker broker = new SessionBroker();
DatabaseSession session1 = new DatabaseSessionImpl(getLogin1());
DatabaseSession session2 = new DatabaseSessionImpl(getLogin2());
session1.addDescriptors(new EmployeeProject1());
session2.addDescriptors(new EmployeeProject2());
broker.registerSession("broker1", session1);
broker.registerSession("broker2", session2);
broker.setLog(oldSession.getLog());
broker.setLogLevel(oldSession.getLogLevel());
broker.login();
org.eclipse.persistence.testing.models.employee.relational.EmployeeTableCreator tables =
new org.eclipse.persistence.testing.models.employee.relational.EmployeeTableCreator();
tables.replaceTables(session1);
tables.replaceTables(session2);
tables.dropConstraints(session1);
tables.dropConstraints(session2);
new SchemaManager(session1).createSequences();
new SchemaManager(session2).createSequences();
broker.logout();
}
public static DatabaseLogin getLogin1() {
//Oracle 11.1
DatabaseLogin login = new DatabaseLogin();
try {
login.usePlatform(OracleDBPlatformHelper.getInstance().getOracle9Platform());
} catch (Exception e) {
}
login.useOracleThinJDBCDriver();
login.setDatabaseURL("ottvm028.ca.oracle.com:1521:toplink");
login.setUserName("QA7");
login.setPassword("password");
login.useNativeSequencing();
login.getDefaultSequence().setPreallocationSize(1);
return login;
}
public static DatabaseLogin getLogin2() {
//Oracle 11.1
DatabaseLogin login = new DatabaseLogin();
try {
login.usePlatform(OracleDBPlatformHelper.getInstance().getOracle9Platform());
} catch (Exception e) {
}
login.useOracleThinJDBCDriver();
login.setDatabaseURL("ottvm028.ca.oracle.com:1521:toplink");
login.setUserName("QA8");
login.setPassword("password");
login.useNativeSequencing();
login.getDefaultSequence().setPreallocationSize(1);
return login;
}
public static TestSuite getReadAllTestSuite() {
TestSuite suite = new TestSuite();
suite.setName("BrokerReadAllTestSuite");
suite.setDescription("This suite tests read all sql.");
PopulationManager manager = PopulationManager.getDefaultManager();
suite.addTest(new ReadAllTest(Employee.class, 12));
suite.addTest(new ReadAllTest(org.eclipse.persistence.testing.models.employee.domain.Project.class, 15));
suite.addTest(new ReadAllTest(LargeProject.class, 5));
suite.addTest(new ReadAllTest(SmallProject.class, 10));
return suite;
}
public static TestSuite getReadObjectTestSuite() {
TestSuite suite = new TestSuite();
suite.setName("BrokerReadObjectTestSuite");
suite.setDescription("This suite test read sql.");
PopulationManager manager = PopulationManager.getDefaultManager();
suite.addTest(new ReadObjectTest(manager.getObject(Employee.class, "0001")));
suite.addTest(new ReadObjectTest(manager.getObject(Employee.class, "0004")));
suite.addTest(new ReadObjectTest(manager.getObject(LargeProject.class, "0001")));
suite.addTest(new ReadObjectTest(manager.getObject(SmallProject.class, "0004")));
return suite;
}
public static TestSuite getSessionBrokerClientQueryTestSuite() {
TestSuite suite = new TestSuite();
suite.setName("SessionBrokerClientQueryTestSuite");
suite.setDescription("This suite tests queries.");
PopulationManager manager = PopulationManager.getDefaultManager();
suite.addTest(new SessionBrokerClientQueryTestCase());
return suite;
}
/**
* Replace the executor with the old session.
*/
public void reset() {
try {
if (getExecutor().getSession() instanceof SessionBroker) {
((DatabaseSession)((ClientSession)((SessionBroker)getExecutor().getSession()).getSessionForName("broker1")).getParent()).logout();
((DatabaseSession)((ClientSession)((SessionBroker)getExecutor().getSession()).getSessionForName("broker2")).getParent()).logout();
}
} finally {
getExecutor().setSession(oldSession);
}
}
/**
* Replace the executor with the broker session.
*/
public void setup() {
oldSession = getSession();
getExecutor().setSession(buildBroker());
}
}