forked from community/device-mgt-core
Merge pull request #999 from pasindujw/master
Unit tests for GeoLocationProviderService.revert-70aa11f8
Megala Uthayakumar
7 years ago
committed by
GitHub
commit
121936f47f
@ -0,0 +1,244 @@
|
||||
/*
|
||||
* Copyright (c) 2017, 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.0
|
||||
*
|
||||
* 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.geo.service;
|
||||
|
||||
import org.mockito.Mockito;
|
||||
|
||||
import org.testng.Assert;
|
||||
import org.testng.annotations.BeforeClass;
|
||||
import org.testng.annotations.Test;
|
||||
import org.wso2.carbon.device.mgt.common.Device;
|
||||
import org.wso2.carbon.device.mgt.common.DeviceIdentifier;
|
||||
import org.wso2.carbon.device.mgt.common.DeviceManagementConstants;
|
||||
import org.wso2.carbon.device.mgt.common.DeviceManagementException;
|
||||
import org.wso2.carbon.device.mgt.common.geo.service.Alert;
|
||||
import org.wso2.carbon.device.mgt.common.geo.service.GeoFence;
|
||||
import org.wso2.carbon.device.mgt.common.geo.service.GeoLocationBasedServiceException;
|
||||
import org.wso2.carbon.device.mgt.core.TestDeviceManagementService;
|
||||
import org.wso2.carbon.device.mgt.core.common.TestDataHolder;
|
||||
import org.wso2.carbon.device.mgt.core.internal.DeviceManagementDataHolder;
|
||||
import org.wso2.carbon.device.mgt.core.service.DeviceManagementProviderService;
|
||||
import org.wso2.carbon.event.processor.stub.EventProcessorAdminServiceStub;
|
||||
import org.wso2.carbon.identity.jwt.client.extension.exception.JWTClientException;
|
||||
import org.wso2.carbon.utils.multitenancy.MultitenantConstants;
|
||||
|
||||
import java.rmi.RemoteException;
|
||||
import java.util.List;
|
||||
|
||||
public class GeoLocationProviderServiceTest {
|
||||
|
||||
private static final String DEVICE_TYPE = "GL_TEST_TYPE";
|
||||
private static final String DEVICE_ID = "GL-TEST-DEVICE-ID-1";
|
||||
private static final String SAMPLE_GEO_JSON = "12121";
|
||||
private static final String SAMPLE_AREA_NAME = "CUSTOM_NAME";
|
||||
private static final String SAMPLE_QUERY_NAME = "QUERY_NAME";
|
||||
private static final String SAMPLE_PROXIMITY_DISATANCE = "100";
|
||||
private static final String SAMPLE_PROXIMITY_TIME = "50";
|
||||
private static final String SAMPLE_SPEED_ALERT_VALUE = "120";
|
||||
private static final String SAMPLE_STATIONARY_TIME = "1500";
|
||||
private static final String SAMPLE_FLUCTUATION_RADIUS = "2000";
|
||||
|
||||
private EventProcessorAdminServiceStub mockEventProcessorAdminServiceStub;
|
||||
private GeoLocationProviderServiceImpl geoLocationProviderServiceImpl;
|
||||
|
||||
@BeforeClass
|
||||
public void init() throws Exception {
|
||||
initMocks();
|
||||
enrollDevice();
|
||||
}
|
||||
|
||||
@Test (description = "Create a sample geo exit-alert with relevant details.")
|
||||
public void createGeoExitAlert() throws GeoLocationBasedServiceException {
|
||||
Boolean result = geoLocationProviderServiceImpl.
|
||||
createGeoAlert(getExitAlert(), getDeviceIdentifier(), DeviceManagementConstants.GeoServices.ALERT_TYPE_EXIT);
|
||||
Assert.assertEquals(result, Boolean.TRUE);
|
||||
}
|
||||
|
||||
@Test (description = "Create a sample geo within-alert with relevant details.")
|
||||
public void createGeoWithinAlert() throws GeoLocationBasedServiceException {
|
||||
Boolean result = geoLocationProviderServiceImpl.
|
||||
createGeoAlert(getWithinAlert(), getDeviceIdentifier(), DeviceManagementConstants.GeoServices.ALERT_TYPE_WITHIN);
|
||||
Assert.assertEquals(result, Boolean.TRUE);
|
||||
}
|
||||
|
||||
@Test (description = "Create a sample geo proximity-alert with relevant details.")
|
||||
public void createGeoProximityAlert() throws GeoLocationBasedServiceException {
|
||||
Boolean result = geoLocationProviderServiceImpl.
|
||||
createGeoAlert(getProximityAlert(), getDeviceIdentifier(), DeviceManagementConstants.GeoServices.ALERT_TYPE_PROXIMITY);
|
||||
Assert.assertEquals(result, Boolean.TRUE);
|
||||
}
|
||||
|
||||
@Test (description = "Create a sample geo speed-alert with relevant details.")
|
||||
public void createGeoSpeedAlert() throws GeoLocationBasedServiceException {
|
||||
Boolean result = geoLocationProviderServiceImpl.
|
||||
createGeoAlert(getSpeedAlert(), getDeviceIdentifier(), DeviceManagementConstants.GeoServices.ALERT_TYPE_SPEED);
|
||||
Assert.assertEquals(result, Boolean.TRUE);
|
||||
}
|
||||
|
||||
@Test (description = "Create a sample geo stationary-alert with relevant details.")
|
||||
public void createGeoStationaryAlert() throws GeoLocationBasedServiceException {
|
||||
Boolean result = geoLocationProviderServiceImpl.
|
||||
createGeoAlert(getStationaryAlert(), getDeviceIdentifier(), DeviceManagementConstants.GeoServices.ALERT_TYPE_STATIONARY);
|
||||
Assert.assertEquals(result, Boolean.TRUE);
|
||||
}
|
||||
|
||||
@Test (description = "Create a sample geo traffic-alert with relevant details.")
|
||||
public void createGeoTrafficAlert() throws GeoLocationBasedServiceException {
|
||||
Boolean result = geoLocationProviderServiceImpl.
|
||||
createGeoAlert(getTrafficAlert(), getDeviceIdentifier(), DeviceManagementConstants.GeoServices.ALERT_TYPE_TRAFFIC);
|
||||
Assert.assertEquals(result, Boolean.TRUE);
|
||||
}
|
||||
|
||||
@Test(dependsOnMethods = "createGeoSpeedAlert", description = "retrieve saved geo speed-alert.")
|
||||
public void getGeoSpeedAlerts() throws GeoLocationBasedServiceException {
|
||||
String result;
|
||||
result = geoLocationProviderServiceImpl.getSpeedAlerts(getDeviceIdentifier());
|
||||
Assert.assertNotNull(result);
|
||||
Assert.assertEquals(result, "{'speedLimit':" + SAMPLE_SPEED_ALERT_VALUE + "}");
|
||||
}
|
||||
|
||||
@Test(dependsOnMethods = "createGeoTrafficAlert" , description = "retrieve saved geo exit-alert.")
|
||||
public void getGeoTrafficAlerts() throws GeoLocationBasedServiceException {
|
||||
List<GeoFence> geoFences;
|
||||
geoFences = geoLocationProviderServiceImpl.getTrafficAlerts(getDeviceIdentifier());
|
||||
Assert.assertNotNull(geoFences);
|
||||
GeoFence geoFenceNode = geoFences.get(0);
|
||||
Assert.assertEquals(geoFenceNode.getGeoJson(), "{\n" +
|
||||
" \"" + DeviceManagementConstants.GeoServices.GEO_FENCE_GEO_JSON + "\": \"" + SAMPLE_GEO_JSON + "\"\n" +
|
||||
"}");
|
||||
}
|
||||
|
||||
@Test(dependsOnMethods = "createGeoStationaryAlert", description = "retrieve saved geo stationary-alert.")
|
||||
public void getGeoStationaryAlerts() throws GeoLocationBasedServiceException {
|
||||
List<GeoFence> geoFences;
|
||||
geoFences = geoLocationProviderServiceImpl.getStationaryAlerts(getDeviceIdentifier());
|
||||
Assert.assertNotNull(geoFences);
|
||||
GeoFence geoFenceNode = geoFences.get(0);
|
||||
Assert.assertEquals(geoFenceNode.getAreaName(), SAMPLE_AREA_NAME);
|
||||
Assert.assertEquals(geoFenceNode.getQueryName(), SAMPLE_QUERY_NAME);
|
||||
Assert.assertEquals(geoFenceNode.getStationaryTime(), SAMPLE_STATIONARY_TIME);
|
||||
}
|
||||
|
||||
private void initMocks() throws JWTClientException, RemoteException {
|
||||
mockEventProcessorAdminServiceStub = Mockito.mock(EventProcessorAdminServiceStub.class);
|
||||
geoLocationProviderServiceImpl = Mockito.mock(GeoLocationProviderServiceImpl.class, Mockito.CALLS_REAL_METHODS);
|
||||
Mockito.doReturn(mockEventProcessorAdminServiceStub).
|
||||
when(geoLocationProviderServiceImpl).getEventProcessorAdminServiceStub();
|
||||
Mockito.doReturn("success").
|
||||
when(mockEventProcessorAdminServiceStub).validateExecutionPlan(Mockito.anyString());
|
||||
}
|
||||
|
||||
private DeviceIdentifier getDeviceIdentifier() {
|
||||
DeviceIdentifier deviceIdentifier = new DeviceIdentifier();
|
||||
deviceIdentifier.setId("1234");
|
||||
deviceIdentifier.setType("TEST");
|
||||
return deviceIdentifier;
|
||||
}
|
||||
|
||||
private Alert getWithinAlert() {
|
||||
Alert alert = new Alert();
|
||||
alert.setDeviceId(DEVICE_ID);
|
||||
alert.setCepAction("CEP_ACTION");
|
||||
alert.setParseData("{\n" +
|
||||
" \" " + DeviceManagementConstants.GeoServices.GEO_FENCE_GEO_JSON + "\": \"" + SAMPLE_GEO_JSON + "\"\n" +
|
||||
"}");
|
||||
alert.setCustomName(SAMPLE_AREA_NAME);
|
||||
alert.setExecutionPlan("EXECUTION_PLAN");
|
||||
alert.setQueryName(SAMPLE_QUERY_NAME);
|
||||
return alert;
|
||||
}
|
||||
|
||||
private Alert getExitAlert() {
|
||||
Alert alert = new Alert();
|
||||
alert.setDeviceId(DEVICE_ID);
|
||||
alert.setQueryName(SAMPLE_QUERY_NAME);
|
||||
alert.setCustomName(SAMPLE_AREA_NAME);
|
||||
alert.setStationeryTime(SAMPLE_STATIONARY_TIME);
|
||||
alert.setFluctuationRadius(SAMPLE_FLUCTUATION_RADIUS);
|
||||
alert.setParseData("{\n" +
|
||||
" \" " + DeviceManagementConstants.GeoServices.GEO_FENCE_GEO_JSON + "\": \"" + SAMPLE_GEO_JSON + "\"\n" +
|
||||
"}");
|
||||
alert.setExecutionPlan("EXECUTION_PLAN");
|
||||
return alert;
|
||||
}
|
||||
|
||||
private Alert getProximityAlert() {
|
||||
Alert alert = new Alert();
|
||||
alert.setDeviceId(DEVICE_ID);
|
||||
alert.setProximityTime(SAMPLE_PROXIMITY_TIME);
|
||||
alert.setProximityDistance(SAMPLE_PROXIMITY_DISATANCE);
|
||||
alert.setParseData("{\n" +
|
||||
" \" " + DeviceManagementConstants.GeoServices.GEO_FENCE_GEO_JSON + "\": \"" + SAMPLE_GEO_JSON + "\"\n" +
|
||||
"}");
|
||||
return alert;
|
||||
}
|
||||
|
||||
private Alert getSpeedAlert() {
|
||||
Alert alert = new Alert();
|
||||
alert.setDeviceId(DEVICE_ID);
|
||||
alert.setParseData("{\n" +
|
||||
" \"" + DeviceManagementConstants.GeoServices.GEO_FENCE_GEO_JSON + "\": \"" + SAMPLE_GEO_JSON + "\",\n" +
|
||||
" \"" + DeviceManagementConstants.GeoServices.SPEED_ALERT_VALUE + "\": \"" + SAMPLE_SPEED_ALERT_VALUE + "\"\n" +
|
||||
"}");
|
||||
return alert;
|
||||
}
|
||||
|
||||
private Alert getStationaryAlert() {
|
||||
Alert alert = new Alert();
|
||||
alert.setDeviceId(DEVICE_ID);
|
||||
alert.setQueryName(SAMPLE_QUERY_NAME);
|
||||
alert.setCustomName(SAMPLE_AREA_NAME);
|
||||
alert.setStationeryTime(SAMPLE_STATIONARY_TIME);
|
||||
alert.setFluctuationRadius(SAMPLE_FLUCTUATION_RADIUS);
|
||||
alert.setParseData("{\n" +
|
||||
" \"" + DeviceManagementConstants.GeoServices.GEO_FENCE_GEO_JSON + "\": \"" + SAMPLE_GEO_JSON + "\"\n" +
|
||||
"}");
|
||||
return alert;
|
||||
}
|
||||
|
||||
private Alert getTrafficAlert() {
|
||||
Alert alert = new Alert();
|
||||
alert.setDeviceId(DEVICE_ID);
|
||||
alert.setParseData("{\n" +
|
||||
" \"" + DeviceManagementConstants.GeoServices.GEO_FENCE_GEO_JSON +"\": \"" + SAMPLE_GEO_JSON + "\"\n" +
|
||||
"}");
|
||||
alert.setCustomName(SAMPLE_AREA_NAME);
|
||||
alert.setExecutionPlan("EXECUTION_PLAN");
|
||||
alert.setQueryName(SAMPLE_QUERY_NAME);
|
||||
return alert;
|
||||
}
|
||||
|
||||
private void enrollDevice() throws Exception {
|
||||
DeviceIdentifier deviceIdentifier = new DeviceIdentifier(DEVICE_ID, DEVICE_TYPE);
|
||||
Device device = TestDataHolder.generateDummyDeviceData(deviceIdentifier);
|
||||
DeviceManagementProviderService deviceMgtService = DeviceManagementDataHolder.getInstance().
|
||||
getDeviceManagementProvider();
|
||||
deviceMgtService.registerDeviceType(new TestDeviceManagementService(DEVICE_TYPE,
|
||||
MultitenantConstants.SUPER_TENANT_DOMAIN_NAME));
|
||||
deviceMgtService.enrollDevice(device);
|
||||
|
||||
Device returnedDevice = deviceMgtService.getDevice(deviceIdentifier);
|
||||
|
||||
if (!returnedDevice.getDeviceIdentifier().equals(deviceIdentifier.getId())) {
|
||||
throw new Exception("Incorrect device with ID - " + device.getDeviceIdentifier() + " returned!");
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in new issue