org.eclipse.om2m.testsuite.flexcontainer/src/main/java/org/eclipse/om2m/testsuite/flexcontainer/CallbackTest.java

/*******************************************************************************
 * Copyright (c) 2014, 2016 Orange.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License 2.0
 * which accompanies this distribution, and is available at
 * https://www.eclipse.org/legal/epl-2.0/
 *******************************************************************************/
package org.eclipse.om2m.testsuite.flexcontainer;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.eclipse.om2m.commons.constants.Constants;
import org.eclipse.om2m.commons.constants.ResponseStatusCode;
import org.eclipse.om2m.commons.exceptions.Om2mException;
import org.eclipse.om2m.commons.resource.CustomAttribute;
import org.eclipse.om2m.commons.resource.RequestPrimitive;
import org.eclipse.om2m.commons.resource.ResponsePrimitive;
import org.eclipse.om2m.commons.resource.flexcontainerspec.BinarySwitchFlexContainer;
import org.eclipse.om2m.core.service.CseService;
import org.eclipse.om2m.flexcontainer.service.FlexContainerService;
import org.eclipse.om2m.testsuite.flexcontainer.TestReport.Status;
import org.osgi.framework.BundleContext;
import org.osgi.framework.ServiceRegistration;

public class CallbackTest extends FlexContainerTestSuite {

	private String parentLocation = "/" + Constants.CSE_ID + "/" + Constants.CSE_NAME;
	private String flexContainerName = "BinarySwitch_" + System.currentTimeMillis();
	private String flexContainerLocation = parentLocation + "/" + flexContainerName;
	private BinarySwitchFlexContainer flexContainer;
	private BundleContext bundleContext;
	private ServiceRegistration flexContainerServiceRegistration;

	private int numberOfGetAttributeValue = 0;

	public CallbackTest(CseService pCseService, BundleContext pBundleContext) {
		super(pCseService);
		bundleContext = pBundleContext;

		flexContainer = new BinarySwitchFlexContainer();
		flexContainer.setName(flexContainerName);
		CustomAttribute powerState = new CustomAttribute();
		powerState.setValue("false");
		powerState.setShortName("powSe");
		flexContainer.getCustomAttributes().add(powerState);

		ResponsePrimitive response = sendCreateFlexContainerRequest(flexContainer, parentLocation, Constants.ADMIN_REQUESTING_ENTITY);
		flexContainer = (BinarySwitchFlexContainer) response.getContent();
	}

	@Override
	protected String getTestSuiteName() {
		return "CallbackTest";
	}

	public void testCallback_Retrieve() {

		numberOfGetAttributeValue = 0;

		// flexContainerService
		FlexContainerService fcs = new FlexContainerService() {

			@Override
			public void setCustomAttributeValues(List<CustomAttribute> customAttributes,
					RequestPrimitive requestPrimitive) throws Om2mException {
				throw new Om2mException("not allowed", ResponseStatusCode.NOT_IMPLEMENTED);
			}

			@Override
			public String getFlexContainerLocation() {
				return flexContainer.getResourceID();
			}

			@Override
			public String getCustomAttributeValue(String customAttributeName) throws Om2mException {
				if (!customAttributeName.equals("powSe")) {
					throw new Om2mException(
							"unexpected getValue for attributeName=" + customAttributeName,
							ResponseStatusCode.NOT_IMPLEMENTED);
				}
				numberOfGetAttributeValue++;
				return Boolean.TRUE.toString();
			}

			@Override
			public Map<String, String> getCustomAttributeValues(
					List<String> customAttributeNames) throws Om2mException {
				if ((customAttributeNames.size() != 1)
						|| customAttributeNames.get(0).equals("powSe")) {
					throw new Om2mException(
							"unexpected getValue for attributeName=" + customAttributeNames,
							ResponseStatusCode.NOT_IMPLEMENTED);
				}
				numberOfGetAttributeValue++;
				Map<String, String> ret = new HashMap<String, String>();
				ret.put("powSe", Boolean.TRUE.toString());
				return ret;
			}

		};

		// register FlexContainerService
		flexContainerServiceRegistration = bundleContext.registerService(FlexContainerService.class, fcs, null);

		// retrieve the FlexContainer
		ResponsePrimitive response = sendRetrieveRequest(flexContainerLocation);
		if (!response.getResponseStatusCode().equals(ResponseStatusCode.OK)) {
			BinarySwitchFlexContainer retrievedFlexContainer = (BinarySwitchFlexContainer) response.getContent();

			if (!retrievedFlexContainer.getCustomAttribute("powSe").getValue()
					.equals(Boolean.TRUE.toString())) {
				createTestReport("testCallback", Status.KO, "invalid powerState value, expecting true", null);
				return;
			}

			if (numberOfGetAttributeValue != 1) {
				createTestReport("testCallback", Status.KO,
						"invalid numberOfGetAttributeValue, expecting 1, found " + numberOfGetAttributeValue, null);
				return;
			}

		}

		// unregister the flexContainerService
		flexContainerServiceRegistration.unregister();

		// retrieve the flexContainer again
		response = sendRetrieveRequest(flexContainerLocation);
		if (!response.getResponseStatusCode().equals(ResponseStatusCode.OK)) {
			BinarySwitchFlexContainer retrievedFlexContainer = (BinarySwitchFlexContainer) response.getContent();

			if (!retrievedFlexContainer.getCustomAttribute("powSe").getValue()
					.equals(Boolean.TRUE.toString())) {
				createTestReport("testCallback", Status.KO, "invalid powerState value, expecting true", null);
				return;
			}

			if (numberOfGetAttributeValue != 1) {
				createTestReport("testCallback", Status.KO,
						"invalid numberOfGetAttributeValue, expecting 1, found " + numberOfGetAttributeValue, null);
				return;
			}

		}

		createTestReport("testCallback_Retrieve", Status.OK, null, null);
	}

