Added redis to consumerlocator
This commit is contained in:
16
test.py
16
test.py
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user