mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
8190ff1fd0
* build_credis.sh: use an up-to-date credis commit. * build_credis.sh: leveldb is updated, so update build cmds for it * WIP: make monitor.py issue flush; switch gcs client to use credis * Experimental: enable automatic GCS flushing with configurable policy. * Fix linux compilation error * Fix leveldb build * Use optimized build for credis * Address comments * Attempt to fix tests
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
from __future__ import absolute_import, division, print_function
|
|
|
|
import os
|
|
import unittest
|
|
|
|
import redis
|
|
|
|
import ray
|
|
|
|
|
|
def parse_client(addr_port_str):
|
|
redis_address, redis_port = addr_port_str.split(":")
|
|
return redis.StrictRedis(host=redis_address, port=redis_port)
|
|
|
|
|
|
@unittest.skipIf(not os.environ.get('RAY_USE_NEW_GCS', False),
|
|
"Tests functionality of the new GCS.")
|
|
class CredisTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.config = ray.init(num_workers=0)
|
|
|
|
def tearDown(self):
|
|
ray.worker.cleanup()
|
|
|
|
def test_credis_started(self):
|
|
assert "redis_address" in self.config
|
|
primary = parse_client(self.config['redis_address'])
|
|
assert primary.ping() is True
|
|
member = primary.lrange('RedisShards', 0, -1)[0]
|
|
shard = parse_client(member.decode())
|
|
|
|
# TODO(zongheng): remove these next four lines of horror, once task
|
|
# table is correctly placed in the data shard & swapping master and
|
|
# member modules.
|
|
member = self.config['redis_address']
|
|
temp = primary
|
|
primary = shard
|
|
shard = temp
|
|
|
|
# Check that primary has loaded credis' master module.
|
|
chain = primary.execute_command('MASTER.GET_CHAIN')
|
|
assert len(chain) == 1
|
|
|
|
# Check that the shard has loaded credis' member module.
|
|
assert chain[0].decode() == member
|
|
assert shard.execute_command('MEMBER.SN') == -1
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main(verbosity=2)
|