fix unit tests

.drone.yml

@@ -15,9 +15,11 @@ steps:
 
 - name: unit_test
   image: python:3.8
+  environment:
+    PRODUCER_REDIS: cache
   commands:
   - pip3 install -r requirements.txt
-  - echo -e 'pytest test.py'
+  - pytest test.py
 
 - name: build-app
   image: banzaicloud/drone-kaniko
@@ -75,6 +77,8 @@ services:
   volumes:
   - name: dockersock
     path: /var/run
+- name: cache
+  image: redis
 
 volumes:
 - name: dockersock

communicator.py

@@ -16,6 +16,7 @@ __version__text__ = "1"
 logging.basicConfig(level=logging.INFO)
 LOGGER = logging.getLogger(__name__)
 
+
 class Communicator:
     """
     Class handling low level communication with consumers.

consumerlocator.py

@@ -18,6 +18,7 @@ __version__text__ = "1"
 KNOWNCONSUMER = os.getenv("PRODUCER_KNOWNCONSUMER", '10.69.42.1')
 LOGGER = logging.getLogger(__name__)
 
+
 class ConsumerLocator:
 
     """

test.py

@@ -17,8 +17,8 @@ __module_name__ = "test"
 __version__text__ = "1"
 
 generateduuid = 'c959ad81-58f9-4445-aab4-8f3d68aee1ad'
-redis_proc = factories.redis_proc(port=6379)
-redis_db = factories.redisdb('redis_db')
+redis_proc = factories.redis_proc(host='cache', port=6379)
+redis_db = factories.redisdb('redis_nooproc')
 
 
 def test_generate_string(mocker):
@@ -31,7 +31,7 @@ def test_generate_string(mocker):
     comm = communicator.Communicator(
         currentconsumer="localhost",
         uuid=generateduuid)
-    mess = messagesender.MessageSender(communicator=comm,  uuid=generateduuid)
+    mess = messagesender.MessageSender(communicator=comm, uuid=generateduuid)
     msg = mess.randomstring(stringlength=32)
     assert isinstance(msg, str)
     assert len(msg) == 32
@@ -71,7 +71,7 @@ def test_send_message(mocker):
     comm = communicator.Communicator(
         currentconsumer="127.0.0.1",
         uuid=generateduuid)
-    mess = messagesender.MessageSender(communicator=comm,  uuid=generateduuid)
+    mess = messagesender.MessageSender(communicator=comm, uuid=generateduuid)
     messa = "SENDING"
     msg = mess.sendmessage(message=messa)
     assert msg is None