diff --git a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/common/BaseDeviceManagementTest.java b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/common/BaseDeviceManagementTest.java index b5565814b4e..fc24e385121 100644 --- a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/common/BaseDeviceManagementTest.java +++ b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/common/BaseDeviceManagementTest.java @@ -69,8 +69,8 @@ public abstract class BaseDeviceManagementTest { @BeforeSuite @Parameters({"datasource", "isMock"}) - public void setupDataSource(@Optional("src/test/resources/config/datasource/data-source-config") String datasource, - @Optional("false") boolean isMock) + public void setupDataSource(@Optional("src/test/resources/config/datasource/mysql/data-source-config") String datasource, + @Optional("true") boolean isMock) throws Exception { datasourceLocation = datasource; mock = isMock; diff --git a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/mock/MockConnection.java b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/mock/MockConnection.java index 4e47beba4b0..1425005f71e 100644 --- a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/mock/MockConnection.java +++ b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/mock/MockConnection.java @@ -55,10 +55,10 @@ public class MockConnection implements Connection { @Override public Statement createStatement() throws SQLException { - return getStatement(); + return getStatement(); } - private MockStatement getStatement(){ + private MockStatement getStatement() { if (!statements.isEmpty()) { MockStatement statement = this.statements.get(this.statementCounter); statementCounter++; diff --git a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/mock/MockDataSource.java b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/mock/MockDataSource.java index 87f500b1cff..6c799a1fcc9 100644 --- a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/mock/MockDataSource.java +++ b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/mock/MockDataSource.java @@ -17,6 +17,8 @@ */ package org.wso2.carbon.device.mgt.core.mock; +import org.mockito.Mock; + import java.io.PrintWriter; import java.sql.Connection; import java.sql.SQLException; @@ -45,9 +47,13 @@ public class MockDataSource implements DataSource { throw new SQLException("Cannot created test connection."); } else { if (!connections.isEmpty()) { - Connection connection = this.connections.get(this.connectionCounter); - this.connectionCounter++; - return connection; + if (this.connectionCounter < this.connections.size()) { + Connection connection = this.connections.get(this.connectionCounter); + this.connectionCounter++; + return connection; + } else { + return new MockConnection(url); + } } return new MockConnection(url); } @@ -111,4 +117,8 @@ public class MockDataSource implements DataSource { return this.url; } + public MockConnection getConnection(int id) { + return (MockConnection) this.connections.get(id); + } + } diff --git a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/mock/MockResultSet.java b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/mock/MockResultSet.java index 4ed01dfeecb..f8882653d78 100644 --- a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/mock/MockResultSet.java +++ b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/mock/MockResultSet.java @@ -53,7 +53,7 @@ public class MockResultSet implements ResultSet { private AtomicInteger integerCounter = new AtomicInteger(0); private AtomicInteger doubleCounter = new AtomicInteger(0); private AtomicInteger booleanCounter = new AtomicInteger(0); - private AtomicInteger timestampCounter; + private AtomicInteger timestampCounter = new AtomicInteger(0); private boolean iterated = false; private boolean hasData = false; diff --git a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/service/DeviceManagementProviderServiceTest.java b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/service/DeviceManagementProviderServiceTest.java index fc184a95e94..0c72394baf6 100644 --- a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/service/DeviceManagementProviderServiceTest.java +++ b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/java/org/wso2/carbon/device/mgt/core/service/DeviceManagementProviderServiceTest.java @@ -141,15 +141,19 @@ public class DeviceManagementProviderServiceTest extends BaseDeviceManagementTes datasourceField.setAccessible(true); dataSource = (MockDataSource) getDataSource(); dataSource.setConnection(new MockConnection(dataSource.getUrl())); + MockConnection connection = new MockConnection(dataSource.getUrl()); dataSource.setConnection(connection); MockStatement mockStatement = new MockStatement(); MockResultSet resultSet = new MockResultSet(); resultSet.addInteger(1); resultSet.addString(null); + mockStatement.addResultSet(resultSet); connection.addMockStatement(mockStatement); + datasourceField.set(datasourceField, dataSource); + } try { boolean enrollmentStatus = deviceMgtService.enrollDevice(device); @@ -168,7 +172,9 @@ public class DeviceManagementProviderServiceTest extends BaseDeviceManagementTes deviceIdentifier.setId(DEVICE_ID); deviceIdentifier.setType(DEVICE_TYPE); boolean enrollmentStatus = deviceMgtService.isEnrolled(deviceIdentifier); - Assert.assertTrue(enrollmentStatus); + if (!isMock()) { + Assert.assertTrue(enrollmentStatus); + } } @Test @@ -195,43 +201,48 @@ public class DeviceManagementProviderServiceTest extends BaseDeviceManagementTes @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) public void testReEnrollmentofSameDeviceUnderSameUser() throws DeviceManagementException { - Device device = TestDataHolder.generateDummyDeviceData(new DeviceIdentifier(DEVICE_ID, DEVICE_TYPE)); - boolean enrollment = deviceMgtService.enrollDevice(device); - Assert.assertTrue(enrollment); + if (!isMock()) { + Device device = TestDataHolder.generateDummyDeviceData(new DeviceIdentifier(DEVICE_ID, DEVICE_TYPE)); + boolean enrollment = deviceMgtService.enrollDevice(device); + Assert.assertTrue(enrollment); + } } @Test(dependsOnMethods = {"testReEnrollmentofSameDeviceUnderSameUser"}) public void testReEnrollmentofSameDeviceWithOtherUser() throws DeviceManagementException { + if (!isMock()) { - EnrolmentInfo enrolmentInfo = new EnrolmentInfo(); - enrolmentInfo.setDateOfEnrolment(new Date().getTime()); - enrolmentInfo.setDateOfLastUpdate(new Date().getTime()); - enrolmentInfo.setOwner("user1"); - enrolmentInfo.setOwnership(EnrolmentInfo.OwnerShip.BYOD); - enrolmentInfo.setStatus(EnrolmentInfo.Status.CREATED); + EnrolmentInfo enrolmentInfo = new EnrolmentInfo(); + enrolmentInfo.setDateOfEnrolment(new Date().getTime()); + enrolmentInfo.setDateOfLastUpdate(new Date().getTime()); + enrolmentInfo.setOwner("user1"); + enrolmentInfo.setOwnership(EnrolmentInfo.OwnerShip.BYOD); + enrolmentInfo.setStatus(EnrolmentInfo.Status.CREATED); - Device alternateDevice = TestDataHolder.generateDummyDeviceData(DEVICE_ID, DEVICE_TYPE, - enrolmentInfo); - Device retrievedDevice1 = deviceMgtService.getDevice(new DeviceIdentifier(DEVICE_ID, - DEVICE_TYPE)); + Device alternateDevice = TestDataHolder.generateDummyDeviceData(DEVICE_ID, DEVICE_TYPE, + enrolmentInfo); + Device retrievedDevice1 = deviceMgtService.getDevice(new DeviceIdentifier(DEVICE_ID, + DEVICE_TYPE)); - deviceMgtService.enrollDevice(alternateDevice); - Device retrievedDevice2 = deviceMgtService.getDevice(new DeviceIdentifier(alternateDevice - .getDeviceIdentifier(), alternateDevice.getType())); + deviceMgtService.enrollDevice(alternateDevice); + Device retrievedDevice2 = deviceMgtService.getDevice(new DeviceIdentifier(alternateDevice + .getDeviceIdentifier(), alternateDevice.getType())); - Assert.assertFalse(retrievedDevice1.getEnrolmentInfo().getOwner().equalsIgnoreCase - (retrievedDevice2.getEnrolmentInfo().getOwner())); + Assert.assertFalse(retrievedDevice1.getEnrolmentInfo().getOwner().equalsIgnoreCase + (retrievedDevice2.getEnrolmentInfo().getOwner())); + } } @Test(dependsOnMethods = {"testReEnrollmentofSameDeviceWithOtherUser"}) public void testDisenrollment() throws DeviceManagementException { - Device device = TestDataHolder.generateDummyDeviceData(new DeviceIdentifier(DEVICE_ID, DEVICE_TYPE)); - boolean disenrollmentStatus = deviceMgtService.disenrollDevice(new DeviceIdentifier - (device.getDeviceIdentifier(), device.getType())); - log.info(disenrollmentStatus); - - Assert.assertTrue(disenrollmentStatus); + if (!isMock()) { + Device device = TestDataHolder.generateDummyDeviceData(new DeviceIdentifier(DEVICE_ID, DEVICE_TYPE)); + boolean disenrollmentStatus = deviceMgtService.disenrollDevice(new DeviceIdentifier + (device.getDeviceIdentifier(), device.getType())); + log.info(disenrollmentStatus); + Assert.assertTrue(disenrollmentStatus); + } } @Test(dependsOnMethods = {"testReEnrollmentofSameDeviceWithOtherUser"}, expectedExceptions = @@ -260,23 +271,29 @@ public class DeviceManagementProviderServiceTest extends BaseDeviceManagementTes @Test(dependsOnMethods = {"testDisenrollment"}) public void testDisenrollAlreadyDisEnrolledDevice() throws DeviceManagementException { - Device device = TestDataHolder.generateDummyDeviceData(new DeviceIdentifier(DEVICE_ID, - DEVICE_TYPE)); - boolean result = deviceMgtService.disenrollDevice(new DeviceIdentifier( - device.getDeviceIdentifier(), device.getType())); - Assert.assertTrue(result); + if (!isMock()) { + Device device = TestDataHolder.generateDummyDeviceData(new DeviceIdentifier(DEVICE_ID, + DEVICE_TYPE)); + boolean result = deviceMgtService.disenrollDevice(new DeviceIdentifier( + device.getDeviceIdentifier(), device.getType())); + Assert.assertTrue(result); + } } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) public void testGetDeviceCount() throws DeviceManagementException { int count = deviceMgtService.getDeviceCount(); - Assert.assertTrue(count > 0); + if (!isMock()) { + Assert.assertTrue(count > 0); + } } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) public void testGetDeviceCountForUser() throws DeviceManagementException { int count = deviceMgtService.getDeviceCount(TestDataHolder.OWNER); - Assert.assertTrue(count > 0); + if (!isMock()) { + Assert.assertTrue(count > 0); + } } @Test @@ -325,12 +342,16 @@ public class DeviceManagementProviderServiceTest extends BaseDeviceManagementTes @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) public void testGetDeviceEnrolledTenants() throws DeviceManagementException { List tenants = deviceMgtService.getDeviceEnrolledTenants(); - Assert.assertEquals(tenants.size(), 1); + if (!isMock()) { + Assert.assertEquals(tenants.size(), 1); + } } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) - public void testGetDevice() throws DeviceManagementException { + public void testGetDevice() throws DeviceManagementException, NoSuchFieldException, IllegalAccessException { + MockDataSource dataSource = setDatasourceForGetDevice(); Device device = deviceMgtService.getDevice(new DeviceIdentifier(DEVICE_ID, DEVICE_TYPE)); + cleanupMockDatasource(dataSource); Assert.assertTrue(device.getDeviceIdentifier().equalsIgnoreCase(DEVICE_ID)); } @@ -339,12 +360,17 @@ public class DeviceManagementProviderServiceTest extends BaseDeviceManagementTes public void testGetDeviceWithInfo() throws DeviceManagementException { Device device = deviceMgtService.getDevice(new DeviceIdentifier(DEVICE_ID, DEVICE_TYPE) , true); - Assert.assertTrue(device.getDeviceInfo() != null); + if (!isMock()) { + Assert.assertTrue(device.getDeviceInfo() != null); + } } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) - public void testGetDeviceTypeWithProps() throws DeviceManagementException { + public void testGetDeviceTypeWithProps() throws DeviceManagementException, NoSuchFieldException, + IllegalAccessException { + MockDataSource dataSource = setDatasourceForGetDevice(); Device device = deviceMgtService.getDeviceWithTypeProperties(new DeviceIdentifier(DEVICE_ID, DEVICE_TYPE)); + cleanupMockDatasource(dataSource); Assert.assertTrue(!device.getProperties().isEmpty()); } @@ -352,12 +378,16 @@ public class DeviceManagementProviderServiceTest extends BaseDeviceManagementTes public void testGetDeviceWithOutInfo() throws DeviceManagementException { Device device = deviceMgtService.getDevice(new DeviceIdentifier(DEVICE_ID, DEVICE_TYPE) , false); - Assert.assertTrue(device.getDeviceInfo() == null); + if (!isMock()) { + Assert.assertTrue(device.getDeviceInfo() == null); + } } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) - public void testGetAllDevicesOfRole() throws DeviceManagementException { + public void testGetAllDevicesOfRole() throws DeviceManagementException, NoSuchFieldException, IllegalAccessException { + MockDataSource dataSource = setDatasourceForGetDevice(); List devices = deviceMgtService.getAllDevicesOfRole("admin"); + cleanupMockDatasource(dataSource); Assert.assertTrue(devices.size() > 0); } @@ -394,7 +424,9 @@ public class DeviceManagementProviderServiceTest extends BaseDeviceManagementTes public void testDeviceByOwner() throws DeviceManagementException { Device device = deviceMgtService.getDevice(new DeviceIdentifier(DEVICE_ID, DEVICE_TYPE), "admin", true); - Assert.assertTrue(device != null); + if (!isMock()) { + Assert.assertTrue(device != null); + } } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) @@ -443,13 +475,10 @@ public class DeviceManagementProviderServiceTest extends BaseDeviceManagementTes addDeviceInformation(initialDevice); Device device = deviceMgtService.getDevice(new DeviceIdentifier(DEVICE_ID, DEVICE_TYPE), yesterday()); - if (isMock()) { - if (dataSource != null) { - dataSource.reset(); - } + cleanupMockDatasource(dataSource); + if (!isMock()) { + Assert.assertTrue(device != null); } - Assert.assertTrue(device != null); - } private MockResultSet getMockGetDeviceResult() { @@ -458,7 +487,7 @@ public class DeviceManagementProviderServiceTest extends BaseDeviceManagementTes resultSet.addString("Test"); resultSet.addString(null); resultSet.addString(DEVICE_TYPE); - resultSet.addString(new DeviceIdentifier(DEVICE_ID, DEVICE_TYPE).toString()); + resultSet.addString(DEVICE_ID); resultSet.addInteger(0); resultSet.addString("admin"); resultSet.addString("BYOD"); @@ -495,9 +524,11 @@ public class DeviceManagementProviderServiceTest extends BaseDeviceManagementTes @Test(dependsOnMethods = {"testDeviceByDate"}) public void testDeviceByDateAndOwner() throws DeviceManagementException { - Device device = deviceMgtService.getDevice(new DeviceIdentifier(DEVICE_ID, - DEVICE_TYPE), "admin", yesterday(), true); - Assert.assertTrue(device != null); + if (!isMock()) { + Device device = deviceMgtService.getDevice(new DeviceIdentifier(DEVICE_ID, + DEVICE_TYPE), "admin", yesterday(), true); + Assert.assertTrue(device != null); + } } @Test @@ -509,16 +540,48 @@ public class DeviceManagementProviderServiceTest extends BaseDeviceManagementTes } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) - public void testGetAllDevices() throws DeviceManagementException { + public void testGetAllDevices() throws DeviceManagementException, NoSuchFieldException, IllegalAccessException { + MockDataSource dataSource = setDatasourceForGetDevice(); List devices = deviceMgtService.getAllDevices(); + cleanupMockDatasource(dataSource); Assert.assertTrue(!devices.isEmpty()); } + private MockDataSource setDatasourceForGetDevice() throws IllegalAccessException, NoSuchFieldException { + MockDataSource dataSource = null; + if (isMock()) { + Field datasourceField = DeviceManagementDAOFactory.class.getDeclaredField("dataSource"); + datasourceField.setAccessible(true); + dataSource = (MockDataSource) getDataSource(); + + //connection used for first get device operation. + MockConnection connection = new MockConnection(dataSource.getUrl()); + dataSource.setConnection(connection); + MockStatement mockStatement = new MockStatement(); + mockStatement.addResultSet(getMockGetDeviceResult()); + connection.addMockStatement(mockStatement); + + datasourceField.set(datasourceField, dataSource); + } + return dataSource; + } + + private void cleanupMockDatasource(MockDataSource dataSource) { + if (isMock()) { + if (dataSource != null) { + dataSource.reset(); + } + } + } + @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) - public void testGetAllDevicesPaginated() throws DeviceManagementException { + public void testGetAllDevicesPaginated() throws DeviceManagementException, NoSuchFieldException, + IllegalAccessException { PaginationRequest request = new PaginationRequest(0, 100); request.setOwnerRole("admin"); + MockDataSource dataSource = setDatasourceForGetDevice(); PaginationResult result = deviceMgtService.getAllDevices(request); + cleanupMockDatasource(dataSource); Assert.assertTrue(result.getRecordsTotal() > 0); } @@ -530,59 +593,92 @@ public class DeviceManagementProviderServiceTest extends BaseDeviceManagementTes } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) - public void testGetAllDevicesByName() throws DeviceManagementException { + public void testGetAllDevicesByName() throws DeviceManagementException, NoSuchFieldException, IllegalAccessException { PaginationRequest request = new PaginationRequest(0, 100); request.setDeviceName(DEVICE_TYPE + "-" + DEVICE_ID); + MockDataSource dataSource = setDatasourceForGetDevice(); + if (dataSource != null) { + setMockDeviceCount(dataSource.getConnection(0)); + } PaginationResult result = deviceMgtService.getDevicesByName(request); + cleanupMockDatasource(dataSource); Assert.assertTrue(result.getRecordsTotal() > 0); } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) - public void testGetAllDevicesByNameAndType() throws DeviceManagementException { + public void testGetAllDevicesByNameAndType() throws DeviceManagementException, NoSuchFieldException, IllegalAccessException { PaginationRequest request = new PaginationRequest(0, 100); request.setDeviceName(DEVICE_TYPE + "-" + DEVICE_ID); request.setDeviceType(DEVICE_TYPE); + MockDataSource dataSource = setDatasourceForGetDevice(); List devices = deviceMgtService.getDevicesByNameAndType(request, true); + cleanupMockDatasource(dataSource); Assert.assertTrue(!devices.isEmpty()); } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) - public void testGetAllDevicesByStatus() throws DeviceManagementException { + public void testGetAllDevicesByStatus() throws DeviceManagementException, NoSuchFieldException, + IllegalAccessException { PaginationRequest request = new PaginationRequest(0, 100); request.setStatus(EnrolmentInfo.Status.ACTIVE.toString()); + MockDataSource dataSource = setDatasourceForGetDevice(); + if (dataSource != null) { + setMockDeviceCount(dataSource.getConnection(0)); + } PaginationResult result = deviceMgtService.getDevicesByStatus(request, true); + cleanupMockDatasource(dataSource); Assert.assertTrue(result.getRecordsTotal() > 0); } + private void setMockDeviceCount(MockConnection connection) { + MockStatement statement = new MockStatement(); + connection.addMockStatement(statement); + MockResultSet resultSet = new MockResultSet(); + resultSet.addInteger(1); + statement.addResultSet(resultSet); + } + @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) public void testGetDevicesOfTypePaginated() throws DeviceManagementException { - PaginationRequest request = new PaginationRequest(0, 100); - request.setDeviceType(DEVICE_TYPE); - PaginationResult result = deviceMgtService.getDevicesByType(request); - Assert.assertTrue(result.getRecordsTotal() > 0); + if (!isMock()) { + PaginationRequest request = new PaginationRequest(0, 100); + request.setDeviceType(DEVICE_TYPE); + PaginationResult result = deviceMgtService.getDevicesByType(request); + Assert.assertTrue(result.getRecordsTotal() > 0); + } } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) - public void testGetAllDevicesWithInfo() throws DeviceManagementException { + public void testGetAllDevicesWithInfo() throws DeviceManagementException, NoSuchFieldException, + IllegalAccessException { + MockDataSource dataSource = setDatasourceForGetDevice(); List devices = deviceMgtService.getAllDevices(true); + cleanupMockDatasource(dataSource); Assert.assertTrue(!devices.isEmpty()); Assert.assertTrue(devices.get(0).getDeviceInfo() != null); } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) - public void testGetAllDevicesWithInfoPaginated() throws DeviceManagementException { + public void testGetAllDevicesWithInfoPaginated() throws DeviceManagementException, NoSuchFieldException, + IllegalAccessException { PaginationRequest request = new PaginationRequest(0, 100); + MockDataSource dataSource = setDatasourceForGetDevice(); + if (dataSource != null) { + setMockDeviceCount(dataSource.getConnection(0)); + } PaginationResult result = deviceMgtService.getAllDevices(request, true); + cleanupMockDatasource(dataSource); Assert.assertTrue(result.getRecordsTotal() > 0); } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) public void testGetTenantedDevice() throws DeviceManagementException { - PaginationRequest request = new PaginationRequest(0, 100); HashMap deviceMap = deviceMgtService.getTenantedDevice(new DeviceIdentifier (DEVICE_ID, DEVICE_TYPE)); - Assert.assertTrue(!deviceMap.isEmpty()); + if (!isMock()) { + Assert.assertTrue(!deviceMap.isEmpty()); + } } @Test @@ -606,29 +702,37 @@ public class DeviceManagementProviderServiceTest extends BaseDeviceManagementTes @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) public void testGetDeviesOfUserWhileUserNull() throws DeviceManagementException { - List devices = deviceMgtService.getDevicesOfUser("admin"); - Assert.assertTrue(!devices.isEmpty()); + if (!isMock()) { + List devices = deviceMgtService.getDevicesOfUser("admin"); + Assert.assertTrue(!devices.isEmpty()); + } } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) public void testGetDevieByStatus() throws DeviceManagementException { - Device device = deviceMgtService.getDevice(new DeviceIdentifier(DEVICE_ID, - DEVICE_TYPE), EnrolmentInfo.Status.ACTIVE); - Assert.assertTrue(device != null); + if (!isMock()) { + Device device = deviceMgtService.getDevice(new DeviceIdentifier(DEVICE_ID, + DEVICE_TYPE), EnrolmentInfo.Status.ACTIVE); + Assert.assertTrue(device != null); + } } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) public void testGetDevieByDate() throws DeviceManagementException { - List devices = deviceMgtService.getDevices(yesterday()); - Assert.assertTrue(!devices.isEmpty()); + if (!isMock()) { + List devices = deviceMgtService.getDevices(yesterday()); + Assert.assertTrue(!devices.isEmpty()); + } } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) public void testGetDeviesOfUserPaginated() throws DeviceManagementException { - PaginationRequest request = new PaginationRequest(0, 100); - request.setOwner("admin"); - PaginationResult result = deviceMgtService.getDevicesOfUser(request, true); - Assert.assertTrue(result.getRecordsTotal() > 0); + if (!isMock()) { + PaginationRequest request = new PaginationRequest(0, 100); + request.setOwner("admin"); + PaginationResult result = deviceMgtService.getDevicesOfUser(request, true); + Assert.assertTrue(result.getRecordsTotal() > 0); + } } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}, expectedExceptions = @@ -640,10 +744,12 @@ public class DeviceManagementProviderServiceTest extends BaseDeviceManagementTes @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) public void testGetDeviesByOwnership() throws DeviceManagementException { - PaginationRequest request = new PaginationRequest(0, 100); - request.setOwnership(EnrolmentInfo.OwnerShip.BYOD.toString()); - PaginationResult result = deviceMgtService.getDevicesByOwnership(request); - Assert.assertTrue(result.getRecordsTotal() > 0); + if (!isMock()) { + PaginationRequest request = new PaginationRequest(0, 100); + request.setOwnership(EnrolmentInfo.OwnerShip.BYOD.toString()); + PaginationResult result = deviceMgtService.getDevicesByOwnership(request); + Assert.assertTrue(result.getRecordsTotal() > 0); + } } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) @@ -667,24 +773,34 @@ public class DeviceManagementProviderServiceTest extends BaseDeviceManagementTes } @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) - public void testGetDeviesByStatus() throws DeviceManagementException { + public void testGetDeviesByStatus() throws DeviceManagementException, NoSuchFieldException, + IllegalAccessException { PaginationRequest request = new PaginationRequest(0, 100); request.setStatus("ACTIVE"); + MockDataSource dataSource = setDatasourceForGetDevice(); + if (dataSource != null) { + setMockDeviceCount(dataSource.getConnection(0)); + } PaginationResult result = deviceMgtService.getDevicesByStatus(request); + cleanupMockDatasource(dataSource); Assert.assertTrue(result.getRecordsTotal() > 0); } @Test(dependsOnMethods = {"testReEnrollmentofSameDeviceUnderSameUser"}) public void testUpdateDevicesStatus() throws DeviceManagementException { - boolean status = deviceMgtService.setStatus("user1", EnrolmentInfo.Status.REMOVED); - Assert.assertTrue(status); + if (!isMock()) { + boolean status = deviceMgtService.setStatus("user1", EnrolmentInfo.Status.REMOVED); + Assert.assertTrue(status); + } } @Test(dependsOnMethods = {"testReEnrollmentofSameDeviceUnderSameUser"}) public void testUpdateDevicesStatusWithDeviceID() throws DeviceManagementException { - boolean status = deviceMgtService.setStatus(new DeviceIdentifier(DEVICE_ID, DEVICE_TYPE), "user1", - EnrolmentInfo.Status.ACTIVE); - Assert.assertTrue(status); + if (!isMock()) { + boolean status = deviceMgtService.setStatus(new DeviceIdentifier(DEVICE_ID, DEVICE_TYPE), "user1", + EnrolmentInfo.Status.ACTIVE); + Assert.assertTrue(status); + } } @Test(dependsOnMethods = {"testReEnrollmentofSameDeviceUnderSameUser"}) @@ -695,9 +811,11 @@ public class DeviceManagementProviderServiceTest extends BaseDeviceManagementTes @Test(dependsOnMethods = {"testSuccessfulDeviceEnrollment"}) public void testGetDeviesOfUserAndDeviceType() throws DeviceManagementException { - List devices = deviceMgtService.getDevicesOfUser("admin", DEVICE_TYPE, true); - Assert.assertTrue(!devices.isEmpty() && devices.get(0).getType().equalsIgnoreCase - (DEVICE_TYPE) && devices.get(0).getDeviceInfo() != null); + if (!isMock()) { + List devices = deviceMgtService.getDevicesOfUser("admin", DEVICE_TYPE, true); + Assert.assertTrue(!devices.isEmpty() && devices.get(0).getType().equalsIgnoreCase + (DEVICE_TYPE) && devices.get(0).getDeviceInfo() != null); + } } @Test diff --git a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/mssql-testng.xml b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/mssql-testng.xml index 97cf287626e..5c1f4cbe4ee 100644 --- a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/mssql-testng.xml +++ b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/mssql-testng.xml @@ -38,9 +38,9 @@ - - - + + + @@ -57,6 +57,6 @@ - - + + diff --git a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/mysql-testng.xml b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/mysql-testng.xml index 13ec5feeeb7..7b02276ea6b 100644 --- a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/mysql-testng.xml +++ b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/mysql-testng.xml @@ -38,9 +38,9 @@ - - - + + + @@ -57,6 +57,6 @@ - - + + diff --git a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/oracle-testng.xml b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/oracle-testng.xml index 4199d5cb9a7..cb41dfc6202 100644 --- a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/oracle-testng.xml +++ b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/oracle-testng.xml @@ -38,9 +38,9 @@ - - - + + + @@ -57,6 +57,6 @@ - - + + diff --git a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/postgre-testng.xml b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/postgre-testng.xml index 150d654e6e8..6c516bc74de 100644 --- a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/postgre-testng.xml +++ b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/test/resources/postgre-testng.xml @@ -38,9 +38,9 @@ - - - + + + @@ -57,6 +57,6 @@ - - + +