cloud/resourcedirectory/src/test/java/org/iotivity/cloud/testrdserver/DiscoveryResourceTest.java

   1 /*
   2  * //******************************************************************
   3  * //
   4  * // Copyright 2016 Samsung Electronics All Rights Reserved.
   5  * //
   6  * //-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=
   7  * //
   8  * // Licensed under the Apache License, Version 2.0 (the "License");
   9  * // you may not use this file except in compliance with the License.
  10  * // You may obtain a copy of the License at
  11  * //
  12  * //      http://www.apache.org/licenses/LICENSE-2.0
  13  * //
  14  * // Unless required by applicable law or agreed to in writing, software
  15  * // distributed under the License is distributed on an "AS IS" BASIS,
  16  * // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  * // See the License for the specific language governing permissions and
  18  * // limitations under the License.
  19  * //
  20  * //-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=
  21  */
  22 package org.iotivity.cloud.testrdserver;
  23
  24 import static java.util.concurrent.TimeUnit.SECONDS;
  25 import static org.junit.Assert.assertTrue;
  26 import static org.mockito.Mockito.mock;
  27
  28 import java.util.ArrayList;
  29 import java.util.HashMap;
  30 import java.util.concurrent.CountDownLatch;
  31
  32 import org.iotivity.cloud.base.device.CoapDevice;
  33 import org.iotivity.cloud.base.protocols.IRequest;
  34 import org.iotivity.cloud.base.protocols.IResponse;
  35 import org.iotivity.cloud.base.protocols.MessageBuilder;
  36 import org.iotivity.cloud.base.protocols.coap.CoapResponse;
  37 import org.iotivity.cloud.base.protocols.enums.RequestMethod;
  38 import org.iotivity.cloud.base.protocols.enums.ResponseStatus;
  39 import org.iotivity.cloud.rdserver.resources.directory.rd.ResourceDirectoryResource;
  40 import org.iotivity.cloud.rdserver.resources.directory.res.DiscoveryResource;
  41 import org.iotivity.cloud.util.Cbor;
  42 import org.junit.After;
  43 import org.junit.Before;
  44 import org.junit.Test;
  45 import org.mockito.Mockito;
  46 import org.mockito.invocation.InvocationOnMock;
  47 import org.mockito.stubbing.Answer;
  48
  49 public class DiscoveryResourceTest {
  50     private Cbor<ArrayList<Object>>   mCbor              = new Cbor<>();
  51     private ResourceDirectoryResource mRDResource        = null;
  52     private DiscoveryResource         mDiscoveryResource = null;
  53     private CoapDevice                mockDevice         = null;
  54     private CountDownLatch            mLatch             = null;
  55     private IResponse                 mResponse;
  56
  57     @Before
  58     public void setUp() throws Exception {
  59         mResponse = null;
  60         mockDevice = mock(CoapDevice.class);
  61         mLatch = new CountDownLatch(1);
  62         mRDResource = new ResourceDirectoryResource();
  63         mDiscoveryResource = new DiscoveryResource();
  64         // callback mock
  65         Mockito.doAnswer(new Answer<Object>() {
  66             @Override
  67             public CoapResponse answer(InvocationOnMock invocation)
  68                     throws Throwable {
  69                 CoapResponse resp = (CoapResponse) invocation.getArguments()[0];
  70                 mLatch.countDown();
  71                 mResponse = resp;
  72                 return resp;
  73             }
  74         }).when(mockDevice).sendResponse(Mockito.anyObject());
  75     }
  76
  77     @After
  78     public void tearDown() throws Exception {
  79         RDServerTestUtils.resetRDDatabase();
  80     }
  81
  82     @Test
  83     public void testHandleGetRequest_notExistVaule() throws Exception {
  84         IRequest request = MessageBuilder.createRequest(RequestMethod.GET,
  85                 RDServerTestUtils.DISCOVERY_REQ_URI,
  86                 "rt=core.light;di=" + RDServerTestUtils.DI);
  87         mDiscoveryResource.onDefaultRequestReceived(mockDevice, request);
  88         // assertion: if the response status is "NOT_FOUND"
  89         assertTrue(mLatch.await(2L, SECONDS));
  90         assertTrue(methodCheck(mResponse, ResponseStatus.NOT_FOUND));
  91     }
  92
  93     @Test
  94     public void testHandleGetRequest_existValue() throws Exception {
  95         IRequest request = MessageBuilder.createRequest(RequestMethod.GET,
  96                 RDServerTestUtils.DISCOVERY_REQ_URI,
  97                 "rt=core.light;di=" + RDServerTestUtils.DI);
  98         mRDResource.onDefaultRequestReceived(mockDevice,
  99                 RDServerTestUtils.makePublishRequest());
 100         mDiscoveryResource.onDefaultRequestReceived(mockDevice, request);
 101         // assertion: if the response status is "CONTENT"
 102         // assertion : if the payload contains resource info
 103         assertTrue(mLatch.await(2L, SECONDS));
 104         assertTrue(methodCheck(mResponse, ResponseStatus.CONTENT));
 105         assertTrue(discoverHashmapCheck(mResponse, "di"));
 106         assertTrue(discoveredResourceCheck(mResponse, "href"));
 107         assertTrue(discoveredResourceCheck(mResponse, "rt"));
 108         assertTrue(discoveredResourceCheck(mResponse, "if"));
 109     }
 110
 111     private boolean discoverHashmapCheck(IResponse response,
 112             String propertyName) {
 113         ArrayList<Object> resourceList = mCbor
 114                 .parsePayloadFromCbor(response.getPayload(), ArrayList.class);
 115         HashMap<Object, Object> firstArray = (HashMap<Object, Object>) resourceList
 116                 .get(0);
 117         if (firstArray.get(propertyName) != null)
 118             return true;
 119         else
 120             return false;
 121     }
 122
 123     private boolean discoveredResourceCheck(IResponse response,
 124             String propertyName) {
 125         ArrayList<Object> resourceList = mCbor
 126                 .parsePayloadFromCbor(response.getPayload(), ArrayList.class);
 127         HashMap<Object, Object> firstArray = (HashMap<Object, Object>) resourceList
 128                 .get(0);
 129         ArrayList<HashMap<Object, Object>> linkData = (ArrayList<HashMap<Object, Object>>) firstArray
 130                 .get("links");
 131         HashMap<Object, Object> linkMap = linkData.get(0);
 132         if (linkMap.get(propertyName) != null)
 133             return true;
 134         else
 135             return false;
 136     }
 137
 138     private boolean methodCheck(IResponse response,
 139             ResponseStatus responseStatus) {
 140         if (responseStatus == response.getStatus())
 141             return true;
 142         else
 143             return false;
 144     }
 145
 146     private boolean nullPayloadCheck(IResponse response) {
 147         ArrayList<Object> payloadData = mCbor
 148                 .parsePayloadFromCbor(mResponse.getPayload(), ArrayList.class);
 149         return (payloadData.isEmpty());
 150     }
 151 }