forked from community/device-mgt-core
megala21
7 years ago
commit
2dabd4cfe0
@ -0,0 +1,110 @@
|
|||||||
|
/*
|
||||||
|
* 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.notification.mgt;
|
||||||
|
|
||||||
|
import org.apache.commons.logging.Log;
|
||||||
|
import org.apache.commons.logging.LogFactory;
|
||||||
|
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.notification.mgt.Notification;
|
||||||
|
import org.wso2.carbon.device.mgt.core.TestDeviceManagementService;
|
||||||
|
import org.wso2.carbon.device.mgt.core.authorization.DeviceAccessAuthorizationServiceImpl;
|
||||||
|
import org.wso2.carbon.device.mgt.core.common.TestDataHolder;
|
||||||
|
import org.wso2.carbon.device.mgt.core.config.DeviceConfigurationManager;
|
||||||
|
import org.wso2.carbon.device.mgt.core.internal.DeviceManagementDataHolder;
|
||||||
|
import org.wso2.carbon.device.mgt.core.internal.DeviceManagementServiceComponent;
|
||||||
|
import org.wso2.carbon.device.mgt.core.service.DeviceManagementProviderService;
|
||||||
|
import org.wso2.carbon.device.mgt.core.service.DeviceManagementProviderServiceImpl;
|
||||||
|
import org.wso2.carbon.device.mgt.core.service.GroupManagementProviderServiceImpl;
|
||||||
|
import org.wso2.carbon.registry.core.config.RegistryContext;
|
||||||
|
import org.wso2.carbon.registry.core.exceptions.RegistryException;
|
||||||
|
import org.wso2.carbon.registry.core.internal.RegistryDataHolder;
|
||||||
|
import org.wso2.carbon.registry.core.jdbc.realm.InMemoryRealmService;
|
||||||
|
import org.wso2.carbon.registry.core.service.RegistryService;
|
||||||
|
import org.wso2.carbon.user.core.service.RealmService;
|
||||||
|
import org.wso2.carbon.utils.multitenancy.MultitenantConstants;
|
||||||
|
|
||||||
|
import java.io.InputStream;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* This class is used to test NotificationManagementServiceImpl.
|
||||||
|
*/
|
||||||
|
public class NotificationManagementTests {
|
||||||
|
|
||||||
|
private static final Log log = LogFactory.getLog(NotificationManagementTests.class);
|
||||||
|
private static final String DEVICE_TYPE = "NOTIFICATION_TEST_DEVICE";
|
||||||
|
private static final String DEVICE_ID_PREFIX = "NOTIFICATION-TEST-DEVICE-ID-";
|
||||||
|
private static final int NO_OF_DEVICES = 10;
|
||||||
|
private List<DeviceIdentifier> deviceIds = new ArrayList<>();
|
||||||
|
|
||||||
|
@BeforeClass
|
||||||
|
public void init() throws Exception {
|
||||||
|
DeviceConfigurationManager.getInstance().initConfig();
|
||||||
|
log.info("Initializing");
|
||||||
|
for (int i = 0; i < NO_OF_DEVICES; i++) {
|
||||||
|
deviceIds.add(new DeviceIdentifier(DEVICE_ID_PREFIX + i, DEVICE_TYPE));
|
||||||
|
}
|
||||||
|
List<Device> devices = TestDataHolder.generateDummyDeviceData(this.deviceIds);
|
||||||
|
DeviceManagementProviderService deviceMgtService = new DeviceManagementProviderServiceImpl();
|
||||||
|
DeviceManagementServiceComponent.notifyStartupListeners();
|
||||||
|
DeviceManagementDataHolder.getInstance().setDeviceManagementProvider(deviceMgtService);
|
||||||
|
DeviceManagementDataHolder.getInstance().setRegistryService(getRegistryService());
|
||||||
|
DeviceManagementDataHolder.getInstance().setDeviceAccessAuthorizationService(new DeviceAccessAuthorizationServiceImpl());
|
||||||
|
DeviceManagementDataHolder.getInstance().setGroupManagementProviderService(new GroupManagementProviderServiceImpl());
|
||||||
|
DeviceManagementDataHolder.getInstance().setDeviceTaskManagerService(null);
|
||||||
|
deviceMgtService.registerDeviceType(new TestDeviceManagementService(DEVICE_TYPE,
|
||||||
|
MultitenantConstants.SUPER_TENANT_DOMAIN_NAME));
|
||||||
|
for (Device device : devices) {
|
||||||
|
deviceMgtService.enrollDevice(device);
|
||||||
|
}
|
||||||
|
List<Device> returnedDevices = deviceMgtService.getAllDevices(DEVICE_TYPE);
|
||||||
|
|
||||||
|
for (Device device : returnedDevices) {
|
||||||
|
if (!device.getDeviceIdentifier().startsWith(DEVICE_ID_PREFIX)) {
|
||||||
|
throw new Exception("Incorrect device with ID - " + device.getDeviceIdentifier() + " returned!");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private RegistryService getRegistryService() throws RegistryException {
|
||||||
|
RealmService realmService = new InMemoryRealmService();
|
||||||
|
RegistryDataHolder.getInstance().setRealmService(realmService);
|
||||||
|
DeviceManagementDataHolder.getInstance().setRealmService(realmService);
|
||||||
|
InputStream is = this.getClass().getClassLoader().getResourceAsStream("carbon-home/repository/conf/registry.xml");
|
||||||
|
RegistryContext context = RegistryContext.getBaseInstance(is, realmService);
|
||||||
|
context.setSetup(true);
|
||||||
|
return context.getEmbeddedRegistryService();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test(description = "Add notifications using addNotification method and check whether it returns true.")
|
||||||
|
public void addNotification() throws Exception {
|
||||||
|
for (int i = 0; i < NO_OF_DEVICES; i++) {
|
||||||
|
DeviceIdentifier testDeviceIdentifier = new DeviceIdentifier(DEVICE_ID_PREFIX + i, DEVICE_TYPE);
|
||||||
|
Notification notification = TestDataHolder.getNotification(i, "CHECKED",
|
||||||
|
testDeviceIdentifier.toString(), DEVICE_ID_PREFIX + i, 1, DEVICE_TYPE);
|
||||||
|
NotificationManagementServiceImpl notificationManagementService = new NotificationManagementServiceImpl();
|
||||||
|
Assert.assertTrue(notificationManagementService.addNotification(testDeviceIdentifier, notification));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@ -0,0 +1,121 @@
|
|||||||
|
/*
|
||||||
|
* 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.permission.mgt;
|
||||||
|
|
||||||
|
import org.apache.commons.logging.Log;
|
||||||
|
import org.apache.commons.logging.LogFactory;
|
||||||
|
import org.powermock.api.mockito.PowerMockito;
|
||||||
|
import org.powermock.core.classloader.annotations.PrepareForTest;
|
||||||
|
import org.testng.Assert;
|
||||||
|
import org.testng.IObjectFactory;
|
||||||
|
import org.testng.annotations.BeforeClass;
|
||||||
|
import org.testng.annotations.ObjectFactory;
|
||||||
|
import org.testng.annotations.Test;
|
||||||
|
import org.wso2.carbon.device.mgt.common.permission.mgt.Permission;
|
||||||
|
import org.wso2.carbon.device.mgt.common.permission.mgt.PermissionManagementException;
|
||||||
|
import org.wso2.carbon.device.mgt.common.permission.mgt.PermissionManagerService;
|
||||||
|
import org.wso2.carbon.registry.core.exceptions.RegistryException;
|
||||||
|
|
||||||
|
import java.util.Properties;
|
||||||
|
|
||||||
|
import static org.mockito.MockitoAnnotations.initMocks;
|
||||||
|
|
||||||
|
@PrepareForTest(PermissionUtils.class)
|
||||||
|
public class PermissionManagerServiceTest {
|
||||||
|
|
||||||
|
private static final Log log = LogFactory.getLog(PermissionManagerServiceTest.class);;
|
||||||
|
private static final String PERMISSION_URL = "permission/admin/device-mgt/test/testPermission";
|
||||||
|
private static final String PERMISSION_PATH = "permission/admin/device-mgt/test/testPermission";
|
||||||
|
private static final String PERMISSION_METHOD = "ui.execute";
|
||||||
|
private static final String PERMISSION_NAME = "Test Permission";
|
||||||
|
|
||||||
|
//For create properties to retrieve permission.
|
||||||
|
private static final String HTTP_METHOD = "HTTP_METHOD";
|
||||||
|
private static final String URL = "URL";
|
||||||
|
|
||||||
|
private Permission permission;
|
||||||
|
private PermissionManagerService permissionManagerService;
|
||||||
|
|
||||||
|
@ObjectFactory
|
||||||
|
public IObjectFactory getObjectFactory() {
|
||||||
|
return new org.powermock.modules.testng.PowerMockObjectFactory();
|
||||||
|
}
|
||||||
|
|
||||||
|
@BeforeClass
|
||||||
|
public void init() throws RegistryException {
|
||||||
|
initMocks(this);
|
||||||
|
permissionManagerService = PermissionManagerServiceImpl.getInstance();
|
||||||
|
this.permission = new Permission();
|
||||||
|
permission.setName(PERMISSION_NAME);
|
||||||
|
permission.setPath(PERMISSION_PATH);
|
||||||
|
permission.setMethod(PERMISSION_METHOD);
|
||||||
|
permission.setUrl(PERMISSION_URL);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test (description = "Create a new permission in the permission tree.")
|
||||||
|
public void testCreatePermission() {
|
||||||
|
try {
|
||||||
|
PowerMockito.mockStatic(PermissionUtils.class);
|
||||||
|
PowerMockito.when(PermissionUtils.putPermission(permission)).thenReturn(true);
|
||||||
|
|
||||||
|
Assert.assertTrue(permissionManagerService.addPermission(permission));
|
||||||
|
} catch (PermissionManagementException e) {
|
||||||
|
log.error("Error creating permission " + e.getErrorMessage());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test (dependsOnMethods = {"testCreatePermission"}, description = "Test for retrieving the created permission " +
|
||||||
|
"from the permission tree.")
|
||||||
|
public void testGetPermission() throws PermissionManagementException {
|
||||||
|
Permission permission = permissionManagerService.getPermission(createProperties());
|
||||||
|
|
||||||
|
Assert.assertEquals(permission.getMethod(), PERMISSION_METHOD);
|
||||||
|
Assert.assertEquals(permission.getName(), PERMISSION_NAME);
|
||||||
|
Assert.assertEquals(permission.getPath(), PERMISSION_PATH);
|
||||||
|
Assert.assertEquals(permission.getUrl(), PERMISSION_URL);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test (dependsOnMethods = {"testCreatePermission"},
|
||||||
|
expectedExceptions = {PermissionManagementException.class},
|
||||||
|
expectedExceptionsMessageRegExp = "Resource URI/HTTP method is empty")
|
||||||
|
public void testGetPermissionError() throws PermissionManagementException {
|
||||||
|
Permission permission = permissionManagerService.getPermission(createErrorProperty());
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Create a Property object which will be passed to getPermission method to retrieve a permission.
|
||||||
|
* @return : Property object which contains permission url and method.
|
||||||
|
* */
|
||||||
|
private Properties createProperties() {
|
||||||
|
Properties properties = new Properties();
|
||||||
|
properties.setProperty(URL, PERMISSION_URL);
|
||||||
|
properties.setProperty(HTTP_METHOD, PERMISSION_METHOD);
|
||||||
|
return properties;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Creates property object with empty properties.
|
||||||
|
* @return : Properties object with empty set of properties.
|
||||||
|
* */
|
||||||
|
private Properties createErrorProperty() {
|
||||||
|
Properties properties = new Properties();
|
||||||
|
properties.setProperty(URL, "");
|
||||||
|
properties.setProperty(HTTP_METHOD, "");
|
||||||
|
return properties;
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in new issue