Search in sources :

Example 1 with ZigBeeOtaStatusCallback

use of com.zsmartsystems.zigbee.app.otaserver.ZigBeeOtaStatusCallback in project com.zsmartsystems.zigbee by zsmartsystems.

the class ZigBeeOtaServerTest method testNotify.

@Test
public void testNotify() {
    ArgumentCaptor<ZigBeeCommand> mockedCommandCaptor = ArgumentCaptor.forClass(ZigBeeCommand.class);
    NodeDescriptor nodeDescriptor = new NodeDescriptor();
    IeeeAddress ieeeAddress = new IeeeAddress("1234567890ABCDEF");
    ZigBeeEndpointAddress networkAddress = new ZigBeeEndpointAddress(1234, 56);
    ZigBeeNetworkManager mockedNetworkManager = Mockito.mock(ZigBeeNetworkManager.class);
    ZigBeeNode node = new ZigBeeNode(mockedNetworkManager, ieeeAddress);
    node.setNetworkAddress(networkAddress.getAddress());
    node.setNodeDescriptor(nodeDescriptor);
    ZigBeeEndpoint endpoint = new ZigBeeEndpoint(mockedNetworkManager, node, networkAddress.getEndpoint());
    // device.setIeeeAddress(ieeeAddress);
    List<Integer> outClusters = new ArrayList<Integer>();
    outClusters.add(ZclOtaUpgradeCluster.CLUSTER_ID);
    endpoint.setOutputClusterIds(outClusters);
    ZigBeeOtaStatusCallback mockedCallback = Mockito.mock(ZigBeeOtaStatusCallback.class);
    otaStatusCapture = new ArrayList<ZigBeeOtaServerStatus>();
    Set<ZigBeeEndpoint> devices = new HashSet<ZigBeeEndpoint>();
    devices.add(endpoint);
    Mockito.when(mockedNetworkManager.getNode((IeeeAddress) Matchers.anyObject())).thenReturn(node);
    // Mockito.when(mockedNetworkManager.getDevice((ZigBeeAddress) Matchers.anyObject())).thenReturn(endpoint);
    // Mockito.when(mockedNetworkManager.getNodeDevices((IeeeAddress) Matchers.anyObject())).thenReturn(devices);
    // ZigBeeTransportTransmit mockedTransport = Mockito.mock(ZigBeeTransportTransmit.class);
    // ArgumentCaptor<ZigBeeApsFrame> mockedApsFrameListener = ArgumentCaptor.forClass(ZigBeeApsFrame.class);
    // try {
    // Mockito.doNothing().when(mockedTransport).sendCommand(mockedApsFrameListener.capture());
    // } catch (ZigBeeException e) {
    // e.printStackTrace();
    // }
    Mockito.doAnswer(new Answer<Integer>() {

        @Override
        public Integer answer(InvocationOnMock invocation) {
            return 0;
        }
    }).when(mockedNetworkManager).sendCommand(mockedCommandCaptor.capture());
    Mockito.doAnswer(new Answer<Future<CommandResult>>() {

        @Override
        public Future<CommandResult> answer(InvocationOnMock invocation) {
            return null;
        }
    }).when(mockedNetworkManager).unicast(mockedCommandCaptor.capture(), (ZigBeeTransactionMatcher) Matchers.anyObject());
    ZclOtaUpgradeCluster cluster = new ZclOtaUpgradeCluster(mockedNetworkManager, endpoint);
    ZigBeeOtaServer server = new ZigBeeOtaServer();
    assertTrue(server.appStartup(cluster));
    server.addListener(this);
    ZigBeeOtaFile otaFile = Mockito.mock(ZigBeeOtaFile.class);
    // Set the firmware and send notification
    server.setFirmware(otaFile);
    assertEquals(1, mockedCommandCaptor.getAllValues().size());
    org.awaitility.Awaitility.await().until(otaListenerUpdated(), org.hamcrest.Matchers.equalTo(1));
    assertEquals(1, otaStatusCapture.size());
    ZigBeeOtaServerStatus status = otaStatusCapture.get(0);
    assertEquals(ZigBeeOtaServerStatus.OTA_WAITING, status);
    ZigBeeCommand command = mockedCommandCaptor.getValue();
    assertEquals(Integer.valueOf(0x19), command.getClusterId());
    assertTrue(command instanceof ImageNotifyCommand);
    ImageNotifyCommand notifyCommand = (ImageNotifyCommand) command;
    assertEquals(new ZigBeeEndpointAddress(1234, 56), notifyCommand.getDestinationAddress());
    assertTrue(notifyCommand.getQueryJitter() >= 1 && notifyCommand.getQueryJitter() <= 100);
}
Also used : ZigBeeOtaFile(com.zsmartsystems.zigbee.app.otaserver.ZigBeeOtaFile) NodeDescriptor(com.zsmartsystems.zigbee.zdo.field.NodeDescriptor) ArrayList(java.util.ArrayList) ZigBeeNetworkManager(com.zsmartsystems.zigbee.ZigBeeNetworkManager) ZigBeeOtaStatusCallback(com.zsmartsystems.zigbee.app.otaserver.ZigBeeOtaStatusCallback) ZigBeeCommand(com.zsmartsystems.zigbee.ZigBeeCommand) ZigBeeEndpointAddress(com.zsmartsystems.zigbee.ZigBeeEndpointAddress) ZigBeeOtaServerStatus(com.zsmartsystems.zigbee.app.otaserver.ZigBeeOtaServerStatus) ZigBeeEndpoint(com.zsmartsystems.zigbee.ZigBeeEndpoint) HashSet(java.util.HashSet) ZigBeeOtaServer(com.zsmartsystems.zigbee.app.otaserver.ZigBeeOtaServer) ZigBeeNode(com.zsmartsystems.zigbee.ZigBeeNode) ZclOtaUpgradeCluster(com.zsmartsystems.zigbee.zcl.clusters.ZclOtaUpgradeCluster) IeeeAddress(com.zsmartsystems.zigbee.IeeeAddress) InvocationOnMock(org.mockito.invocation.InvocationOnMock) Future(java.util.concurrent.Future) ImageNotifyCommand(com.zsmartsystems.zigbee.zcl.clusters.otaupgrade.ImageNotifyCommand) Test(org.junit.Test)

