def test_reset():
"""
Test that reset method resets variables back to correct default values.
"""
n = NodeManager(startup_nodes=[{}])
n.initialize = Mock()
n.slots = {"foo": "bar"}
n.nodes = ["foo", "bar"]
n.reset()
assert n.slots == {}
assert n.nodes == {}
def test_reset():
"""
Test that reset method resets variables back to correct default values.
"""
n = NodeManager(startup_nodes=[{}])
n.initialize = Mock()
n.slots = {"foo": "bar"}
n.nodes = ["foo", "bar"]
n.reset()
assert n.slots == {}
assert n.nodes == {}
def test_determine_pubsub_node():
"""
Given a set of nodes it should determine the same pubsub node each time.
"""
n = NodeManager(startup_nodes=[{}])
n.nodes = {
"127.0.0.1:7001": {"host": "127.0.0.1", "port": 7001, "server_type": "master"},
"127.0.0.1:7005": {"host": "127.0.0.1", "port": 7005, "server_type": "master"},
"127.0.0.1:7000": {"host": "127.0.0.1", "port": 7000, "server_type": "master"},
"127.0.0.1:7002": {"host": "127.0.0.1", "port": 7002, "server_type": "master"},
}
n.determine_pubsub_node()
assert n.pubsub_node == {"host": "127.0.0.1", "port": 7005, "server_type": "master", "pubsub": True}
def test_determine_pubsub_node():
"""
Given a set of nodes it should determine the same pubsub node each time.
"""
n = NodeManager(startup_nodes=[{}])
n.nodes = {
"127.0.0.1:7001": {
"host": "127.0.0.1",
"port": 7001,
"server_type": "master"
},
"127.0.0.1:7005": {
"host": "127.0.0.1",
"port": 7005,
"server_type": "master"
},
"127.0.0.1:7000": {
"host": "127.0.0.1",
"port": 7000,
"server_type": "master"
},
"127.0.0.1:7002": {
"host": "127.0.0.1",
"port": 7002,
"server_type": "master"
},
}
n.determine_pubsub_node()
assert n.pubsub_node == {
"host": "127.0.0.1",
"port": 7005,
"server_type": "master",
"pubsub": True
}
