[timseries] Fix tests

Author: nepython

.travis.yml

@@ -24,29 +24,28 @@ addons:
 
 services:
   - docker
-  - redis-server
-  - elasticsearch
 
 branches:
   only:
     - master
     - dev
 
 before_install:
-  # - docker run -d --name influxdb -e INFLUXDB_DB=openwisp2 -p 8086:8086 influxdb:alpine
-  # - docker run -p 9200:9200 docker.elastic.co/elasticsearch/elasticsearch:7.8.0
+  - docker-compose up -d
   - pip install -U pip wheel setuptools
   - pip install $DJANGO
   - pip install -U -r requirements-test.txt
 
 install:
-  - pip install -e .
+  - pip install -e .[influxdb,elasticsearch]
   - sh install-dev.sh
 
 script:
   - ./run-qa-checks
   - SAMPLE_APP=1 coverage run --source=openwisp_monitoring runtests.py
   - coverage run -a --source=openwisp_monitoring runtests.py
+  - elasticsearch=1 SAMPLE_APP=1 coverage run -a --source=openwisp_monitoring runtests.py
+  - elasticsearch=1 coverage run -a --source=openwisp_monitoring runtests.py
 
 jobs:
   include:

Dockerfile

@@ -19,6 +19,7 @@ WORKDIR /opt/openwisp/tests/
 ENV NAME=openwisp-monitoring \
     PYTHONBUFFERED=1 \
     INFLUXDB_HOST=influxdb \
-    REDIS_HOST=redis
+    REDIS_HOST=redis \
+    ELASTICSEARCH_HOST=es01
 CMD ["sh", "docker-entrypoint.sh"]
 EXPOSE 8000

docker-compose.yml

@@ -11,6 +11,8 @@ services:
     depends_on:
       - influxdb
       - redis
+      - es01
+      - es02
 
   influxdb:
     image: influxdb:1.8-alpine
@@ -23,24 +25,45 @@ services:
       INFLUXDB_USER: openwisp
       INFLUXDB_USER_PASSWORD: openwisp
   # clustered version of elasticsearch is used as that might be used in production
-  elasticsearch:
+  es01:
     image: docker.elastic.co/elasticsearch/elasticsearch:7.8.0
     container_name: es01
     environment:
-      - node.name: openwisp2
-      - cluster.name: openwisp2
-      - bootstrap.memory_lock: true
+      - "node.name=es01"
+      - "discovery.seed_hosts=es02"
+      - "cluster.initial_master_nodes=es01,es02"
+      - "cluster.name=openwisp2"
+      - "bootstrap.memory_lock=true"
       - "ES_JAVA_OPTS=-Xms512m -Xmx512m"
     ulimits:
       memlock:
         soft: -1
         hard: -1
     volumes:
-      - data01:/usr/share/elasticsearch/data
+      - esdata01:/usr/share/elasticsearch/data
     ports:
       - 9200:9200
     networks:
-      - elastic
+      - esnet
+  es02:
+    image: docker.elastic.co/elasticsearch/elasticsearch:7.8.0
+    container_name: es02
+    environment:
+      - "node.name=es02"
+      - "discovery.seed_hosts=es01"
+      - "cluster.initial_master_nodes=es01,es02"
+      - "cluster.name=openwisp2"
+      - "bootstrap.memory_lock=true"
+      - "ES_JAVA_OPTS=-Xms512m -Xmx512m"
+    ulimits:
+      memlock:
+        soft: -1
+        hard: -1
+    volumes:
+      - esdata02:/usr/share/elasticsearch/data
+    networks:
+      - esnet
+
   redis:
     image: redis:5.0-alpine
     ports:
@@ -49,7 +72,10 @@ services:
 
 volumes:
   influxdb-data: {}
+  esdata01:
+    driver: local
+  esdata02:
+    driver: local
 
 networks:
-  elastic:
-    driver: bridge
+  esnet:

openwisp_monitoring/db/backends/__init__.py

@@ -49,6 +49,7 @@ def load_backend_module(backend_name=TIMESERIES_DB['BACKEND'], module=None):
         # The database backend wasn't found. Display a helpful error message
         # listing all built-in database backends.
         builtin_backends = ['influxdb', 'elasticsearch']
+        raise e
         if backend_name not in [
             f'openwisp_monitoring.db.backends.{b}' for b in builtin_backends
         ]: