components/basys.components/src/test/java/org/eclipse/basyx/regression/processengineconnector/tests/TestDeviceAdministrationShell.java

package org.eclipse.basyx.regression.processengineconnector.tests;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import java.util.Map;

import org.eclipse.basyx.aas.api.modelurn.ModelUrn;
import org.eclipse.basyx.aas.api.resources.IOperation;
import org.eclipse.basyx.aas.api.resources.ISubModel;
import org.eclipse.basyx.aas.backend.connected.ConnectedAssetAdministrationShellManager;
import org.eclipse.basyx.aas.backend.connected.aas.ConnectedAssetAdministrationShell;
import org.eclipse.basyx.aas.backend.connector.http.HTTPConnectorProvider;
import org.eclipse.basyx.aas.metamodel.hashmap.aas.descriptor.AASDescriptor;
import org.eclipse.basyx.aas.metamodel.hashmap.aas.descriptor.SubmodelDescriptor;
import org.eclipse.basyx.aas.metamodel.hashmap.aas.identifier.IdentifierType;
import org.eclipse.basyx.regression.support.server.context.ComponentsRegressionContext;
import org.eclipse.basyx.testsuite.support.backend.servers.AASHTTPServerResource;
import org.eclipse.basyx.testsuite.support.vab.stub.AASRegistryStub;
import org.junit.Before;
import org.junit.ClassRule;
import org.junit.Test;


/**
 * 
 * 
 * @author zhangzai
 *
 */
public class TestDeviceAdministrationShell {
	private  ConnectedAssetAdministrationShellManager manager;
	ConnectedAssetAdministrationShell connectedAAS;
	/**
	 * Makes sure Tomcat Server is started
	 */
	@ClassRule
	public static AASHTTPServerResource res = new AASHTTPServerResource(new ComponentsRegressionContext());

	@Before
	public void setupConnection() {
		AASRegistryStub registry = new AASRegistryStub();
		ModelUrn coilcarUrn = new ModelUrn("coilcar");
		AASDescriptor ccDescriptor = new AASDescriptor("coilcar", IdentifierType.URI,
				"http://localhost:8080/basys.components/Testsuite/Processengine/coilcar/");
		SubmodelDescriptor smDescriptor = new SubmodelDescriptor("submodel1", IdentifierType.URI,
				"http://localhost:8080/basys.components/Testsuite/Processengine/coilcar/");
		ccDescriptor.addSubmodelDescriptor(smDescriptor);
		registry.register(coilcarUrn, ccDescriptor);
		
		//set-up the administration shell manager to create connected aas
		manager = new ConnectedAssetAdministrationShellManager(registry, new HTTPConnectorProvider());
		
		// create the connected AAS using the manager
		try {
			connectedAAS = manager.retrieveAAS(coilcarUrn);
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
	
	@Test
	public void operationTest() throws Exception {
		assertEquals(1, connectedAAS.getSubModels().size());

		// Check if the contained SubModel id is as expected
		assertTrue(connectedAAS.getSubModels().containsKey("submodel1"));
		ISubModel sm = connectedAAS
				.getSubModels()
				.get("submodel1");
		
		Map<String, IOperation> operations = sm.getOperations();
		assertEquals(2, operations.size());
		
		IOperation op1 = operations.get("liftTo");
		op1.invoke(5);
		IOperation op2 = operations.get("moveTo");
		op2.invoke(55);
	}
}