|
|
|
@ -21,6 +21,7 @@ package org.wso2.iot.integration.samples;
|
|
|
|
|
import com.google.gson.JsonArray;
|
|
|
|
|
import com.google.gson.JsonObject;
|
|
|
|
|
import com.google.gson.JsonParser;
|
|
|
|
|
import com.google.gson.JsonPrimitive;
|
|
|
|
|
import junit.framework.Assert;
|
|
|
|
|
import org.apache.commons.httpclient.HttpStatus;
|
|
|
|
|
import org.apache.commons.logging.Log;
|
|
|
|
@ -35,10 +36,13 @@ import org.testng.annotations.Test;
|
|
|
|
|
import org.wso2.carbon.automation.engine.context.TestUserMode;
|
|
|
|
|
import org.wso2.carbon.automation.test.utils.http.client.HttpResponse;
|
|
|
|
|
import org.wso2.iot.integration.common.Constants;
|
|
|
|
|
import org.wso2.iot.integration.common.MqttSubscriberClient;
|
|
|
|
|
import org.wso2.iot.integration.common.PayloadGenerator;
|
|
|
|
|
import org.wso2.iot.integration.common.RestClient;
|
|
|
|
|
import org.wso2.iot.integration.common.TestBase;
|
|
|
|
|
|
|
|
|
|
import java.util.ArrayList;
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* This class tests the functionality of the virtual fire alarm.
|
|
|
|
|
*/
|
|
|
|
@ -46,7 +50,12 @@ public class VirtualFireAlarmTestCase extends TestBase {
|
|
|
|
|
private static Log log = LogFactory.getLog(VirtualFireAlarmTestCase.class);
|
|
|
|
|
public static String deviceId1;
|
|
|
|
|
public static String deviceId2;
|
|
|
|
|
public static String tenantDeviceId1;
|
|
|
|
|
public static String tenantDeviceId2;
|
|
|
|
|
public static long currentTime;
|
|
|
|
|
private String broker = "tcp://localhost:1886";
|
|
|
|
|
private String DEVICE_TYPE = "virtual_firealarm";
|
|
|
|
|
private RestClient restClient;
|
|
|
|
|
|
|
|
|
|
@Factory(dataProvider = "userModeProvider")
|
|
|
|
|
public VirtualFireAlarmTestCase(TestUserMode userMode) {
|
|
|
|
@ -55,10 +64,10 @@ public class VirtualFireAlarmTestCase extends TestBase {
|
|
|
|
|
|
|
|
|
|
@BeforeClass(alwaysRun = true)
|
|
|
|
|
public void initTest() throws Exception {
|
|
|
|
|
super.init(userMode);;
|
|
|
|
|
RestClient client = new RestClient(backendHTTPSURL, Constants.APPLICATION_JSON, accessTokenString);
|
|
|
|
|
super.init(userMode);
|
|
|
|
|
restClient = new RestClient(backendHTTPSURL, Constants.APPLICATION_JSON, accessTokenString);
|
|
|
|
|
if (userMode == TestUserMode.TENANT_ADMIN) {
|
|
|
|
|
HttpResponse response = client
|
|
|
|
|
HttpResponse response = restClient
|
|
|
|
|
.post(Constants.VirtualFireAlarmConstants.ANALYTICS_ARTIFACTS_DEPLOYMENT_ENDPOINT, "");
|
|
|
|
|
Assert.assertEquals(HttpStatus.SC_CREATED, response.getResponseCode());
|
|
|
|
|
// Time for deploying the carbon apps
|
|
|
|
@ -85,20 +94,24 @@ public class VirtualFireAlarmTestCase extends TestBase {
|
|
|
|
|
JsonArray jsonArray = new JsonParser().parse(response.getData()).getAsJsonObject().getAsJsonArray("devices");
|
|
|
|
|
Assert.assertEquals("Virtual fire alarm enrollment failed ", 2, jsonArray.size());
|
|
|
|
|
|
|
|
|
|
deviceId1 = jsonArray.get(0).getAsJsonObject().getAsJsonPrimitive("deviceIdentifier").getAsString();
|
|
|
|
|
deviceId2 = jsonArray.get(1).getAsJsonObject().getAsJsonPrimitive("deviceIdentifier").getAsString();
|
|
|
|
|
if (userMode != TestUserMode.TENANT_ADMIN) {
|
|
|
|
|
deviceId1 = jsonArray.get(0).getAsJsonObject().getAsJsonPrimitive("deviceIdentifier").getAsString();
|
|
|
|
|
deviceId2 = jsonArray.get(1).getAsJsonObject().getAsJsonPrimitive("deviceIdentifier").getAsString();
|
|
|
|
|
} else {
|
|
|
|
|
tenantDeviceId1 = jsonArray.get(0).getAsJsonObject().getAsJsonPrimitive("deviceIdentifier").getAsString();
|
|
|
|
|
tenantDeviceId2 = jsonArray.get(1).getAsJsonObject().getAsJsonPrimitive("deviceIdentifier").getAsString();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Test(description = "Test whether the publishing to a mqtt broker works fine without exceptions", dependsOnMethods =
|
|
|
|
|
{"testEnrollment"} )
|
|
|
|
|
public void testEventPublishing() throws Exception {
|
|
|
|
|
String DEVICE_TYPE = "virtual_firealarm";
|
|
|
|
|
|
|
|
|
|
String deviceId1 = userMode == TestUserMode.TENANT_ADMIN ? tenantDeviceId1 : VirtualFireAlarmTestCase.deviceId1;
|
|
|
|
|
String deviceId2 = userMode == TestUserMode.TENANT_ADMIN ? tenantDeviceId2 : VirtualFireAlarmTestCase.deviceId2;
|
|
|
|
|
// Publishing message as a device with simple agent (device 1)
|
|
|
|
|
String topic = automationContext.getContextTenant().getDomain() + "/" + DEVICE_TYPE + "/" + deviceId1 +
|
|
|
|
|
"/temperature";
|
|
|
|
|
int qos = 2;
|
|
|
|
|
String broker = "tcp://localhost:1886";
|
|
|
|
|
String clientId = deviceId1 + ":" + DEVICE_TYPE;
|
|
|
|
|
MemoryPersistence persistence = new MemoryPersistence();
|
|
|
|
|
MqttClient sampleClient = new MqttClient(broker, clientId, persistence);
|
|
|
|
@ -154,4 +167,35 @@ public class VirtualFireAlarmTestCase extends TestBase {
|
|
|
|
|
log.info("Mqtt Client is Disconnected");
|
|
|
|
|
currentTime = System.currentTimeMillis();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Test(description = "Test whether the policy publishing from the server to device works", dependsOnMethods =
|
|
|
|
|
{"testEnrollment"} )
|
|
|
|
|
public void testPolicyPublishing() throws Exception {
|
|
|
|
|
String deviceId2 = userMode == TestUserMode.TENANT_ADMIN ? tenantDeviceId2 : VirtualFireAlarmTestCase.deviceId2;
|
|
|
|
|
String topic = automationContext.getContextTenant().getDomain() + "/" + DEVICE_TYPE + "/" + deviceId2 + "/#";
|
|
|
|
|
String clientId = deviceId2 + ":" + DEVICE_TYPE;
|
|
|
|
|
HttpResponse response = restClient.post("/api/device-mgt/v1.0/policies", PayloadGenerator
|
|
|
|
|
.getJsonPayload(Constants.VirtualFireAlarmConstants.PAYLOAD_FILE,
|
|
|
|
|
Constants.VirtualFireAlarmConstants.POLICY_DATA).toString());
|
|
|
|
|
Assert.assertEquals("Policy creation for virtual fire alarm failed", HttpStatus.SC_CREATED,
|
|
|
|
|
response.getResponseCode());
|
|
|
|
|
JsonObject jsonObject = new JsonParser().parse(response.getData()).getAsJsonObject();
|
|
|
|
|
String policyId = jsonObject.getAsJsonPrimitive("id").getAsString();
|
|
|
|
|
JsonArray jsonArray = new JsonArray();
|
|
|
|
|
jsonArray.add(new JsonPrimitive(policyId));
|
|
|
|
|
response = restClient.post(Constants.VirtualFireAlarmConstants.ACTIVATE_POLICY_ENDPOINT,
|
|
|
|
|
jsonArray.toString());
|
|
|
|
|
Assert.assertEquals("Policy activation for virtual fire alarm failed", HttpStatus.SC_OK,
|
|
|
|
|
response.getResponseCode());
|
|
|
|
|
MqttSubscriberClient mqttSubscriberClient = new MqttSubscriberClient(broker, clientId, topic, accessToken);
|
|
|
|
|
response = restClient.put(Constants.VirtualFireAlarmConstants.APPLY_CHANGES_ENDPOINT, "");
|
|
|
|
|
Assert.assertEquals("Applying changes to policy for virtual fire alarm failed", HttpStatus.SC_OK,
|
|
|
|
|
response.getResponseCode());
|
|
|
|
|
// Allow some time for message delivery
|
|
|
|
|
Thread.sleep(20000);
|
|
|
|
|
ArrayList<MqttMessage> mqttMessages = mqttSubscriberClient.getMqttMessages();
|
|
|
|
|
Assert.assertEquals("Policy published message is not received by the mqtt listener. ", 1, mqttMessages.size());
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|