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

   1 /**
   2  * Copyright (c) 2010-2023 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.dbquery.internal.dbimpl.influx2;
  14
  15 import static org.mockito.Mockito.mock;
  16
  17 import java.time.Instant;
  18 import java.util.function.BiConsumer;
  19 import java.util.function.Consumer;
  20
  21 import org.eclipse.jdt.annotation.NonNullByDefault;
  22 import org.openhab.binding.dbquery.internal.error.DatabaseException;
  23
  24 import com.influxdb.Cancellable;
  25 import com.influxdb.query.FluxRecord;
  26
  27 /**
  28  *
  29  * @author Joan Pujol - Initial contribution
  30  */
  31 @NonNullByDefault
  32 public class InfluxDBClientFacadeMock implements InfluxDBClientFacade {
  33     public static final String INVALID_QUERY = "invalid";
  34     public static final String EMPTY_QUERY = "empty";
  35     public static final String SCALAR_QUERY = "scalar";
  36     public static final String MULTIPLE_ROWS_QUERY = "multiple";
  37
  38     public static final String SCALAR_RESULT = "scalarResult";
  39     public static final int MULTIPLE_ROWS_SIZE = 3;
  40     public static final String VALUE_COLUMN = "_value";
  41     public static final String TIME_COLUMN = "_time";
  42     public static final String MULTIPLE_ROWS_VALUE_PREFIX = "value";
  43
  44     boolean connected;
  45
  46     @Override
  47     public boolean connect() {
  48         connected = true;
  49         return true;
  50     }
  51
  52     @Override
  53     public boolean isConnected() {
  54         return connected;
  55     }
  56
  57     @Override
  58     public boolean disconnect() {
  59         connected = false;
  60         return true;
  61     }
  62
  63     @Override
  64     public void query(String queryString, BiConsumer<Cancellable, FluxRecord> onNext,
  65             Consumer<? super Throwable> onError, Runnable onComplete) {
  66         if (!connected) {
  67             throw new DatabaseException("Client not connected");
  68         }
  69
  70         if (INVALID_QUERY.equals(queryString)) {
  71             onError.accept(new RuntimeException("Invalid query"));
  72         } else if (EMPTY_QUERY.equals(queryString)) {
  73             onComplete.run();
  74         } else if (SCALAR_QUERY.equals(queryString)) {
  75             FluxRecord scalar = new FluxRecord(0);
  76             scalar.getValues().put("result", "_result");
  77             scalar.getValues().put("table", 0);
  78             scalar.getValues().put(VALUE_COLUMN, SCALAR_RESULT);
  79             onNext.accept(mock(Cancellable.class), scalar);
  80             onComplete.run();
  81         } else if (MULTIPLE_ROWS_QUERY.equals(queryString)) {
  82             onNext.accept(mock(Cancellable.class), createRowRecord(0, MULTIPLE_ROWS_VALUE_PREFIX + 1));
  83             onNext.accept(mock(Cancellable.class), createRowRecord(0, MULTIPLE_ROWS_VALUE_PREFIX + 2));
  84             onNext.accept(mock(Cancellable.class), createRowRecord(1, MULTIPLE_ROWS_VALUE_PREFIX + 3));
  85             onComplete.run();
  86         }
  87     }
  88
  89     private static FluxRecord createRowRecord(int table, String value) {
  90         FluxRecord record = new FluxRecord(0);
  91         record.getValues().put("result", "_result");
  92         record.getValues().put("table", table);
  93         record.getValues().put(VALUE_COLUMN, value);
  94         record.getValues().put(TIME_COLUMN, Instant.now());
  95         record.getValues().put("_start", Instant.now());
  96         record.getValues().put("_stop", Instant.now());
  97         record.getValues().put("_measurement", "measurementName");
  98         return record;
  99     }
 100 }