@ -18,9 +18,9 @@
package org.wso2.carbon.device.mgt.mobile.impl.dao ;
import org.apache.commons.dbcp.BasicDataSource ;
import org.apache.commons.logging.Log ;
import org.apache.commons.logging.LogFactory ;
import org.apache.tomcat.jdbc.pool.PoolProperties ;
import org.testng.Assert ;
import org.testng.annotations.BeforeClass ;
import org.testng.annotations.Parameters ;
@ -36,6 +36,7 @@ import org.wso2.carbon.device.mgt.mobile.impl.common.TestDBConfiguration;
import org.wso2.carbon.device.mgt.mobile.impl.common.TestDBConfigurations ;
import org.wso2.carbon.device.mgt.mobile.util.MobileDeviceManagementUtil ;
import javax.sql.DataSource ;
import javax.xml.bind.JAXBContext ;
import javax.xml.bind.JAXBException ;
import javax.xml.bind.Unmarshaller ;
@ -46,237 +47,237 @@ import java.util.List;
public class MobileFeatureDAOTestSuite {
private static final Log log = LogFactory . getLog ( MobileFeatureDAOTestSuite . class ) ;
public static final String MBL_FEATURE_NAME = "Camera" ;
private static final String MBL_FEATURE_CODE = "500A" ;
public static final String MBL_FEATURE_DESCRIPTION = "Camera enable or disable" ;
public static final String MBL_FEATURE_UPDATED_CODE = "501B" ;
private TestDBConfiguration testDBConfiguration ;
private Connection conn = null ;
private Statement stmt = null ;
private MobileFeatureDAOImpl mblFeatureDAO ;
private int mblFeatureId ;
private static final Log log = LogFactory . getLog ( MobileFeatureDAOTestSuite . class ) ;
public static final String MBL_FEATURE_NAME = "Camera" ;
private static final String MBL_FEATURE_CODE = "500A" ;
public static final String MBL_FEATURE_DESCRIPTION = "Camera enable or disable" ;
public static final String MBL_FEATURE_UPDATED_CODE = "501B" ;
private TestDBConfiguration testDBConfiguration ;
private Connection conn = null ;
private Statement stmt = null ;
private MobileFeatureDAOImpl mblFeatureDAO ;
private int mblFeatureId ;
@BeforeClass
@Parameters ( "dbType" )
public void setUpDB ( String dbTypeStr ) throws Exception {
@BeforeClass
@Parameters ( "dbType" )
public void setUpDB ( String dbTypeStr ) throws Exception {
DBTypes dbType = DBTypes . valueOf ( dbTypeStr ) ;
testDBConfiguration = getTestDBConfiguration ( dbType ) ;
DBTypes dbType = DBTypes . valueOf ( dbTypeStr ) ;
testDBConfiguration = getTestDBConfiguration ( dbType ) ;
switch ( dbType ) {
case H2 :
createH2DB ( testDBConfiguration ) ;
BasicDataSource testDataSource = new BasicDataSource ( ) ;
testDataSource . setDriverClassName ( testDBConfiguration . getDriverClass ( ) ) ;
testDataSource . setUrl ( testDBConfiguration . getConnectionUrl ( ) ) ;
testDataSource . setUsername ( testDBConfiguration . getUserName ( ) ) ;
testDataSource . setPassword ( testDBConfiguration . getPwd ( ) ) ;
mblFeatureDAO = new MobileFeatureDAOImpl ( testDataSource ) ;
default :
}
}
switch ( dbType ) {
case H2 :
createH2DB ( testDBConfiguration ) ;
DataSource testDataSource = new org . apache . tomcat . jdbc . pool . DataSource ( ) ;
PoolProperties properties = new PoolProperties ( ) ;
properties . setUrl ( testDBConfiguration . getConnectionURL ( ) ) ;
properties . setDriverClassName ( testDBConfiguration . getDriverClassName ( ) ) ;
properties . setUsername ( testDBConfiguration . getUsername ( ) ) ;
properties . setPassword ( testDBConfiguration . getPassword ( ) ) ;
mblFeatureDAO = new MobileFeatureDAOImpl ( testDataSource ) ;
default :
}
}
private TestDBConfiguration getTestDBConfiguration ( DBTypes dbType ) throws
MobileDeviceManagementDAOException ,
DeviceManagementException {
private TestDBConfiguration getTestDBConfiguration ( DBTypes dbType ) throws
MobileDeviceManagementDAOException ,
DeviceManagementException {
File deviceMgtConfig = new File ( "src/test/resources/testdbconfig.xml" ) ;
Document doc = null ;
testDBConfiguration = null ;
TestDBConfigurations testDBConfigurations = null ;
File deviceMgtConfig = new File ( "src/test/resources/testdbconfig.xml" ) ;
Document doc = null ;
testDBConfiguration = null ;
TestDBConfigurations testDBConfigurations = null ;
doc = MobileDeviceManagementUtil . convertToDocument ( deviceMgtConfig ) ;
JAXBContext testDBContext = null ;
doc = MobileDeviceManagementUtil . convertToDocument ( deviceMgtConfig ) ;
JAXBContext testDBContext = null ;
try {
testDBContext = JAXBContext . newInstance ( TestDBConfigurations . class ) ;
Unmarshaller unmarshaller = testDBContext . createUnmarshaller ( ) ;
testDBConfigurations = ( TestDBConfigurations ) unmarshaller . unmarshal ( doc ) ;
} catch ( JAXBException e ) {
throw new MobileDeviceManagementDAOException ( "Error parsing test db configurations" , e ) ;
}
try {
testDBContext = JAXBContext . newInstance ( TestDBConfigurations . class ) ;
Unmarshaller unmarshaller = testDBContext . createUnmarshaller ( ) ;
testDBConfigurations = ( TestDBConfigurations ) unmarshaller . unmarshal ( doc ) ;
} catch ( JAXBException e ) {
throw new MobileDeviceManagementDAOException ( "Error parsing test db configurations" , e ) ;
}
Iterator < TestDBConfiguration > itrDBConfigs =
testDBConfigurations . getDbTypesList ( ) . iterator ( ) ;
while ( itrDBConfigs . hasNext ( ) ) {
testDBConfiguration = itrDBConfigs . next ( ) ;
if ( testDBConfiguration . get Db Type( ) . equals ( dbType . toString ( ) ) ) {
break ;
}
}
Iterator < TestDBConfiguration > itrDBConfigs =
testDBConfigurations . getDbTypesList ( ) . iterator ( ) ;
while ( itrDBConfigs . hasNext ( ) ) {
testDBConfiguration = itrDBConfigs . next ( ) ;
if ( testDBConfiguration . get Type( ) . equals ( dbType . toString ( ) ) ) {
break ;
}
}
return testDBConfiguration ;
}
return testDBConfiguration ;
}
private void createH2DB ( TestDBConfiguration testDBConf ) throws Exception {
Class . forName ( testDBConf . getDriverClass ( ) ) ;
conn = DriverManager . getConnection ( testDBConf . getConnectionU rl ( ) ) ;
stmt = conn . createStatement ( ) ;
stmt . executeUpdate ( "RUNSCRIPT FROM './src/test/resources/sql/CreateH2TestDB.sql'" ) ;
stmt . close ( ) ;
conn . close ( ) ;
}
private void createH2DB ( TestDBConfiguration testDBConf ) throws Exception {
Class . forName ( testDBConf . getDriverClass Name ( ) ) ;
conn = DriverManager . getConnection ( testDBConf . getConnectionU RL ( ) ) ;
stmt = conn . createStatement ( ) ;
stmt . executeUpdate ( "RUNSCRIPT FROM './src/test/resources/sql/CreateH2TestDB.sql'" ) ;
stmt . close ( ) ;
conn . close ( ) ;
}
@Test
public void addMobileFeatureTest ( )
throws MobileDeviceManagementDAOException {
@Test
public void addMobileFeatureTest ( )
throws MobileDeviceManagementDAOException {
MobileFeature mobileFeature = new MobileFeature ( ) ;
MobileFeature testMblFeature = new MobileFeature ( ) ;
mobileFeature . setCode ( MBL_FEATURE_CODE ) ;
mobileFeature . setDescription ( MBL_FEATURE_DESCRIPTION ) ;
mobileFeature . setName ( MBL_FEATURE_NAME ) ;
boolean added = mblFeatureDAO . addMobileFeature ( mobileFeature ) ;
try {
conn = DriverManager . getConnection ( testDBConfiguration . getConnectionU rl ( ) ) ;
stmt = conn . createStatement ( ) ;
ResultSet resultSet = stmt
. executeQuery (
"SELECT FEATURE_ID, CODE, NAME, DESCRIPTION FROM MBL_FEATURE WHERE CODE = '500A'" ) ;
while ( resultSet . next ( ) ) {
testMblFeature . setId ( resultSet . getInt ( 1 ) ) ;
testMblFeature . setCode ( resultSet . getString ( 2 ) ) ;
testMblFeature . setName ( resultSet . getString ( 3 ) ) ;
testMblFeature . setDescription ( resultSet . getString ( 4 ) ) ;
}
conn . close ( ) ;
} catch ( SQLException e ) {
log . error ( "Error in retrieving Mobile Feature data " , e ) ;
throw new MobileDeviceManagementDAOException ( "Error in retrieving Mobile Feature data " ,
e ) ;
} finally {
TestUtils . cleanupResources ( conn , stmt , null ) ;
}
mblFeatureId = testMblFeature . getId ( ) ;
Assert . assertTrue ( added , "MobileFeature is added" ) ;
Assert . assertEquals ( MBL_FEATURE_CODE , testMblFeature . getCode ( ) ,
"MobileFeature code has persisted successfully" ) ;
Assert . assertEquals ( MBL_FEATURE_NAME , testMblFeature . getName ( ) ,
"MobileFeature name has persisted successfully" ) ;
Assert . assertEquals ( MBL_FEATURE_DESCRIPTION , testMblFeature . getDescription ( ) ,
"MobileFeature description has persisted successfully" ) ;
}
MobileFeature mobileFeature = new MobileFeature ( ) ;
MobileFeature testMblFeature = new MobileFeature ( ) ;
mobileFeature . setCode ( MBL_FEATURE_CODE ) ;
mobileFeature . setDescription ( MBL_FEATURE_DESCRIPTION ) ;
mobileFeature . setName ( MBL_FEATURE_NAME ) ;
boolean added = mblFeatureDAO . addMobileFeature ( mobileFeature ) ;
try {
conn = DriverManager . getConnection ( testDBConfiguration . getConnectionU RL ( ) ) ;
stmt = conn . createStatement ( ) ;
ResultSet resultSet = stmt
. executeQuery (
"SELECT FEATURE_ID, CODE, NAME, DESCRIPTION FROM MBL_FEATURE WHERE CODE = '500A'" ) ;
while ( resultSet . next ( ) ) {
testMblFeature . setId ( resultSet . getInt ( 1 ) ) ;
testMblFeature . setCode ( resultSet . getString ( 2 ) ) ;
testMblFeature . setName ( resultSet . getString ( 3 ) ) ;
testMblFeature . setDescription ( resultSet . getString ( 4 ) ) ;
}
conn . close ( ) ;
} catch ( SQLException e ) {
log . error ( "Error in retrieving Mobile Feature data " , e ) ;
throw new MobileDeviceManagementDAOException ( "Error in retrieving Mobile Feature data " ,
e ) ;
} finally {
TestUtils . cleanupResources ( conn , stmt , null ) ;
}
mblFeatureId = testMblFeature . getId ( ) ;
Assert . assertTrue ( added , "MobileFeature is added" ) ;
Assert . assertEquals ( MBL_FEATURE_CODE , testMblFeature . getCode ( ) ,
"MobileFeature code has persisted successfully" ) ;
Assert . assertEquals ( MBL_FEATURE_NAME , testMblFeature . getName ( ) ,
"MobileFeature name has persisted successfully" ) ;
Assert . assertEquals ( MBL_FEATURE_DESCRIPTION , testMblFeature . getDescription ( ) ,
"MobileFeature description has persisted successfully" ) ;
}
@Test ( dependsOnMethods = { "addMobileFeatureTest" } )
public void getMobileFeatureByCodeTest ( )
throws MobileDeviceManagementDAOException {
@Test ( dependsOnMethods = { "addMobileFeatureTest" } )
public void getMobileFeatureByCodeTest ( )
throws MobileDeviceManagementDAOException {
MobileFeature mobileFeature = mblFeatureDAO . getMobileFeatureByCode ( MBL_FEATURE_CODE ) ;
Assert . assertEquals ( MBL_FEATURE_CODE , mobileFeature . getCode ( ) ,
"MobileFeature code has retrieved successfully" ) ;
Assert . assertEquals ( MBL_FEATURE_NAME , mobileFeature . getName ( ) ,
"MobileFeature name has retrieved successfully" ) ;
Assert . assertEquals ( MBL_FEATURE_DESCRIPTION , mobileFeature . getDescription ( ) ,
"MobileFeature description has retrieved successfully" ) ;
}
MobileFeature mobileFeature = mblFeatureDAO . getMobileFeatureByCode ( MBL_FEATURE_CODE ) ;
Assert . assertEquals ( MBL_FEATURE_CODE , mobileFeature . getCode ( ) ,
"MobileFeature code has retrieved successfully" ) ;
Assert . assertEquals ( MBL_FEATURE_NAME , mobileFeature . getName ( ) ,
"MobileFeature name has retrieved successfully" ) ;
Assert . assertEquals ( MBL_FEATURE_DESCRIPTION , mobileFeature . getDescription ( ) ,
"MobileFeature description has retrieved successfully" ) ;
}
@Test ( dependsOnMethods = { "addMobileFeatureTest" } )
public void deleteMobileFeatureByCodeTest ( )
throws MobileDeviceManagementDAOException {
boolean status = mblFeatureDAO . deleteMobileFeatureByCode ( MBL_FEATURE_CODE ) ;
try {
conn = DriverManager . getConnection ( testDBConfiguration . getConnectionU rl ( ) ) ;
stmt = conn . createStatement ( ) ;
ResultSet resultSet = stmt
. executeQuery (
"SELECT FEATURE_ID, CODE FROM MBL_FEATURE WHERE CODE = '500A'" ) ;
while ( resultSet . next ( ) ) {
status = false ;
}
conn . close ( ) ;
} catch ( SQLException e ) {
log . error ( "Error in deleting Mobile Feature data " , e ) ;
throw new MobileDeviceManagementDAOException ( "Error in deleting Mobile Feature data " ,
e ) ;
} finally {
TestUtils . cleanupResources ( conn , stmt , null ) ;
}
Assert . assertTrue ( status , "MobileFeature has deleted successfully" ) ;
}
@Test ( dependsOnMethods = { "addMobileFeatureTest" } )
public void deleteMobileFeatureByCodeTest ( )
throws MobileDeviceManagementDAOException {
boolean status = mblFeatureDAO . deleteMobileFeatureByCode ( MBL_FEATURE_CODE ) ;
try {
conn = DriverManager . getConnection ( testDBConfiguration . getConnectionU RL ( ) ) ;
stmt = conn . createStatement ( ) ;
ResultSet resultSet = stmt
. executeQuery (
"SELECT FEATURE_ID, CODE FROM MBL_FEATURE WHERE CODE = '500A'" ) ;
while ( resultSet . next ( ) ) {
status = false ;
}
conn . close ( ) ;
} catch ( SQLException e ) {
log . error ( "Error in deleting Mobile Feature data " , e ) ;
throw new MobileDeviceManagementDAOException ( "Error in deleting Mobile Feature data " ,
e ) ;
} finally {
TestUtils . cleanupResources ( conn , stmt , null ) ;
}
Assert . assertTrue ( status , "MobileFeature has deleted successfully" ) ;
}
@Test ( dependsOnMethods = { "addMobileFeatureTest" } )
public void getMobileFeatureByIdTest ( )
throws MobileDeviceManagementDAOException {
@Test ( dependsOnMethods = { "addMobileFeatureTest" } )
public void getMobileFeatureByIdTest ( )
throws MobileDeviceManagementDAOException {
MobileFeature mobileFeature = mblFeatureDAO . getMobileFeatureById ( mblFeatureId ) ;
Assert . assertEquals ( MBL_FEATURE_CODE , mobileFeature . getCode ( ) ,
"MobileFeature code has retrieved successfully" ) ;
Assert . assertEquals ( MBL_FEATURE_NAME , mobileFeature . getName ( ) ,
"MobileFeature name has retrieved successfully" ) ;
Assert . assertEquals ( MBL_FEATURE_DESCRIPTION , mobileFeature . getDescription ( ) ,
"MobileFeature description has retrieved successfully" ) ;
}
MobileFeature mobileFeature = mblFeatureDAO . getMobileFeatureById ( mblFeatureId ) ;
Assert . assertEquals ( MBL_FEATURE_CODE , mobileFeature . getCode ( ) ,
"MobileFeature code has retrieved successfully" ) ;
Assert . assertEquals ( MBL_FEATURE_NAME , mobileFeature . getName ( ) ,
"MobileFeature name has retrieved successfully" ) ;
Assert . assertEquals ( MBL_FEATURE_DESCRIPTION , mobileFeature . getDescription ( ) ,
"MobileFeature description has retrieved successfully" ) ;
}
@Test ( dependsOnMethods = { "addMobileFeatureTest" } )
public void getAllMobileFeaturesTest ( )
throws MobileDeviceManagementDAOException {
@Test ( dependsOnMethods = { "addMobileFeatureTest" } )
public void getAllMobileFeaturesTest ( )
throws MobileDeviceManagementDAOException {
List < MobileFeature > mobileFeatures = mblFeatureDAO . getAllMobileFeatures ( ) ;
Assert . assertNotNull ( mobileFeatures , "MobileFeature list is not null" ) ;
Assert . assertTrue ( mobileFeatures . size ( ) > 0 , "MobileFeature list has 1 MobileFeature" ) ;
}
List < MobileFeature > mobileFeatures = mblFeatureDAO . getAllMobileFeatures ( ) ;
Assert . assertNotNull ( mobileFeatures , "MobileFeature list is not null" ) ;
Assert . assertTrue ( mobileFeatures . size ( ) > 0 , "MobileFeature list has 1 MobileFeature" ) ;
}
@Test ( dependsOnMethods = { "addMobileFeatureTest" } )
public void deleteMobileFeatureByIdTest ( )
throws MobileDeviceManagementDAOException {
boolean status = mblFeatureDAO . deleteMobileFeatureById ( mblFeatureId ) ;
try {
conn = DriverManager . getConnection ( testDBConfiguration . getConnectionUrl ( ) ) ;
stmt = conn . createStatement ( ) ;
ResultSet resultSet = stmt
. executeQuery (
"SELECT FEATURE_ID, CODE FROM MBL_FEATURE WHERE FEATURE_ID = " +
mblFeatureId ) ;
while ( resultSet . next ( ) ) {
status = false ;
}
conn . close ( ) ;
} catch ( SQLException e ) {
log . error ( "Error in deleting Mobile Feature data " , e ) ;
throw new MobileDeviceManagementDAOException ( "Error in deleting Mobile Feature data " ,
e ) ;
} finally {
TestUtils . cleanupResources ( conn , stmt , null ) ;
}
Assert . assertTrue ( status , "MobileFeature has deleted successfully" ) ;
}
@Test ( dependsOnMethods = { "addMobileFeatureTest" } )
public void deleteMobileFeatureByIdTest ( )
throws MobileDeviceManagementDAOException {
boolean status = mblFeatureDAO . deleteMobileFeatureById ( mblFeatureId ) ;
try {
conn = DriverManager . getConnection ( testDBConfiguration . getConnectionURL ( ) ) ;
ResultSet resultSet = stmt
. executeQuery (
"SELECT FEATURE_ID, CODE FROM MBL_FEATURE WHERE FEATURE_ID = " +
mblFeatureId ) ;
while ( resultSet . next ( ) ) {
status = false ;
}
conn . close ( ) ;
} catch ( SQLException e ) {
log . error ( "Error in deleting Mobile Feature data " , e ) ;
throw new MobileDeviceManagementDAOException ( "Error in deleting Mobile Feature data " ,
e ) ;
} finally {
TestUtils . cleanupResources ( conn , stmt , null ) ;
}
Assert . assertTrue ( status , "MobileFeature has deleted successfully" ) ;
}
@Test ( dependsOnMethods = { "deleteMobileFeatureByCodeTest" , "addMobileFeatureTest" } )
public void updateMobileFeatureTest ( )
throws MobileDeviceManagementDAOException {
@Test ( dependsOnMethods = { "deleteMobileFeatureByCodeTest" , "addMobileFeatureTest" } )
public void updateMobileFeatureTest ( )
throws MobileDeviceManagementDAOException {
MobileFeature mobileFeature = new MobileFeature ( ) ;
MobileFeature testMblFeature = new MobileFeature ( ) ;
mobileFeature . setCode ( MBL_FEATURE_UPDATED_CODE ) ;
mobileFeature . setDescription ( MBL_FEATURE_DESCRIPTION ) ;
mobileFeature . setName ( MBL_FEATURE_NAME ) ;
mobileFeature . setId ( mblFeatureId ) ;
boolean updated = mblFeatureDAO . updateMobileFeature ( mobileFeature ) ;
try {
conn = DriverManager . getConnection ( testDBConfiguration . get ConnectionUrl ( ) ) ;
stmt = conn . createStatement ( ) ;
ResultSet resultSet = stmt
. executeQuery (
"SELECT FEATURE_ID, CODE, NAME, DESCRIPTION FROM MBL_FEATURE WHERE CODE = '" +
MBL_FEATURE_UPDATED_CODE + "'" ) ;
while ( resultSet . next ( ) ) {
testMblFeature . setId ( resultSet . getInt ( 1 ) ) ;
testMblFeature . setCode ( resultSet . getString ( 2 ) ) ;
testMblFeature . setName ( resultSet . getString ( 3 ) ) ;
testMblFeature . setDescription ( resultSet . getString ( 4 ) ) ;
}
conn . close ( ) ;
} catch ( SQLException e ) {
log . error ( "Error in updating Mobile Feature data " , e ) ;
throw new MobileDeviceManagementDAOException ( "Error in updating Mobile Feature data " ,
e ) ;
} finally {
TestUtils . cleanupResources ( conn , stmt , null ) ;
}
Assert . assertTrue ( updated , "MobileFeature has updated" ) ;
Assert . assertEquals ( MBL_FEATURE_UPDATED_CODE , testMblFeature . getCode ( ) ,
"MobileFeature data has updated successfully" ) ;
}
MobileFeature mobileFeature = new MobileFeature ( ) ;
MobileFeature testMblFeature = new MobileFeature ( ) ;
mobileFeature . setCode ( MBL_FEATURE_UPDATED_CODE ) ;
mobileFeature . setDescription ( MBL_FEATURE_DESCRIPTION ) ;
mobileFeature . setName ( MBL_FEATURE_NAME ) ;
mobileFeature . setId ( mblFeatureId ) ;
boolean updated = mblFeatureDAO . updateMobileFeature ( mobileFeature ) ;
try {
conn = DriverManager . getConnection ( testDBConfiguration . get DriverClassName ( ) ) ;
stmt = conn . createStatement ( ) ;
ResultSet resultSet = stmt
. executeQuery (
"SELECT FEATURE_ID, CODE, NAME, DESCRIPTION FROM MBL_FEATURE WHERE CODE = '" +
MBL_FEATURE_UPDATED_CODE + "'" ) ;
while ( resultSet . next ( ) ) {
testMblFeature . setId ( resultSet . getInt ( 1 ) ) ;
testMblFeature . setCode ( resultSet . getString ( 2 ) ) ;
testMblFeature . setName ( resultSet . getString ( 3 ) ) ;
testMblFeature . setDescription ( resultSet . getString ( 4 ) ) ;
}
conn . close ( ) ;
} catch ( SQLException e ) {
log . error ( "Error in updating Mobile Feature data " , e ) ;
throw new MobileDeviceManagementDAOException ( "Error in updating Mobile Feature data " ,
e ) ;
} finally {
TestUtils . cleanupResources ( conn , stmt , null ) ;
}
Assert . assertTrue ( updated , "MobileFeature has updated" ) ;
Assert . assertEquals ( MBL_FEATURE_UPDATED_CODE , testMblFeature . getCode ( ) ,
"MobileFeature data has updated successfully" ) ;
}
}