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 }