2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2017 AT&T Intellectual Property. All rights
7 * ================================================================================
8 * Licensed under the Apache License, Version 2.0 (the "License");
9 * you may not use this file except in compliance with the License.
10 * You may obtain a copy of the License at
12 * http://www.apache.org/licenses/LICENSE-2.0
14 * Unless required by applicable law or agreed to in writing, software
15 * distributed under the License is distributed on an "AS IS" BASIS,
16 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17 * See the License for the specific language governing permissions and
18 * limitations under the License.
19 * ============LICENSE_END=========================================================
22 package org.openecomp.appc.lockmanager.impl.sql;
24 import org.junit.Rule;
25 import org.junit.rules.TestName;
26 import org.openecomp.appc.dao.util.DefaultJdbcConnectionFactory;
27 import org.openecomp.appc.lockmanager.api.LockManager;
28 import org.openecomp.appc.lockmanager.api.LockManagerBaseTests;
29 import org.openecomp.appc.lockmanager.impl.sql.JdbcLockManager;
30 import org.openecomp.appc.lockmanager.impl.sql.MySqlConnectionFactory;
32 import java.sql.Connection;
33 import java.sql.PreparedStatement;
34 import java.sql.SQLException;
36 public abstract class MySqlLockManagerBaseTests extends LockManagerBaseTests {
38 private static final boolean USE_REAL_DB = Boolean.getBoolean("lockmanager.tests.useRealDb");
39 private static final String TABLE_LOCK_MANAGEMENT = "TEST_LOCK_MANAGEMENT";
40 private static final String JDBC_URL = System.getProperty("lockmanager.tests.jdbcUrl", "jdbc:mysql://192.168.1.2/test");
41 private static final String JDBC_USERNAME = System.getProperty("lockmanager.tests.jdbcUsername", "test");
42 private static final String JDBC_PASSWORD = System.getProperty("lockmanager.tests.jdbcPassword", "123456");
44 protected static final int CONCURRENT_TEST_WAIT_TIME = 10; // secs
47 public TestName testName = new TestName();
50 protected LockManager createLockManager() {
51 JdbcLockManager jdbcLockManager = createJdbcLockManager(USE_REAL_DB);
52 DefaultJdbcConnectionFactory connectionFactory = new MySqlConnectionFactory();
53 connectionFactory.setJdbcURL(JDBC_URL);
54 connectionFactory.setJdbcUserName(JDBC_USERNAME);
55 connectionFactory.setJdbcPassword(JDBC_PASSWORD);
56 jdbcLockManager.setConnectionFactory(connectionFactory);
57 jdbcLockManager.setTableName(TABLE_LOCK_MANAGEMENT);
58 System.out.println("=> Running LockManager test [" + jdbcLockManager.getClass().getName() + "." + testName.getMethodName() + "]" + (USE_REAL_DB ? ". JDBC URL is [" + JDBC_URL + "]" : ""));
59 clearTestLocks(jdbcLockManager);
60 return jdbcLockManager;
63 protected abstract JdbcLockManager createJdbcLockManager(boolean useRealDb);
65 protected boolean setSynchronizer(Synchronizer synchronizer) {
66 if(!(lockManager instanceof SynchronizerReceiver)) {
67 System.err.println("Skipping concurrency test [" + testName.getMethodName() + "] for LockManager of type " + lockManager.getClass());
70 ((SynchronizerReceiver)lockManager).setSynchronizer(synchronizer);
74 private static final String SQL_DELETE_LOCK_RECORD = String.format("DELETE FROM %s WHERE RESOURCE_ID=?", TABLE_LOCK_MANAGEMENT);
75 private void clearTestLocks(JdbcLockManager jdbcLockManager) {
76 Connection connection = jdbcLockManager.openDbConnection();
77 if(connection == null) {
81 for(Resource resource: Resource.values()) {
82 try(PreparedStatement statement = connection.prepareStatement(SQL_DELETE_LOCK_RECORD)) {
83 statement.setString(1, resource.name());
84 statement.executeUpdate();
87 } catch(SQLException e) {
88 throw new RuntimeException("Cannot clear test resources in table", e);
90 jdbcLockManager.closeDbConnection(connection);