Unit test framework for DAO tests

components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/common/DBTypes.java

@@ -0,0 +1,26 @@
+/*
+ * Copyright (c) 2014, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
+ * WSO2 Inc. 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.
+ * 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.wso2.carbon.device.mgt.core.common;
+
+
+public enum DBTypes {
+    Oracle("Oracle"),H2("H2"),MySql("MySql");
+
+    String dbName ;
+    DBTypes(String dbStrName) {
+        dbName = dbStrName;
+    }
+}

components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/common/TestDBConfiguration.java

@@ -0,0 +1,86 @@
+/*
+ * Copyright (c) 2014, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
+ * WSO2 Inc. 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.
+ * 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.wso2.carbon.device.mgt.core.common;
+
+import javax.xml.bind.annotation.XmlAttribute;
+import javax.xml.bind.annotation.XmlElement;
+import javax.xml.bind.annotation.XmlRootElement;
+
+@XmlRootElement(name = "DBType")
+public class TestDBConfiguration {
+
+    private String connectionUrl;
+    private String driverClass;
+    private String userName;
+    private String pwd;
+
+    @Override public String toString() {
+        return "TestDBConfiguration{" +
+                "connectionUrl='" + connectionUrl + '\'' +
+                ", driverClass='" + driverClass + '\'' +
+                ", userName='" + userName + '\'' +
+                ", pwd='" + pwd + '\'' +
+                ", dbType='" + dbType + '\'' +
+                '}';
+    }
+
+    private String dbType;
+
+    @XmlElement(name = "connectionurl", nillable = false)
+    public String getConnectionUrl() {
+        return connectionUrl;
+    }
+
+    public void setConnectionUrl(String connectionUrl) {
+        this.connectionUrl = connectionUrl;
+    }
+    @XmlElement(name = "driverclass", nillable = false)
+    public String getDriverClass() {
+        return driverClass;
+    }
+
+    public void setDriverClass(String driverClass) {
+        this.driverClass = driverClass;
+    }
+
+    @XmlElement(name = "userName", nillable = false)
+    public String getUserName() {
+        return userName;
+    }
+
+    public void setUserName(String userName) {
+        this.userName = userName;
+    }
+
+    @XmlElement(name = "pwd", nillable = false)
+    public String getPwd() {
+        return pwd;
+    }
+
+    public void setPwd(String pwd) {
+        this.pwd = pwd;
+    }
+
+    @XmlAttribute(name = "typeName")
+    public String getDbType() {
+        return dbType;
+    }
+
+    public void setDbType(String dbType) {
+        this.dbType = dbType;
+    }
+
+}

components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/common/TestDBConfigurations.java

@@ -0,0 +1,36 @@
+/*
+ * Copyright (c) 2014, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
+ * WSO2 Inc. 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.
+ * 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.wso2.carbon.device.mgt.core.common;
+
+import javax.xml.bind.annotation.XmlElement;
+import javax.xml.bind.annotation.XmlRootElement;
+import java.util.List;
+
+@XmlRootElement(name = "DeviceMgtTestDBConfigurations")
+public class TestDBConfigurations {
+
+    private List<TestDBConfiguration> dbTypesList;
+
+    @XmlElement(name = "DBType")
+    public List<TestDBConfiguration> getDbTypesList() {
+        return dbTypesList;
+    }
+
+    public void setDbTypesList(List<TestDBConfiguration> dbTypesList) {
+        this.dbTypesList = dbTypesList;
+    }
+
+}

components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/dao/DeviceDAOTest.java

@@ -0,0 +1,28 @@
+/*
+ * Copyright (c) 2014, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
+ * WSO2 Inc. 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.
+ * 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.wso2.carbon.device.mgt.core.dao;
+
+import org.testng.Assert;
+
+public class DeviceDAOTest {
+
+  //  @Test
+    public void setUp() throws Exception {
+   //     log.info("Testing started.");
+        Assert.assertEquals("A", "A");
+    }
+
+}

components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/dao/DeviceMgtDAOTestSuite.java

@@ -0,0 +1,171 @@
+/*
+ * Copyright (c) 2014, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
+ * WSO2 Inc. 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.
+ * 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.wso2.carbon.device.mgt.core.dao;
+
+import org.apache.commons.dbcp.BasicDataSource;
+import org.testng.Assert;
+import org.testng.annotations.BeforeClass;
+import org.testng.annotations.Parameters;
+import org.testng.annotations.Test;
+import org.w3c.dom.Document;
+import org.wso2.carbon.device.mgt.common.DeviceManagementException;
+import org.wso2.carbon.device.mgt.core.common.DBTypes;
+import org.wso2.carbon.device.mgt.core.common.TestDBConfiguration;
+import org.wso2.carbon.device.mgt.core.common.TestDBConfigurations;
+import org.wso2.carbon.device.mgt.core.dto.Device;
+import org.wso2.carbon.device.mgt.core.dto.DeviceType;
+import org.wso2.carbon.device.mgt.core.dto.Status;
+import org.wso2.carbon.device.mgt.core.util.DeviceManagerUtil;
+
+import javax.xml.bind.JAXBContext;
+import javax.xml.bind.JAXBException;
+import javax.xml.bind.Unmarshaller;
+import java.io.File;
+import java.sql.*;
+import java.util.Date;
+import java.util.Iterator;
+
+public class DeviceMgtDAOTestSuite {
+
+    private TestDBConfiguration testDBConfiguration;
+    private DeviceType devType = new DeviceType();
+    private Connection conn = null;
+    private Statement stmt = null;
+
+    @BeforeClass
+    @Parameters("dbType")
+    public void setUpDB(String dbTypeStr) throws Exception {
+
+        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());
+            DeviceManagementDAOFactory.init(testDataSource);
+        default:
+        }
+    }
+
+    private TestDBConfiguration getTestDBConfiguration(DBTypes dbType)
+            throws DeviceManagementDAOException, DeviceManagementException {
+
+        File deviceMgtConfig = new File("src/test/resources/testdbconfig.xml");
+        Document doc = null;
+        testDBConfiguration = null;
+        TestDBConfigurations testDBConfigurations = null;
+
+        doc = DeviceManagerUtil.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 DeviceManagementDAOException("Error parsing test db configurations", e);
+        }
+
+        Iterator<TestDBConfiguration> itrDBConfigs = testDBConfigurations.getDbTypesList().iterator();
+        while (itrDBConfigs.hasNext()) {
+            testDBConfiguration = itrDBConfigs.next();
+            if (testDBConfiguration.getDbType().equals(dbType.toString())) {
+                break;
+            }
+        }
+
+        return testDBConfiguration;
+    }
+
+    private void createH2DB(TestDBConfiguration testDBConf) throws Exception {
+
+        Class.forName(testDBConf.getDriverClass());
+        conn = DriverManager.getConnection(testDBConf.getConnectionUrl());
+        stmt = conn.createStatement();
+        stmt.executeUpdate("RUNSCRIPT FROM './src/test/resources/sql/CreateH2TestDB.sql'");
+        stmt.close();
+        conn.close();
+
+    }
+
+    @Test
+    public void addDeviceTypeTest() throws DeviceManagementDAOException, DeviceManagementException {
+
+        DeviceTypeDAO deviceTypeMgtDAO = DeviceManagementDAOFactory.getDeviceTypeDAO();
+
+        devType.setName("IOS");
+
+        deviceTypeMgtDAO.addDeviceType(devType);
+        Long deviceTypeId = null;
+
+        try {
+            conn = DeviceManagementDAOFactory.getDataSource().getConnection();
+            stmt = conn.createStatement();
+            ResultSet resultSet = stmt.executeQuery("SELECT ID,NAME from DM_DEVICE_TYPE DType where DType.NAME='IOS'");
+
+            while (resultSet.next()) {
+                deviceTypeId = resultSet.getLong(1);
+            }
+            conn.close();
+        } catch (SQLException sqlEx) {
+            throw new DeviceManagementDAOException("error in fetch device type by name IOS", sqlEx);
+        }
+
+        Assert.assertNotNull(deviceTypeId, "Device Type Id is null");
+        devType.setId(deviceTypeId);
+
+    }
+
+    @Test(dependsOnMethods = { "addDeviceTypeTest" })
+    public void addDeviceTest() throws DeviceManagementDAOException, DeviceManagementException {
+
+        DeviceDAO deviceMgtDAO = DeviceManagementDAOFactory.getDeviceDAO();
+
+        Device device = new Device();
+
+        device.setDateOfEnrollment(new Date().getTime());
+        device.setDateOfLastUpdate(new Date().getTime());
+        device.setDescription("test description");
+        device.setStatus(Status.ACTIVE);
+        device.setDeviceIdentificationId("111");
+        device.setDeviceType(devType.getId().toString());
+        device.setOwnerId("111");
+        deviceMgtDAO.addDevice(device);
+
+        Long deviceId = null;
+        try {
+            conn = DeviceManagementDAOFactory.getDataSource().getConnection();
+            stmt = conn.createStatement();
+            ResultSet resultSet = stmt
+                    .executeQuery("SELECT ID from DM_DEVICE DEVICE where DEVICE.DEVICE_IDENTIFICATION='111'");
+
+            while (resultSet.next()) {
+                deviceId = resultSet.getLong(1);
+            }
+            conn.close();
+        } catch (SQLException sqlEx) {
+            throw new DeviceManagementDAOException("error in fetch device by device identification id", sqlEx);
+        }
+
+        Assert.assertNotNull(deviceId, "Device Id is null");
+    }
+
+}

components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/sql/CreateH2TestDB.sql

@@ -0,0 +1,24 @@
+CREATE TABLE IF NOT EXISTS DM_DEVICE_TYPE
+(
+     ID   INT auto_increment NOT NULL,
+     NAME VARCHAR(300) NULL DEFAULT NULL,
+     PRIMARY KEY (ID)
+);
+
+CREATE TABLE IF NOT EXISTS DM_DEVICE
+(
+     ID                    INT auto_increment NOT NULL,
+     DESCRIPTION           TEXT NULL DEFAULT NULL,
+     NAME                  VARCHAR(100) NULL DEFAULT NULL,
+     DATE_OF_ENROLLMENT    INTEGER NULL DEFAULT NULL,
+     DATE_OF_LAST_UPDATE   INTEGER NULL DEFAULT NULL,
+     OWNERSHIP             VARCHAR(45) NULL DEFAULT NULL,
+     STATUS                VARCHAR(15) NULL DEFAULT NULL,
+     DEVICE_TYPE_ID        INT(11) NULL DEFAULT NULL,
+     DEVICE_IDENTIFICATION VARCHAR(300) NULL DEFAULT NULL,
+     OWNER                 VARCHAR(45) NULL DEFAULT NULL,
+     TENANT_ID INTEGER DEFAULT 0,
+     PRIMARY KEY (ID),
+     CONSTRAINT fk_DM_DEVICE_DM_DEVICE_TYPE2 FOREIGN KEY (DEVICE_TYPE_ID )
+     REFERENCES DM_DEVICE_TYPE (ID ) ON DELETE NO ACTION ON UPDATE NO ACTION
+);

components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/testdbconfig.xml

@@ -0,0 +1,24 @@
+<?xml version="1.0" encoding="ISO-8859-1"?>
+<!--
+  ~ Copyright (c) 2014, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
+  ~ WSO2 Inc. 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.
+  ~ 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.
+  -->
+
+<DeviceMgtTestDBConfigurations>
+    <DBType typeName="H2">
+        <connectionurl>jdbc:h2:mem:cdm-test-db;DB_CLOSE_DELAY=-1</connectionurl>
+        <driverclass>org.h2.Driver</driverclass>
+        <userName></userName>
+        <pwd></pwd>
+    </DBType>
+</DeviceMgtTestDBConfigurations>

components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/testng.xml

@@ -0,0 +1,28 @@
+<!--
+  ~ Copyright (c) 2014, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
+  ~ WSO2 Inc. 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.
+  ~ 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.
+  -->
+
+<!DOCTYPE suite SYSTEM "http://testng.org/testng-1.0.dtd" >
+
+<suite name="EMM-core-initializer">
+    <parameter name="useDefaultListeners" value="false"/>
+
+    <test name="DAO Unit Tests" preserve-order="true">
+        <parameter name="dbType" value="H2"/>
+        <classes>
+
+            <class name="org.wso2.carbon.device.mgt.core.dao.DeviceMgtDAOTestSuite"/>
+        </classes>
+    </test>
+</suite>