|
4 | 4 |
|
5 | 5 | require 'redis_client'
|
6 | 6 |
|
7 |
| -TEST_REDIS_HOST = '127.0.0.1' |
| 7 | +TEST_REDIS_HOST = ENV.fetch('REDIS_HOST', '127.0.0.1') |
8 | 8 | TEST_REDIS_PORT = 6379
|
9 | 9 | TEST_TIMEOUT_SEC = 5.0
|
10 | 10 | TEST_RECONNECT_ATTEMPTS = 3
|
|
56 | 56 | TEST_NUMBER_OF_REPLICAS = TEST_REPLICA_SIZE * TEST_SHARD_SIZE
|
57 | 57 | TEST_NUMBER_OF_NODES = TEST_SHARD_SIZE + TEST_NUMBER_OF_REPLICAS
|
58 | 58 |
|
59 |
| -TEST_REDIS_PORTS = TEST_REDIS_PORT.upto(TEST_REDIS_PORT + TEST_NUMBER_OF_NODES - 1).to_a.freeze |
60 |
| -TEST_NODE_URIS = TEST_REDIS_PORTS.map { |v| "#{TEST_REDIS_SCHEME}://#{TEST_REDIS_HOST}:#{v}" }.freeze |
61 |
| -TEST_NODE_OPTIONS = TEST_REDIS_PORTS.to_h { |v| ["#{TEST_REDIS_HOST}:#{v}", { host: TEST_REDIS_HOST, port: v }] }.freeze |
| 59 | +case TEST_REDIS_HOST |
| 60 | +when '127.0.0.1', 'localhost' |
| 61 | + TEST_REDIS_PORTS = TEST_REDIS_PORT.upto(TEST_REDIS_PORT + TEST_NUMBER_OF_NODES - 1).to_a.freeze |
| 62 | + TEST_NODE_URIS = TEST_REDIS_PORTS.map { |v| "#{TEST_REDIS_SCHEME}://#{TEST_REDIS_HOST}:#{v}" }.freeze |
| 63 | + TEST_NODE_OPTIONS = TEST_REDIS_PORTS.to_h { |v| ["#{TEST_REDIS_HOST}:#{v}", { host: TEST_REDIS_HOST, port: v }] }.freeze |
| 64 | +when 'node1' |
| 65 | + TEST_REDIS_PORTS = Array.new(TEST_NUMBER_OF_NODES) { TEST_REDIS_PORT }.freeze |
| 66 | + TEST_NODE_URIS = Array.new(TEST_NUMBER_OF_NODES) do |i| |
| 67 | + host = "node#{i + 1}" |
| 68 | + "#{TEST_REDIS_SCHEME}://#{host}:#{TEST_REDIS_PORT}" |
| 69 | + end.freeze |
| 70 | + |
| 71 | + TEST_NODE_OPTIONS = Array.new(TEST_NUMBER_OF_NODES) do |i| |
| 72 | + host = "node#{format("%#{TEST_NUMBER_OF_NODES}d", i + 1)}" |
| 73 | + ["#{host}:#{TEST_REDIS_PORT}", { host: host, port: TEST_REDIS_PORT }] |
| 74 | + end.to_h.freeze |
| 75 | +else |
| 76 | + raise NotImplementedError, TEST_REDIS_HOST |
| 77 | +end |
62 | 78 |
|
63 | 79 | TEST_GENERIC_OPTIONS = (TEST_REDIS_SSL ? _base_opts.merge(_ssl_opts) : _base_opts).freeze
|
64 | 80 |
|
|
0 commit comments