Added redis to consumerlocator

This commit is contained in:
2020-04-17 16:47:27 +02:00
parent eff5f4191c
commit 4a3fc3b1ba
3 changed files with 33 additions and 23 deletions

16
test.py
View File

@@ -8,6 +8,7 @@ import re
import consumerlocator
import communicator
import messagesender
import redisconnector
__author__ = "@tormakris"
@@ -196,7 +197,8 @@ def test_learnconsumerlist(httpserver):
uuid=generateduuid)
consumerlocator.KNOWNCONSUMER = f"127.0.0.1:{port}"
locator = consumerlocator.ConsumerLocator(
uuid=generateduuid, communicator=comm)
uuid=generateduuid, communicator=comm,
redisconnector=redisconnector.RedisConnector())
ret = locator.learnconsumerlist()
assert ret is None
@@ -213,7 +215,8 @@ def test_getcurrentconsumer(mocker):
currentconsumer="127.0.0.1",
uuid=generateduuid)
locator = consumerlocator.ConsumerLocator(
uuid=generateduuid, communicator=comm)
uuid=generateduuid, communicator=comm,
redisconnector=redisconnector.RedisConnector())
assert locator.getcurrentconsumer() == consumerlocator.KNOWNCONSUMER
@@ -236,7 +239,8 @@ def test_checkcurrentconsumer(httpserver):
uuid=generateduuid)
consumerlocator.KNOWNCONSUMER = f"127.0.0.1:{port}"
locator = consumerlocator.ConsumerLocator(
uuid=generateduuid, communicator=comm)
uuid=generateduuid, communicator=comm,
redisconnector=redisconnector.RedisConnector())
ret = locator.checkcurrentconsumer()
assert ret == True
@@ -260,7 +264,8 @@ def test_updateconsumer(httpserver):
uuid=generateduuid)
consumerlocator.KNOWNCONSUMER = f"127.0.0.1:{port}"
locator = consumerlocator.ConsumerLocator(
uuid=generateduuid, communicator=comm)
uuid=generateduuid, communicator=comm,
redisconnector=redisconnector.RedisConnector())
assert locator.currentconsumer is not None
ret = locator.updateconsumer()
assert ret == f"127.0.0.1:{port}"
@@ -285,6 +290,7 @@ def test_updateconsumerlist(httpserver):
uuid=generateduuid)
consumerlocator.KNOWNCONSUMER = f"127.0.0.1:{port}"
locator = consumerlocator.ConsumerLocator(
uuid=generateduuid, communicator=comm)
uuid=generateduuid, communicator=comm,
redisconnector=redisconnector.RedisConnector())
ret = locator.updateconsumerlist()
assert ret is None