git.basschouten.com Git - openhab-addons.git/blob

   1 /**
   2  * Copyright (c) 2010-2021 Contributors to the openHAB project
   3  *
   4  * See the NOTICE file(s) distributed with this work for additional
   5  * information.
   6  *
   7  * This program and the accompanying materials are made available under the
   8  * terms of the Eclipse Public License 2.0 which is available at
   9  * http://www.eclipse.org/legal/epl-2.0
  10  *
  11  * SPDX-License-Identifier: EPL-2.0
  12  */
  13 package org.openhab.binding.mqtt.handler;
  14
  15 import static org.hamcrest.CoreMatchers.is;
  16 import static org.hamcrest.MatcherAssert.assertThat;
  17 import static org.junit.jupiter.api.Assertions.assertThrows;
  18 import static org.mockito.ArgumentMatchers.*;
  19 import static org.mockito.Mockito.*;
  20
  21 import java.util.concurrent.ScheduledExecutorService;
  22 import java.util.concurrent.ScheduledThreadPoolExecutor;
  23
  24 import org.junit.jupiter.api.AfterEach;
  25 import org.junit.jupiter.api.BeforeEach;
  26 import org.junit.jupiter.api.Test;
  27 import org.junit.jupiter.api.extension.ExtendWith;
  28 import org.mockito.ArgumentCaptor;
  29 import org.mockito.Mock;
  30 import org.mockito.junit.jupiter.MockitoExtension;
  31 import org.mockito.junit.jupiter.MockitoSettings;
  32 import org.mockito.quality.Strictness;
  33 import org.openhab.core.config.core.Configuration;
  34 import org.openhab.core.io.transport.mqtt.MqttBrokerConnection;
  35 import org.openhab.core.io.transport.mqtt.MqttConnectionState;
  36 import org.openhab.core.io.transport.mqtt.MqttException;
  37 import org.openhab.core.io.transport.mqtt.MqttService;
  38 import org.openhab.core.test.java.JavaTest;
  39 import org.openhab.core.thing.Bridge;
  40 import org.openhab.core.thing.ThingStatus;
  41 import org.openhab.core.thing.ThingStatusInfo;
  42 import org.openhab.core.thing.binding.ThingHandlerCallback;
  43 import org.osgi.service.cm.ConfigurationException;
  44
  45 /**
  46  * Test cases for {@link BrokerHandler}.
  47  *
  48  * @author David Graeff - Initial contribution
  49  */
  50 @ExtendWith(MockitoExtension.class)
  51 @MockitoSettings(strictness = Strictness.WARN)
  52 public class BrokerHandlerTest extends JavaTest {
  53     private ScheduledExecutorService scheduler;
  54
  55     private @Mock ThingHandlerCallback callback;
  56     private @Mock Bridge thing;
  57     private @Mock MqttService service;
  58
  59     private MqttBrokerConnectionEx connection;
  60
  61     private BrokerHandler handler;
  62
  63     @BeforeEach
  64     public void setUp() {
  65         scheduler = new ScheduledThreadPoolExecutor(1);
  66         connection = spy(new MqttBrokerConnectionEx("10.10.0.10", 80, false, "BrokerHandlerTest"));
  67         connection.setTimeoutExecutor(scheduler, 10000);
  68         connection.setConnectionCallback(connection);
  69
  70         Configuration config = new Configuration();
  71         when(thing.getConfiguration()).thenReturn(config);
  72
  73         handler = spy(new BrokerHandlerEx(thing, connection));
  74         handler.setCallback(callback);
  75     }
  76
  77     @AfterEach
  78     public void tearDown() {
  79         scheduler.shutdownNow();
  80     }
  81
  82     @Test
  83     public void handlerInitWithoutUrl() throws IllegalArgumentException {
  84         // Assume it is a real handler and not a mock as defined above
  85         handler = new BrokerHandler(thing);
  86         assertThrows(IllegalArgumentException.class, () -> initializeHandlerWaitForTimeout());
  87     }
  88
  89     @Test
  90     public void createBrokerConnection() {
  91         Configuration config = new Configuration();
  92         config.put("host", "10.10.0.10");
  93         config.put("port", 80);
  94         when(thing.getConfiguration()).thenReturn(config);
  95         handler.initialize();
  96         verify(handler).createBrokerConnection();
  97     }
  98
  99     @Test
 100     public void handlerInit() throws InterruptedException, IllegalArgumentException {
 101         assertThat(initializeHandlerWaitForTimeout(), is(true));
 102
 103         ArgumentCaptor<ThingStatusInfo> statusInfoCaptor = ArgumentCaptor.forClass(ThingStatusInfo.class);
 104         verify(callback, atLeast(3)).statusUpdated(eq(thing), statusInfoCaptor.capture());
 105         assertThat(statusInfoCaptor.getValue().getStatus(), is(ThingStatus.ONLINE));
 106     }
 107
 108     /**
 109      * Utility method for tests that need the handler to be initialized to go on.
 110      *
 111      * @return Return true if successful. You usually want to use:
 112      *         assertThat(initializeHandlerWaitForTimeout(), is(true));
 113      * @throws InterruptedException
 114      * @throws IllegalArgumentException
 115      * @throws MqttException
 116      * @throws ConfigurationException
 117      */
 118     boolean initializeHandlerWaitForTimeout() throws InterruptedException, IllegalArgumentException {
 119         MqttBrokerConnection c = connection;
 120
 121         MqttConnectionObserverEx o = new MqttConnectionObserverEx();
 122         c.addConnectionObserver(o);
 123
 124         assertThat(connection.connectionState(), is(MqttConnectionState.DISCONNECTED));
 125         handler.initialize();
 126         waitForAssert(() -> verify(connection, times(2)).addConnectionObserver(any()));
 127         waitForAssert(() -> verify(connection, times(1)).start());
 128         // First we expect a CONNECTING state and then a CONNECTED unique state change
 129         waitForAssert(() -> assertThat(o.counter, is(2)));
 130         // First we expect a CONNECTING state and then a CONNECTED state change
 131         // (and other CONNECTED after the future completes)
 132         waitForAssert(() -> verify(handler, times(3)).connectionStateChanged(any(), any()));
 133         return true;
 134     }
 135 }