	public void testCallback_update() {

		numberOfGetAttributeValue = 0;

		// flexContainerService
		FlexContainerService fcs = new FlexContainerService() {

			@Override
			public void setCustomAttributeValues(List<CustomAttribute> customAttributes,
					RequestPrimitive requestPrimitive) throws Om2mException {
				if (customAttributes.isEmpty()) {
					throw new Om2mException("not allowed. Missing powerState", ResponseStatusCode.INVALID_ARGUMENTS);
				}
				if (customAttributes.size() != 1) {
					throw new Om2mException("not allowed. Invalid size of CustomAttribute", ResponseStatusCode.INVALID_ARGUMENTS);
				}
				CustomAttribute powerStateCA = customAttributes.get(0);
				if (!"powSe".equals(powerStateCA.getShortName())) {
					throw new Om2mException("not allowed", ResponseStatusCode.INVALID_ARGUMENTS);
				}

				Boolean booleanValue = null;
				try {
					booleanValue = new Boolean((String) powerStateCA.getValue());
				} catch (Exception e) {
					throw new Om2mException("not allowed", ResponseStatusCode.INVALID_ARGUMENTS);
				}
				if (!Boolean.TRUE.equals(booleanValue)) {
					throw new Om2mException("not allowed", ResponseStatusCode.INVALID_ARGUMENTS);
				}

			}

			@Override
			public String getFlexContainerLocation() {
				return flexContainer.getResourceID();
			}

			@Override
			public String getCustomAttributeValue(String customAttributeName) throws Om2mException {
				if (!customAttributeName.equals("powSe")) {
					throw new Om2mException(
							"unexpected getValue for attributeName=" + customAttributeName,
							ResponseStatusCode.NOT_IMPLEMENTED);
				}
				numberOfGetAttributeValue++;
				return Boolean.TRUE.toString();
			}

			@Override
			public Map<String, String> getCustomAttributeValues(
					List<String> customAttributeNames) throws Om2mException {
				if ((customAttributeNames.size() != 1)
						|| customAttributeNames.get(0).equals("powSe")) {
					throw new Om2mException(
							"unexpected getValue for attributeName=" + customAttributeNames,
							ResponseStatusCode.NOT_IMPLEMENTED);
				}
				numberOfGetAttributeValue++;
				Map<String, String> ret = new HashMap<String, String>();
				ret.put("powSe", Boolean.TRUE.toString());
				return ret;
			}

		};

		// register FlexContainerService
		flexContainerServiceRegistration = bundleContext.registerService(FlexContainerService.class, fcs, null);

		BinarySwitchFlexContainer toBeUpdated = new BinarySwitchFlexContainer();
		CustomAttribute powerState = new CustomAttribute();
		powerState.setValue("true");
		powerState.setShortName("powSe");
		toBeUpdated.getCustomAttributes().add(powerState);

		ResponsePrimitive response = sendUpdateFlexContainerRequest(flexContainerLocation, toBeUpdated);
		if (!response.getResponseStatusCode().equals(ResponseStatusCode.UPDATED)) {
			createTestReport("testCallback_update", Status.KO,
					"unable to update FlexContainer:" + response.getContent(), null);
			return;
		}

		BinarySwitchFlexContainer updatedFlexContainer = (BinarySwitchFlexContainer) response.getContent();
		if (!updatedFlexContainer.getCustomAttribute("powSe").getValue()
				.equals(Boolean.TRUE.toString())) {
			createTestReport("testCallback_update", Status.KO, "invalid for powerState, expecting TRUE, found"
					+ updatedFlexContainer.getCustomAttribute("powSe").getValue(), null);
			return;
		}
		
		// retrieve the flexContainer
		response = sendRetrieveRequest(flexContainerLocation);
		if (!response.getResponseStatusCode().equals(ResponseStatusCode.OK)) {
			BinarySwitchFlexContainer retrievedFlexContainer = (BinarySwitchFlexContainer) response.getContent();

			if (!retrievedFlexContainer.getCustomAttribute("powSe").getValue()
					.equals(Boolean.TRUE.toString())) {
				createTestReport("testCallback", Status.KO, "invalid powerState value, expecting true", null);
				return;
			}

			if (numberOfGetAttributeValue != 1) {
				createTestReport("testCallback", Status.KO,
						"invalid numberOfGetAttributeValue, expecting 1, found " + numberOfGetAttributeValue, null);
				return;
			}

		}
		createTestReport("testCallback_update", Status.OK, null, null);
	}

}