Aggregations

IeeeAddress (com.zsmartsystems.zigbee.IeeeAddress)1 ZigBeeCommand (com.zsmartsystems.zigbee.ZigBeeCommand)1 ZigBeeEndpoint (com.zsmartsystems.zigbee.ZigBeeEndpoint)1 ZigBeeEndpointAddress (com.zsmartsystems.zigbee.ZigBeeEndpointAddress)1 ZigBeeNetworkManager (com.zsmartsystems.zigbee.ZigBeeNetworkManager)1 ZigBeeNode (com.zsmartsystems.zigbee.ZigBeeNode)1 ZigBeeOtaFile (com.zsmartsystems.zigbee.app.otaserver.ZigBeeOtaFile)1 ZigBeeOtaServer (com.zsmartsystems.zigbee.app.otaserver.ZigBeeOtaServer)1 ZigBeeOtaServerStatus (com.zsmartsystems.zigbee.app.otaserver.ZigBeeOtaServerStatus)1 ZigBeeOtaStatusCallback (com.zsmartsystems.zigbee.app.otaserver.ZigBeeOtaStatusCallback)1 ZclOtaUpgradeCluster (com.zsmartsystems.zigbee.zcl.clusters.ZclOtaUpgradeCluster)1 ImageNotifyCommand (com.zsmartsystems.zigbee.zcl.clusters.otaupgrade.ImageNotifyCommand)1 NodeDescriptor (com.zsmartsystems.zigbee.zdo.field.NodeDescriptor)1 ArrayList (java.util.ArrayList)1 HashSet (java.util.HashSet)1 Future (java.util.concurrent.Future)1 Test (org.junit.Test)1 InvocationOnMock (org.mockito.invocation.InvocationOnMock)1